Support renaming custom classes (python-trio#49)

sethmlarson · pquentin · commit 4c4bf5edd22f · 2019-12-10T21:47:18.000+04:00
diff --git a/src/unasync/__init__.py b/src/unasync/__init__.py
@@ -18,6 +18,9 @@
     "__aiter__": "__iter__",
     "__anext__": "__next__",
     "asynccontextmanager": "contextmanager",
+    "AsyncIterable": "Iterable",
+    "AsyncIterator": "Iterator",
+    "AsyncGenerator": "Generator",
     # TODO StopIteration is still accepted in Python 2, but the right change
     # is 'raise StopAsyncIteration' -> 'return' since we want to use unasynced
     # code in Python 3.7+
@@ -68,8 +71,16 @@ def unasync_tokens(tokens):
             # `print( stuff)`
             used_space = space
         else:
-            if toknum == std_tokenize.NAME and tokval in ASYNC_TO_SYNC:
-                tokval = ASYNC_TO_SYNC[tokval]
+            if toknum == std_tokenize.NAME:
+                if tokval in ASYNC_TO_SYNC:
+                    tokval = ASYNC_TO_SYNC[tokval]
+                # Convert classes prefixed with 'Async' into 'Sync'
+                elif (
+                    len(tokval) > 5
+                    and tokval.startswith("Async")
+                    and tokval[5].isupper()
+                ):
+                    tokval = "Sync" + tokval[5:]
             if used_space is None:
                 used_space = space
             yield (used_space, tokval)
diff --git a/tests/data/async/classes.py b/tests/data/async/classes.py
@@ -0,0 +1,3 @@
+class AsyncSocket(object):
+    async def send_all(self, data):
+        ...
diff --git a/tests/data/async/typing.py b/tests/data/async/typing.py
@@ -0,0 +1,5 @@
+import typing
+
+typing.AsyncIterable[bytes]
+typing.AsyncIterator[bytes]
+typing.AsyncGenerator[bytes]
diff --git a/tests/data/sync/classes.py b/tests/data/sync/classes.py
@@ -0,0 +1,3 @@
+class SyncSocket(object):
+    def send_all(self, data):
+        ...
diff --git a/tests/data/sync/typing.py b/tests/data/sync/typing.py
@@ -0,0 +1,5 @@
+import typing
+
+typing.Iterable[bytes]
+typing.Iterator[bytes]
+typing.Generator[bytes]

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+class AsyncSocket(object):`
	`2`	`+ async def send_all(self, data):`
	`3`	`+ ...`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+class SyncSocket(object):`
	`2`	`+ def send_all(self, data):`
	`3`	`+ ...`