Update _auditwall.py

Author: KRRT7

codeflash/verification/_auditwall.py

@@ -153,14 +153,12 @@ def make_handler(event: str) -> Callable[[str, tuple], None]:
     # Block groups of events.
     event_prefix = event.split(".", 1)[0]
     if event_prefix in (
-        "os",
         "fcntl",
         "ftplib",
         "glob",
         "imaplib",
         "msvcrt",
         "nntplib",
-        "os",
         "pathlib",
         "poplib",
         "shutil",
@@ -173,6 +171,18 @@ def make_handler(event: str) -> Callable[[str, tuple], None]:
         "webbrowser",
     ):
         return reject
+    if event_prefix == "os" and event not in [
+        "os.putenv",
+        "os.unsetenv",
+        "os.listdir",
+        "os.scandir",
+        "os.chdir",
+        "os.fwalk",
+        "os.getxattr",
+        "os.listxattr",
+        "os.walk",
+    ]:
+        return reject
     # Allow other events.
     return accept