Should check for __init__.py in module_to_file_path

gauge-sh · Feb 8, 2024 · 8451d14 · 8451d14
1 parent ec9e21f
commit 8451d14
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/modguard/parsing/utils.py b/modguard/parsing/utils.py
@@ -76,6 +76,11 @@ def module_to_file_path(
         member_name = fs_path[last_sep_index + 1 :]
         return file_path, member_name
 
+    init_file_path = fs_path[:last_sep_index] + "/__init__.py"
+    if os.path.exists(init_file_path):
+        member_name = fs_path[last_sep_index + 1 :]
+        return init_file_path, member_name
+
     raise errors.ModguardParseError(
         f"Failed to translate module path {mod_path} into file path"
     )