From 2b09371cbf689d816684c2f4f1191e676c955b2d Mon Sep 17 00:00:00 2001 From: Mark Bailey Date: Mon, 4 Nov 2024 15:29:56 -0500 Subject: [PATCH] feat: make dir if it doesn't exist --- src/main.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/main.py b/src/main.py index 29f0e84..8716918 100644 --- a/src/main.py +++ b/src/main.py @@ -43,6 +43,9 @@ def post_detect(file: Image, strength: DetectionStrength): print("\nWorking on: " + file_name) print("\nStrength: " + strength + "\n") + if not os.path.exists("data/"): + os.makedirs("data") + with open("data/" + file_name, "wb") as f: f.write(b64decode(file.file_contents))