addon

backend/app/db/init_db.py

@@ -3,8 +3,8 @@ Linux BenchTools - Database Initialization
 """
 
 import os
-from app.db.base import Base
-from app.db.session import engine
+from app.db.base import Base, BasePeripherals
+from app.db.session import engine, engine_peripherals
 from app.core.config import settings
 
 
@@ -24,8 +24,48 @@ def init_db():
         if db_dir:
             os.makedirs(db_dir, exist_ok=True)
 
-    # Create all tables
+    # Create all tables for main database
     Base.metadata.create_all(bind=engine)
 
-    print(f"✅ Database initialized: {settings.DATABASE_URL}")
+    print(f"✅ Main database initialized: {settings.DATABASE_URL}")
     print(f"✅ Upload directory created: {settings.UPLOAD_DIR}")
+
+    # Initialize peripherals database if module is enabled
+    if settings.PERIPHERALS_MODULE_ENABLED:
+        init_peripherals_db()
+
+
+def init_peripherals_db():
+    """
+    Initialize peripherals database:
+    - Create all tables
+    - Create upload directories
+    - Import peripheral models
+    """
+    # Import models to register them
+    from app.models.peripheral import (
+        Peripheral, PeripheralPhoto, PeripheralDocument,
+        PeripheralLink, PeripheralLoan
+    )
+    from app.models.location import Location
+    from app.models.peripheral_history import PeripheralLocationHistory
+
+    # Create peripherals upload directories
+    os.makedirs(settings.PERIPHERALS_UPLOAD_DIR, exist_ok=True)
+    os.makedirs(os.path.join(settings.PERIPHERALS_UPLOAD_DIR, "photos"), exist_ok=True)
+    os.makedirs(os.path.join(settings.PERIPHERALS_UPLOAD_DIR, "documents"), exist_ok=True)
+    os.makedirs(os.path.join(settings.PERIPHERALS_UPLOAD_DIR, "locations", "images"), exist_ok=True)
+    os.makedirs(os.path.join(settings.PERIPHERALS_UPLOAD_DIR, "locations", "qrcodes"), exist_ok=True)
+
+    # Create database directory if using SQLite
+    if "sqlite" in settings.PERIPHERALS_DB_URL:
+        db_path = settings.PERIPHERALS_DB_URL.replace("sqlite:///", "")
+        db_dir = os.path.dirname(db_path)
+        if db_dir:
+            os.makedirs(db_dir, exist_ok=True)
+
+    # Create all tables for peripherals database
+    BasePeripherals.metadata.create_all(bind=engine_peripherals)
+
+    print(f"✅ Peripherals database initialized: {settings.PERIPHERALS_DB_URL}")
+    print(f"✅ Peripherals upload directories created: {settings.PERIPHERALS_UPLOAD_DIR}")