fix: ASync iter classes (#79)

a_sync/iter.py

@@ -1,33 +1,39 @@
 
 import asyncio
-from typing import AsyncIterable, Iterator, TypeVar
+from typing import AsyncIterable, AsyncIterator, Iterable, Iterator, TypeVar
 
 T = TypeVar('T')
 
-class ASyncIterable(AsyncIterable[T]):
-    """An iterable that can be used in both a `for` loop and an `async for` loop."""
+class ASyncIterable(AsyncIterable[T], Iterable[T]):
+    """An abstract iterable implementation that can be used in both a `for` loop and an `async for` loop."""
     def __iter__(self) -> Iterator[T]:
-        return self.__sync_iterator()
-    def __sync_iterator(self) -> Iterator[T]:
         aiterator = self.__aiter__()
         while True:
-            try:
-                yield asyncio.get_event_loop().run_until_complete(aiterator.__anext__())
-            except StopAsyncIteration as e:
-                raise StopIteration(*e.args) from e
+            yield asyncio.get_event_loop().run_until_complete(aiterator.__anext__())
+    @classmethod
+    def wrap(self, aiterable: AsyncIterable[T]) -> "ASyncWrappedIterable[T]":
+        return ASyncWrappedIterable(aiterable)
 
-class ASyncIterator(ASyncIterable[T]):
-    """An iterator that can be used in both a `for` loop and an `async for` loop."""
+class ASyncIterator(AsyncIterator[T], Iterator[T]):
+    """An abstract iterator implementation that can be used in both a `for` loop and an `async for` loop."""
     def __next__(self) -> T:
-        return asyncio.get_event_loop().run_until_complete(self.__anext__())
+        try:
+            return asyncio.get_event_loop().run_until_complete(self.__anext__())
+        except StopAsyncIteration as e:
+            raise StopIteration from e
+    @classmethod
+    def wrap(self, aiterator: AsyncIterator[T]) -> "ASyncWrappedIterator[T]":
+        return ASyncWrappedIterator(aiterator)
 
 class ASyncWrappedIterable(ASyncIterable[T]):
     def __init__(self, async_iterable: AsyncIterable[T]):
-        self.__iterable = async_iterable.__aiter__()
+        self.__aiterable = async_iterable
     def __aiter__(self) -> AsyncIterable[T]:
-        return self.__iterable
-
-class ASyncWrappedIterator(ASyncWrappedIterable[T], ASyncIterator[T]):
+        return self.__aiterable.__aiter__()
+
+class ASyncWrappedIterator(ASyncIterator[T]):
+    def __init__(self, async_iterator: AsyncIterator[T]):
+        self.__aiterator = async_iterator
     async def __anext__(self) -> T:
-        return await self.__iterable.__anext__()
+        return await self.__aiterator.__anext__()