"""
Linux BenchTools - Database Base
"""

from sqlalchemy.ext.declarative import declarative_base

# Base for main database (benchmarks, devices)
Base = declarative_base()

# Base for peripherals database (separate)
BasePeripherals = declarative_base()

# Import all models here for Alembic/migrations
# Main DB models
from app.models.device import Device  # noqa
from app.models.hardware_snapshot import HardwareSnapshot  # noqa
from app.models.benchmark import Benchmark  # noqa
from app.models.disk_smart import DiskSMART  # noqa
from app.models.manufacturer_link import ManufacturerLink  # noqa
from app.models.document import Document  # noqa

# Peripherals DB models (imported when module enabled)
# Will be imported in init_db.py