From 29fb9daa0c2ddf51cb9b372b5bee7d6cbb367ad5 Mon Sep 17 00:00:00 2001
From: w4ffl35 <25737761+w4ffl35@users.noreply.github.com>
Date: Thu, 9 May 2024 11:41:58 -0600
Subject: [PATCH] Fixes whitelisted_directories

---
 setup.py                           | 2 +-
 src/darklock/restrict_os_access.py | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/setup.py b/setup.py
index 418c4e3..9287174 100644
--- 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
index 6fc6f1d..e8c6d9a 100644
--- 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)