create missing directory when creating package init file (#34)

dillon-giacoppo · Apr 21, 2020 · 9c03622 · 9c03622 · alexeagle · Aug 19, 2020
1 parent 825ca18
commit 9c03622
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/extract_wheels/lib/namespace_pkgs.py b/extract_wheels/lib/namespace_pkgs.py
@@ -84,6 +84,8 @@ def add_pkgutil_style_namespace_pkg_init(dir_path: str) -> None:
 
     if os.path.isfile(ns_pkg_init_filepath):
         raise ValueError("%s already contains an __init__.py file." % dir_path)
+    if not os.path.exists(dir_path):
+        os.makedirs(dir_path)
     with open(ns_pkg_init_filepath, "w") as ns_pkg_init_f:
         # See https://packaging.python.org/guides/packaging-namespace-packages/#pkgutil-style-namespace-packages
         ns_pkg_init_f.write(

diff --git a/extract_wheels/lib/namespace_pkgs_test.py b/extract_wheels/lib/namespace_pkgs_test.py
@@ -131,5 +131,13 @@ def test_empty_case(self):
         self.assertEqual(actual, set())
 
 
+class TestaddPkgutilStyleNamespacePkgInit(unittest.TestCase):
+    def test_missing_directory_is_created(self):
+        directory = TempDir()
+        missing_directory = pathlib.Path(directory.root()) / "missing_directory"
+        namespace_pkgs.add_pkgutil_style_namespace_pkg_init(str(missing_directory))
+        self.assertTrue(missing_directory.is_dir())
+
+
 if __name__ == "__main__":
     unittest.main()