flake8'd

Author: smurfix

tests/test_concurrent.py

@@ -52,7 +52,7 @@ async def _test_same_task():
     def get_loop(i):
         loops[i] = (asyncio.get_event_loop(), asyncio.get_event_loop_policy())
 
-    async with trio.open_nursery() as n:
+    async with trio.open_nursery():
         async with trio_asyncio.open_loop() as loop1:
             async with trio_asyncio.open_loop() as loop2:
                 loop1.call_later(0.1, get_loop, 0)

trio_asyncio/async_.py

@@ -1,5 +1,4 @@
 import trio
-import asyncio
 
 from async_generator import async_generator, yield_, asynccontextmanager
 

trio_asyncio/base.py

@@ -1,6 +1,5 @@
 import os
 import sys
-import inspect
 import math
 import trio
 import heapq

trio_asyncio/loop.py

@@ -36,10 +36,6 @@
 class _TrioPolicy(asyncio.events.BaseDefaultEventLoopPolicy):
     _loop_factory = TrioEventLoop
 
-    def __init__(self):
-        super().__init__()
-        current_loop = trio.TaskLocal(_loop=None, _task=False)
-
     def new_event_loop(self):
         try:
             trio.hazmat.current_task()
@@ -102,6 +98,8 @@ def set_event_loop(self, loop):
 
 from asyncio import events as _aio_event
 
+#####
+
 _orig_policy_get = _aio_event.get_event_loop_policy
 
 
@@ -120,10 +118,12 @@ def _new_policy_get():
 _aio_event.get_event_loop_policy = _new_policy_get
 asyncio.get_event_loop_policy = _new_policy_get
 
+#####
+
 _orig_run_get = _aio_event._get_running_loop
 
 
-def _new_loop_get():
+def _new_run_get():
     try:
         return _current_loop.loop
     except RuntimeError:
@@ -132,6 +132,8 @@ def _new_loop_get():
 
 _aio_event._get_running_loop = _new_run_get
 
+#####
+
 _orig_loop_get = _aio_event.get_event_loop
 
 

trio_asyncio/util.py

@@ -63,7 +63,7 @@ async def consume_next():
             item = STOP
 
         trio.hazmat.reschedule(task, trio.hazmat.Value(value=item))
-        #trio.hazmat.reschedule(task, STOP)
+        # trio.hazmat.reschedule(task, STOP)
 
     def abort_cb(raise_cancel_arg):
         # Save the cancel-raising function