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