Allow custom directory in place of '_async' (#54)

Author: sethmlarson

src/unasync/__init__.py

@@ -64,11 +64,7 @@ def unasync_name(name):
     if name in ASYNC_TO_SYNC:
         return ASYNC_TO_SYNC[name]
     # Convert classes prefixed with 'Async' into 'Sync'
-    elif (
-        len(name) > 5
-        and name.startswith("Async")
-        and name[5].isupper()
-    ):
+    elif len(name) > 5 and name.startswith("Async") and name[5].isupper():
         return "Sync" + name[5:]
     return name
 
@@ -131,7 +127,11 @@ class build_py(orig.build_py):
     and saves them in _sync dir.
     """
 
+    RENAME_DIR_FROM_TO = ("_async", "_sync")  # Controls what directory will be renamed.
+
     def run(self):
+        dir_from, dir_to = self.RENAME_DIR_FROM_TO
+
         self._updated_files = []
 
         # Base class code
@@ -143,8 +143,8 @@ def run(self):
 
         # Our modification!
         for f in self._updated_files:
-            if os.sep + "_async" + os.sep in f:
-                unasync_file(f, "_async", "_sync")
+            if os.sep + dir_from + os.sep in f:
+                unasync_file(f, dir_from, dir_to)
 
         # Remaining base class code
         self.byte_compile(self.get_outputs(include_bytecode=0))
@@ -154,3 +154,10 @@ def build_module(self, module, module_file, package):
         if copied:
             self._updated_files.append(outfile)
         return outfile, copied
+
+
+def customize_build_py(rename_dir_from_to=("_async", "_sync")):
+    class _build_py(build_py):
+        RENAME_DIR_FROM_TO = rename_dir_from_to
+
+    return _build_py

tests/data/example_custom_pkg/setup.py

@@ -0,0 +1,17 @@
+import setuptools
+
+import unasync
+
+setuptools.setup(
+    name="ahip",
+    version="0.0.1",
+    author="Example Author",
+    author_email="author@example.com",
+    description="A package used to test customized unasync",
+    url="https://github.com/pypa/sampleproject",
+    packages=["ahip", "ahip.some_dir"],
+    cmdclass={
+        "build_py": unasync.customize_build_py(rename_dir_from_to=("ahip", "hip"))
+    },
+    package_dir={"": "src"},
+)

tests/data/example_custom_pkg/src/__init__.py

@@ -0,0 +1,2 @@
+async def f():
+    return await 1

tests/data/example_custom_pkg/src/ahip/__init__.py

@@ -0,0 +1,2 @@
+async def f():
+    return await 1

tests/data/example_custom_pkg/src/ahip/another_file.py

@@ -0,0 +1,2 @@
+async def f():
+    return await 1

tests/data/example_custom_pkg/src/ahip/some_dir/__init__.py

@@ -0,0 +1,2 @@
+async def f():
+    return await 1

tests/data/example_custom_pkg/src/ahip/some_dir/another_file.py

@@ -0,0 +1,2 @@
+async def f():
+    return await 1

tests/data/example_custom_pkg/src/ahip/some_dir/some_file.py

@@ -0,0 +1,2 @@
+async def f():
+    return await 1

tests/data/example_custom_pkg/src/ahip/some_file.py

@@ -96,3 +96,19 @@ def test_project_structure_after_build_py_packages(tmpdir):
     )
 
     assert _async_dir_tree == unasynced_dir_tree
+
+
+def test_project_structure_after_customized_build_py_packages(tmpdir):
+
+    source_pkg_dir = os.path.join(TEST_DIR, "example_custom_pkg")
+    pkg_dir = str(tmpdir) + "/" + "example_custom_pkg"
+    shutil.copytree(source_pkg_dir, pkg_dir)
+
+    env = copy.copy(os.environ)
+    env["PYTHONPATH"] = os.path.realpath(os.path.join(TEST_DIR, ".."))
+    subprocess.check_call(["python", "setup.py", "build"], cwd=pkg_dir, env=env)
+
+    _async_dir_tree = list_files(os.path.join(source_pkg_dir, "src/ahip/."))
+    unasynced_dir_tree = list_files(os.path.join(pkg_dir, "build/lib/hip/."))
+
+    assert _async_dir_tree == unasynced_dir_tree