highperfocused/lnbits
1
0
Fork 0
mirror of https://github.com/lnbits/lnbits.git synced 2025-11-23 12:29:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

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
7 changed files with 28 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
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):
await self.client.aclose() try:
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)

5
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):
await self.client.aclose() try:
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:

5
lnbits/wallets/lndrest.py
View File

@@ -69,7 +69,10 @@ class LndRestWallet(Wallet):
) )
async def cleanup(self): async def cleanup(self):
await self.client.aclose() try:
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:

5
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):
await self.client.aclose() try:
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}"

5
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):
await self.client.aclose() try:
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)

5
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):
await self.client.aclose() try:
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:

5
lnbits/wallets/spark.py
View File

@@ -37,7 +37,10 @@ class SparkWallet(Wallet):
) )
async def cleanup(self): async def cleanup(self):
await self.client.aclose() try:
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):