network.best_effort_reliable: use curio APIs instead of asyncio

electrum/network.py

@@ -37,6 +37,7 @@ import traceback
 import concurrent
 from concurrent import futures
 import copy
+import functools
 
 import aiorpcx
 from aiorpcx import TaskGroup, ignore_after
@@ -829,24 +830,21 @@ class Network(Logger, NetworkRetryManager[ServerAddr]):
         return True
 
     def best_effort_reliable(func):
+        @functools.wraps(func)
         async def make_reliable_wrapper(self: 'Network', *args, **kwargs):
             for i in range(10):
                 iface = self.interface
                 # retry until there is a main interface
                 if not iface:
-                    try:
-                        await asyncio.wait_for(self.default_server_changed_event.wait(), 1)
-                    except asyncio.TimeoutError:
-                        pass
+                    async with ignore_after(1):
+                        await self.default_server_changed_event.wait()
                     continue  # try again
                 assert iface.ready.done(), "interface not ready yet"
                 # try actual request
-                success_fut = asyncio.ensure_future(func(self, *args, **kwargs))
-                await asyncio.wait([success_fut, iface.got_disconnected.wait()], return_when=asyncio.FIRST_COMPLETED)
-                if success_fut.done() and not success_fut.cancelled():
-                    if success_fut.exception():
                 try:
-                            raise success_fut.exception()
+                    async with TaskGroup(wait=any) as group:
+                        task = await group.spawn(func(self, *args, **kwargs))
+                        await group.spawn(iface.got_disconnected.wait())
                 except RequestTimedOut:
                     await iface.close()
                     await iface.got_disconnected.wait()
@@ -857,12 +855,14 @@ class Network(Logger, NetworkRetryManager[ServerAddr]):
                     await iface.close()
                     await iface.got_disconnected.wait()
                     continue  # try again
-                    return success_fut.result()
+                if task.done() and not task.cancelled():
+                    return task.result()
                 # otherwise; try again
             raise BestEffortRequestFailed('no interface to do request on... gave up.')
         return make_reliable_wrapper
 
     def catch_server_exceptions(func):
+        @functools.wraps(func)
         async def wrapper(self, *args, **kwargs):
             try:
                 return await func(self, *args, **kwargs)