LBRYFoundation/LBRY-Vault
1
0
Fork 0
mirror of https://github.com/LBRYFoundation/LBRY-Vault.git synced 2025-09-01 09:45:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

interface: change how GracefulDisconnect is handled

Browse source
This commit is contained in:
SomberNight 2018-09-12 18:43:50 +02:00
parent 6f7a065081
commit 1419a5c60d
No known key found for this signature in database
GPG key ID: B33B5F232C6271E9
1 changed files with 11 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
electrum/interface.py
View file

@ -96,8 +96,7 @@ class NotificationSession(ClientSession):
return str(method) + repr(params) return str(method) + repr(params)
# FIXME this is often raised inside a TaskGroup, but then it's not silent :( class GracefulDisconnect(Exception): pass
class GracefulDisconnect(AIOSafeSilentException): pass
class ErrorParsingSSLCert(Exception): pass class ErrorParsingSSLCert(Exception): pass
@ -226,7 +225,17 @@ class Interface(PrintError):
sslc.check_hostname = 0 sslc.check_hostname = 0
return sslc return sslc
def handle_graceful_disconnect(func):
async def wrapper_func(self, *args, **kwargs):
try:
return await func(self, *args, **kwargs)
except GracefulDisconnect as e:
self.print_error("disconnecting gracefully. {}".format(e))
self.exception = e
return wrapper_func
@aiosafe @aiosafe
@handle_graceful_disconnect
async def run(self): async def run(self):
try: try:
ssl_context = await self._get_ssl_context() ssl_context = await self._get_ssl_context()