Bump to 2.0.1

aio-libs · Mar 13, 2018 · 68d60a9 · 68d60a9
1 parent c7d61a8
commit 68d60a9
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -1,6 +1,12 @@
 CHANGES
 =======
 
+2.0.1 (2018-03-13)
+------------------
+
+* Fix ``PendingDeprecationWarning`` on Python 3.7 (#33)
+
+
 2.0.0 (2017-10-09)
 ------------------
 

diff --git a/async_timeout/__init__.py b/async_timeout/__init__.py
@@ -1,7 +1,10 @@
 import asyncio
+import sys
 
 
-__version__ = '2.0.0'
+__version__ = '2.0.1'
+
+PY_37 = sys.version_info >= (3, 7)
 
 
 class timeout:
@@ -89,8 +92,12 @@ def _cancel_task(self):
 
 
 def current_task(loop):
-    task = asyncio.Task.current_task(loop=loop)
+    if PY_37:
+        task = asyncio.current_task(loop=loop)
+    else:
+        task = asyncio.Task.current_task(loop=loop)
     if task is None:
+        # this should be removed, tokio must use register_task and family API
         if hasattr(loop, 'current_task'):
             task = loop.current_task()