fix: allow to specify the root folder

Author: aignas

python/private/pypi/namespace_pkgs.bzl

@@ -7,9 +7,19 @@ EXTS = [
     ".pyc",
 ]
 
-def _get_files(files, ignored_dirnames = []):
+def _add_all(dirname, dirs):
+    dir_path = "."
+    for dir_name in dirname.split("/"):
+        dir_path = "{}/{}".format(dir_path, dir_name)
+        dirs[dir_path[2:]] = None
+
+def _get_files(files, ignored_dirnames = [], root = None):
     dirs = {}
     ignored = {i: None for i in ignored_dirnames}
+
+    if root:
+        _add_all(root, ignored)
+
     for file in files:
         dirname, _, filename = file.rpartition("/")
 
@@ -30,10 +40,7 @@ def _get_files(files, ignored_dirnames = []):
         if dirname in dirs or not dirname:
             continue
 
-        dir_path = "."
-        for dir_name in dirname.split("/"):
-            dir_path = "{}/{}".format(dir_path, dir_name)
-            dirs[dir_path[2:]] = None
+        _add_all(dirname, dirs)
 
     return sorted([d for d in dirs if d not in ignored])
 

tests/pypi/namespace_pkgs/namespace_pkgs_tests.bzl

@@ -24,17 +24,17 @@ _tests.append(test_in_current_dir)
 
 def test_find_correct_namespace_packages(env):
     files = [
-        "foo/bar/biz.py",
-        "foo/bee/boo.py",
-        "foo/buu/__init__.py",
-        "foo/buu/bii.py",
+        "nested/root/foo/bar/biz.py",
+        "nested/root/foo/bee/boo.py",
+        "nested/root/foo/buu/__init__.py",
+        "nested/root/foo/buu/bii.py",
     ]
 
-    got = namespace_pkgs.get_files(files)
+    got = namespace_pkgs.get_files(files, root = "nested/root")
     expected = [
-        "foo",
-        "foo/bar",
-        "foo/bee",
+        "nested/root/foo",
+        "nested/root/foo/bar",
+        "nested/root/foo/bee",
     ]
     env.expect.that_collection(got).contains_exactly(expected)