add os.listdir and os.walk wrappers

src/aiofiles/base.py

@@ -1,7 +1,7 @@
-"""Various base classes."""
+import asyncio
 from collections.abc import Awaitable
 from contextlib import AbstractAsyncContextManager
-from asyncio import get_running_loop
+from functools import partial, wraps
 
 
 class AsyncBase:
@@ -12,7 +12,7 @@ def __init__(self, file, loop, executor):
 
     @property
     def _loop(self):
-        return self._ref_loop or get_running_loop()
+        return self._ref_loop or asyncio.get_running_loop()
 
     def __aiter__(self):
         """We are our own iterator."""
@@ -63,7 +63,18 @@ async def __aenter__(self):
         return await self
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
-        await get_running_loop().run_in_executor(
+        await asyncio.get_running_loop().run_in_executor(
             None, self._obj._file.__exit__, exc_type, exc_val, exc_tb
         )
         self._obj = None
+
+
+def wrap(func):
+    @wraps(func)
+    async def run(*args, loop=None, executor=None, **kwargs):
+        if loop is None:
+            loop = asyncio.get_running_loop()
+        pfunc = partial(func, *args, **kwargs)
+        return await loop.run_in_executor(executor, pfunc)
+
+    return run

src/aiofiles/os.py

@@ -2,8 +2,9 @@
 
 import os
 
-from . import ospath as path
-from .ospath import wrap
+from aiofiles.base import wrap
+from aiofiles import ospath as path
+
 
 __all__ = [
     "path",
@@ -25,34 +26,40 @@
     "wrap",
     "getcwd",
 ]
-if hasattr(os, "link"):
-    __all__ += ["link"]
-if hasattr(os, "sendfile"):
-    __all__ += ["sendfile"]
-if hasattr(os, "statvfs"):
-    __all__ += ["statvfs"]
 
 
-stat = wrap(os.stat)
+access = wrap(os.access)
+
+getcwd = wrap(os.getcwd)
+
+listdir = wrap(os.listdir)
+
+makedirs = wrap(os.makedirs)
+mkdir = wrap(os.mkdir)
+
+readlink = wrap(os.readlink)
+remove = wrap(os.remove)
+removedirs = wrap(os.removedirs)
 rename = wrap(os.rename)
 renames = wrap(os.renames)
 replace = wrap(os.replace)
-remove = wrap(os.remove)
-unlink = wrap(os.unlink)
-mkdir = wrap(os.mkdir)
-makedirs = wrap(os.makedirs)
 rmdir = wrap(os.rmdir)
-removedirs = wrap(os.removedirs)
-symlink = wrap(os.symlink)
-readlink = wrap(os.readlink)
-listdir = wrap(os.listdir)
+
 scandir = wrap(os.scandir)
-access = wrap(os.access)
-getcwd = wrap(os.getcwd)
+stat = wrap(os.stat)
+symlink = wrap(os.symlink)
+
+unlink = wrap(os.unlink)
+
+walk = wrap(os.walk)
+
 
 if hasattr(os, "link"):
+    __all__ += ["link"]
     link = wrap(os.link)
 if hasattr(os, "sendfile"):
+    __all__ += ["sendfile"]
     sendfile = wrap(os.sendfile)
 if hasattr(os, "statvfs"):
+    __all__ += ["statvfs"]
     statvfs = wrap(os.statvfs)

src/aiofiles/ospath.py

@@ -1,30 +1,23 @@
 """Async executor versions of file functions from the os.path module."""
 
-import asyncio
-from functools import partial, wraps
 from os import path
 
+from aiofiles.base import wrap
 
-def wrap(func):
-    @wraps(func)
-    async def run(*args, loop=None, executor=None, **kwargs):
-        if loop is None:
-            loop = asyncio.get_running_loop()
-        pfunc = partial(func, *args, **kwargs)
-        return await loop.run_in_executor(executor, pfunc)
-
-    return run
 
+abspath = wrap(path.abspath)
 
 exists = wrap(path.exists)
-isfile = wrap(path.isfile)
-isdir = wrap(path.isdir)
-islink = wrap(path.islink)
-ismount = wrap(path.ismount)
+
 getsize = wrap(path.getsize)
 getmtime = wrap(path.getmtime)
 getatime = wrap(path.getatime)
 getctime = wrap(path.getctime)
+
+isfile = wrap(path.isfile)
+isdir = wrap(path.isdir)
+islink = wrap(path.islink)
+ismount = wrap(path.ismount)
+
 samefile = wrap(path.samefile)
 sameopenfile = wrap(path.sameopenfile)
-abspath = wrap(path.abspath)