prevent asyncio from hanging

Author: ziloka

proxybroker/api.py

@@ -64,7 +64,7 @@ def __init__(
         stop_broker_on_sigint=True,
         **kwargs,
     ):
-        self._loop = loop or asyncio.get_event_loop_policy().get_event_loop()
+        self._loop = loop
         self._proxies = queue or asyncio.Queue()
         self._resolver = Resolver(loop=self._loop)
         self._timeout = timeout

proxybroker/checker.py

@@ -43,7 +43,7 @@ def __init__(
         self._strict = strict
         self._dnsbl = dnsbl or []
         self._types = types or {}
-        self._loop = loop or asyncio.get_event_loop()
+        self._loop = loop
         self._resolver = Resolver(loop=self._loop)
 
         self._req_http_proto = not types or bool(

proxybroker/cli.py

@@ -387,7 +387,7 @@ def cli(args=sys.argv[1:]):
         ns.types.remove('HTTP')
         ns.types.append(('HTTP', ns.anon_lvl))
 
-    loop = asyncio.get_event_loop_policy().get_event_loop()
+    loop = asyncio.new_event_loop()
     proxies = asyncio.Queue()
     broker = Broker(
         proxies,
@@ -443,6 +443,7 @@ def cli(args=sys.argv[1:]):
 
     try:
         if tasks:
+            asyncio.set_event_loop(loop)
             loop.run_until_complete(asyncio.gather(*tasks))
             if ns.show_stats:
                 broker.show_stats(verbose=True)

proxybroker/judge.py

@@ -29,7 +29,7 @@ def __init__(self, url, timeout=8, verify_ssl=False, loop=None):
         self.marks = {'via': 0, 'proxy': 0}
         self.timeout = timeout
         self.verify_ssl = verify_ssl
-        self._loop = loop or asyncio.get_event_loop()
+        self._loop = loop
         self._resolver = Resolver(loop=self._loop)
 
     def __repr__(self):

proxybroker/providers.py

@@ -45,7 +45,7 @@ def __init__(
         self._proxies = set()
         # concurrent connections on the current provider
         self._sem_provider = asyncio.Semaphore(max_conn)
-        self._loop = loop or asyncio.get_event_loop()
+        self._loop = loop
 
     @property
     def proxies(self):

proxybroker/resolver.py

@@ -44,7 +44,7 @@ class Resolver:
 
     def __init__(self, timeout=5, loop=None):
         self._timeout = timeout
-        self._loop = loop or asyncio.get_event_loop()
+        self._loop = loop
         self._resolver = aiodns.DNSResolver(loop=self._loop)
 
     @staticmethod

proxybroker/server.py

@@ -131,7 +131,7 @@ def __init__(
     ):
         self.host = host
         self.port = int(port)
-        self._loop = loop or asyncio.get_event_loop()
+        self._loop = loop
         self._timeout = timeout
         self._max_tries = max_tries
         self._backlog = backlog