aiolabs/lnbits
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Wallets: Catch errors during cleanup (#1829)

Browse source 
* jetz aba

* catch RuntimeError
This commit is contained in:
callebtc 2023-07-21 09:50:50 +02:00 • committed by GitHub
parent befb288900
commit 2d4c9f349b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 28 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

3
lnbits/wallets/eclair.py
View file

@ -44,7 +44,10 @@ class EclairWallet(Wallet):
self.client = httpx.AsyncClient(base_url=self.url, headers=self.auth) self.client = httpx.AsyncClient(base_url=self.url, headers=self.auth)
async def cleanup(self): async def cleanup(self):
try:
await self.client.aclose() await self.client.aclose()
except RuntimeError as e:
logger.warning(f"Error closing wallet connection: {e}")
async def status(self) -> StatusResponse: async def status(self) -> StatusResponse:
r = await self.client.post("/globalbalance", timeout=5) r = await self.client.post("/globalbalance", timeout=5)

3
lnbits/wallets/lnbits.py
View file

@ -32,7 +32,10 @@ class LNbitsWallet(Wallet):
self.client = httpx.AsyncClient(base_url=self.endpoint, headers=self.key) self.client = httpx.AsyncClient(base_url=self.endpoint, headers=self.key)
async def cleanup(self): async def cleanup(self):
try:
await self.client.aclose() await self.client.aclose()
except RuntimeError as e:
logger.warning(f"Error closing wallet connection: {e}")
async def status(self) -> StatusResponse: async def status(self) -> StatusResponse:
try: try:

3
lnbits/wallets/lndrest.py
View file

@ -69,7 +69,10 @@ class LndRestWallet(Wallet):
) )
async def cleanup(self): async def cleanup(self):
try:
await self.client.aclose() await self.client.aclose()
except RuntimeError as e:
logger.warning(f"Error closing wallet connection: {e}")
async def status(self) -> StatusResponse: async def status(self) -> StatusResponse:
try: try:

3
lnbits/wallets/lnpay.py
View file

@ -35,7 +35,10 @@ class LNPayWallet(Wallet):
self.client = httpx.AsyncClient(base_url=self.endpoint, headers=self.auth) self.client = httpx.AsyncClient(base_url=self.endpoint, headers=self.auth)
async def cleanup(self): async def cleanup(self):
try:
await self.client.aclose() await self.client.aclose()
except RuntimeError as e:
logger.warning(f"Error closing wallet connection: {e}")
async def status(self) -> StatusResponse: async def status(self) -> StatusResponse:
url = f"/wallet/{self.wallet_key}" url = f"/wallet/{self.wallet_key}"

3
lnbits/wallets/lntips.py
View file

@ -33,7 +33,10 @@ class LnTipsWallet(Wallet):
self.client = httpx.AsyncClient(base_url=self.endpoint, headers=self.auth) self.client = httpx.AsyncClient(base_url=self.endpoint, headers=self.auth)
async def cleanup(self): async def cleanup(self):
try:
await self.client.aclose() await self.client.aclose()
except RuntimeError as e:
logger.warning(f"Error closing wallet connection: {e}")
async def status(self) -> StatusResponse: async def status(self) -> StatusResponse:
r = await self.client.get("/api/v1/balance", timeout=40) r = await self.client.get("/api/v1/balance", timeout=40)

3
lnbits/wallets/opennode.py
View file

@ -37,7 +37,10 @@ class OpenNodeWallet(Wallet):
self.client = httpx.AsyncClient(base_url=self.endpoint, headers=self.auth) self.client = httpx.AsyncClient(base_url=self.endpoint, headers=self.auth)
async def cleanup(self): async def cleanup(self):
try:
await self.client.aclose() await self.client.aclose()
except RuntimeError as e:
logger.warning(f"Error closing wallet connection: {e}")
async def status(self) -> StatusResponse: async def status(self) -> StatusResponse:
try: try:

3
lnbits/wallets/spark.py
View file

@ -37,7 +37,10 @@ class SparkWallet(Wallet):
) )
async def cleanup(self): async def cleanup(self):
try:
await self.client.aclose() await self.client.aclose()
except RuntimeError as e:
logger.warning(f"Error closing wallet connection: {e}")
def __getattr__(self, key): def __getattr__(self, key):
async def call(*args, **kwargs): async def call(*args, **kwargs):