72 lines
2.5 KiB
Python
Executable File
72 lines
2.5 KiB
Python
Executable File
"""
|
|
Linux BenchTools - Database Initialization
|
|
"""
|
|
|
|
import os
|
|
from app.db.base import Base, BasePeripherals
|
|
from app.db.session import engine, engine_peripherals
|
|
from app.core.config import settings
|
|
|
|
|
|
def init_db():
|
|
"""
|
|
Initialize database:
|
|
- Create all tables
|
|
- Create upload directory if it doesn't exist
|
|
"""
|
|
# Create upload directory
|
|
os.makedirs(settings.UPLOAD_DIR, exist_ok=True)
|
|
|
|
# Create database directory if using SQLite
|
|
if "sqlite" in settings.DATABASE_URL:
|
|
db_path = settings.DATABASE_URL.replace("sqlite:///", "")
|
|
db_dir = os.path.dirname(db_path)
|
|
if db_dir:
|
|
os.makedirs(db_dir, exist_ok=True)
|
|
|
|
# Create all tables for main database
|
|
Base.metadata.create_all(bind=engine)
|
|
|
|
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}")
|