Fix py3.9

.travis.yml

@@ -15,7 +15,12 @@ matrix:
   - python: "3.5"
     env: TOXENV=py35
   - python: "3.6"
-    env: TOXENV=py36-cov
+    env: TOXENV=py36
+    after_success:
+    - coverage combine
+    - codecov
+  - python: "3.9"
+    env: TOXENV=py39
     after_success:
     - coverage combine
     - codecov

setup.py

@@ -12,6 +12,8 @@
     'Programming Language :: Python :: 3.5',
     'Programming Language :: Python :: 3.6',
     'Programming Language :: Python :: 3.7',
+    'Programming Language :: Python :: 3.8',
+    'Programming Language :: Python :: 3.9',
     'Programming Language :: Python :: Implementation :: CPython',
     'Programming Language :: Python',
     'Topic :: Software Development',

src/aiocontext/context.py

@@ -1,6 +1,7 @@
 """Context information storage for asyncio."""
 
 import asyncio
+import sys
 from collections import ChainMap
 from collections.abc import MutableMapping
 from contextlib import suppress
@@ -70,7 +71,11 @@ def get_data(self, task=None):
             {'key': 'value'}
         """
         if task is None:
-            task = asyncio.Task.current_task()
+            if sys.version_info < (3, 9):
+                task = asyncio.Task.current_task()
+            else:
+                task = asyncio.current_task()
+
             if task is None:
                 raise EventLoopError("No event loop found")
         data = getattr(task, self._data_attr, None)

src/aiocontext/task_factory.py

@@ -1,12 +1,12 @@
 """Task factory."""
 
 import asyncio
+import sys
 from functools import wraps
 
 from .__about__ import __title__
 from .errors import TaskFactoryError
 
-
 _TASK_FACTORY_ATTR = '_{}_contexts'.format(__title__)
 
 
@@ -56,7 +56,11 @@ def custom_task_factory(loop, coro):
 
     @wraps(task_factory)
     def wrapper(loop, coro):
-        parent_task = asyncio.Task.current_task(loop=loop)
+        if sys.version_info < (3, 9):
+            parent_task = asyncio.Task.current_task(loop=loop)
+        else:
+            parent_task = asyncio.current_task(loop=loop)
+
         child_task = task_factory(loop, coro)
         if child_task._source_traceback:
             del child_task._source_traceback[-1]

tox.ini

@@ -6,6 +6,8 @@ envlist =
     py35
     py36
     py37
+    py38
+    py39
     manifest
     metadata
     doc8