Fix crash if driveoff home folder is not created

UoA-eResearch · Nov 18, 2024 · c97b0dd · c97b0dd
1 parent c5e4d5d
commit c97b0dd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/api/main.py b/src/api/main.py
@@ -17,6 +17,8 @@
 from models.role import prepopulate_roles
 from models.services import ResearchDriveService, Services
 
+# Ensure driveoff directory is created
+(Path.home() / ".driveoff").mkdir(exist_ok=True)
 DB_FILE_NAME = Path.home() / ".driveoff" / "database.db"
 DB_URL = f"sqlite:///{DB_FILE_NAME}"