#4 Rootpath: Fixed Infinite loop for windows

grimen · Mar 31, 2021 · ed6df33 · ed6df33
1 parent f5446bf
commit ed6df33
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/rootpath/detect.py b/rootpath/detect.py
@@ -83,6 +83,13 @@ def find_root_path(current_path, pattern = None):
             if found_system_root:
                 return None
 
+            system_root = sys.executable
+            while os.path.split(system_root)[1]:
+                system_root = os.path.split(system_root)[0]
+
+            if current_path == system_root:
+                return None
+
             current_path = path.abspath(path.join(current_path, '..'))
 
     return find_root_path(current_path, pattern)
diff --git a/rootpath/tests/test_detect.py b/rootpath/tests/test_detect.py
@@ -173,6 +173,10 @@ def test_rootpath_detect_entry_nested_pattern(self):
 
         self.assertEqual(root_path, None)
 
+    def test_system_root(self):
+        root_path = rootpath.detect(r"\Users\Public\Downloads")
+        self.assertEqual(root_path, None)
+
 
 # =========================================
 #       MAIN