Fixes whitelisted_directories

Capsize-Games · May 9, 2024 · 29fb9da · 29fb9da
1 parent 5f29cbe
commit 29fb9da
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="darklock",
-    version="0.1.4",
+    version="0.1.5",
     author="Capsize LLC",
     description="",
     long_description=open("README.md", "r", encoding="utf-8").read(),

diff --git a/src/darklock/restrict_os_access.py b/src/darklock/restrict_os_access.py
@@ -27,10 +27,10 @@ def __init__(self):
         self.logger.addHandler(logging.StreamHandler())
 
     def is_directory_whitelisted(self, directory: str) -> bool:
-        print("IS DIRECTORY WHITELISTED")
-        print(directory)
-        print(self.whitelisted_directories)
-        return directory in self.whitelisted_directories
+        for d in self.whitelisted_directories:
+            if str(d) in str(directory):
+                return True
+        return False
 
     def restrict_os_write(self, *args, **kwargs):
         return self.original_os_write(*args, **kwargs)