cerbervs/doctr-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: make dir if it doesn't exist

Browse Source
This commit is contained in:
Mark Bailey 2024-11-04 15:29:56 -05:00
parent 0af50c21ae
commit 2b09371cbf
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/main.py
View File

@ -43,6 +43,9 @@ def post_detect(file: Image, strength: DetectionStrength):
print("\nWorking on: " + file_name) print("\nWorking on: " + file_name)
print("\nStrength: " + strength + "\n") print("\nStrength: " + strength + "\n")
if not os.path.exists("data/"):
os.makedirs("data")
with open("data/" + file_name, "wb") as f: with open("data/" + file_name, "wb") as f:
f.write(b64decode(file.file_contents)) f.write(b64decode(file.file_contents))