HartOMat/backend/app/utils/seed_templates.py

"""Seed database with 7 Schaeffler product category templates."""
import asyncio
import uuid
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine, async_sessionmaker
from sqlalchemy import select

STANDARD_FIELDS = {
    "0": {"label": "Ebene1", "required": True},
    "1": {"label": "Ebene2", "required": True},
    "2": {"label": "Baureihe", "required": True},
    "3": {"label": "PIM-ID (Klasse)", "required": False},
    "4": {"label": "Produkt (Baureihe)", "required": False},
    "5": {"label": "[Separator]", "required": False, "skip": True},
    "6": {"label": "Gewähltes Produkt", "required": True},
    "7": {"label": "Name CAD-Modell", "required": True},
    "8": {"label": "Gewünschte Bildnummer", "required": False},
    "9": {"label": "Lagertyp", "required": False},
    "10": {"label": "Medias-Rendering", "required": False},
}

TEMPLATES = [
    {
        "name": "Tapered Roller Bearings (TRB)",
        "category_key": "TRB",
        "description": "Kegelrollenlager – Tapered roller bearings",
        "standard_fields": STANDARD_FIELDS,
        "component_schema": {
            "pairs": [
                {"component_type": "Innenring / Inner ring", "required": False},
                {"component_type": "Innenring / Inner ring 2", "required": False},
                {"component_type": "Innenring / Inner ring 3", "required": False},
                {"component_type": "Außenring / Outer ring", "required": False},
                {"component_type": "Außenring / Outer ring 2", "required": False},
                {"component_type": "Außenring / Outer ring 3", "required": False},
                {"component_type": "Außenring / Outer ring 4", "required": False},
                {"component_type": "Käfig / Cage", "required": False},
                {"component_type": "Wälzkörper / Rolling Element", "required": False},
                {"component_type": "Dichtungskern/Dichtungsträger", "required": False},
                {"component_type": "Dichtung Außen / Dichtlippe", "required": False},
            ]
        },
    },
    {
        "name": "Kugellager (Ball Bearings)",
        "category_key": "Kugellager",
        "description": "Kugellager – Ball bearings",
        "standard_fields": STANDARD_FIELDS,
        "component_schema": {
            "pairs": [
                {"component_type": "Innenring / Inner ring", "required": False},
                {"component_type": "Außenring / Outer ring", "required": False},
                {"component_type": "Wälzkörper / Rolling Element", "required": True},
                {"component_type": "Käfig / Cage", "required": False},
                {"component_type": "Dichtungskern/Dichtungsträger", "required": False},
                {"component_type": "Axial - WS", "required": False},
                {"component_type": "Axial - GS", "required": False},
            ]
        },
    },
    {
        "name": "Gleitlager (Plain Bearings)",
        "category_key": "Gleitlager",
        "description": "Gleitlager – Plain / sliding bearings",
        "standard_fields": STANDARD_FIELDS,
        "component_schema": {
            "pairs": [
                {"component_type": "Innenring / Inner ring", "required": False},
                {"component_type": "Außenring / Outer ring", "required": False},
                {"component_type": "Gehause / Housing", "required": False},
                {"component_type": "Sliding Layer", "required": False},
                {"component_type": "Dichtungsträger / Sealing carrier", "required": False},
                {"component_type": "Dichtlippe / Sealing lip", "required": False},
            ]
        },
    },
    {
        "name": "Spherical / Toroidal Roller Bearings (SRB/TORB)",
        "category_key": "SRB_TORB",
        "description": "Pendelrollenlager / Toroidalrollenlager – SRB and TORB bearings",
        "standard_fields": STANDARD_FIELDS,
        "component_schema": {
            "pairs": [
                {"component_type": "Innenring / Inner ring", "required": False},
                {"component_type": "Außenring / Outer ring", "required": False},
                {"component_type": "Käfig / Cage", "required": False},
                {"component_type": "Wälzkörper / Rolling element", "required": False},
                {"component_type": "Bordscheibe IR / Loose Lip IR", "required": False},
                {"component_type": "Dichtungsträger / Sealing carrier", "required": False},
            ]
        },
    },
    {
        "name": "Cylindrical Roller Bearings (CRB)",
        "category_key": "CRB",
        "description": "Zylinderrollenlager – Cylindrical roller bearings",
        "standard_fields": STANDARD_FIELDS,
        "component_schema": {
            "pairs": [
                {"component_type": "Innenring", "required": False},
                {"component_type": "Außenring", "required": False},
                {"component_type": "Rollen", "required": False},
                {"component_type": "Käfig", "required": False},
                {"component_type": "Dichtung", "required": False},
                {"component_type": "Halteringe", "required": False},
                {"component_type": "Bordscheibe", "required": False},
            ]
        },
    },
    {
        "name": "Linear Guide Rails",
        "category_key": "Linear_schiene",
        "description": "Linearsysteme – Linear guide rail systems",
        "standard_fields": STANDARD_FIELDS,
        "component_schema": {
            "pairs": [
                {"component_type": "Rail", "required": True},
            ]
        },
    },
    {
        "name": "End Plates (Anschlagplatten)",
        "category_key": "Anschlagplatten",
        "description": "Anschlagplatten – End plates for guide rails",
        "standard_fields": STANDARD_FIELDS,
        "component_schema": {
            "pairs": [
                {"component_type": "Platte / Plate", "required": True},
                {"component_type": "Schraube / Screw", "required": False},
                {"component_type": "Nut BZ", "required": False},
            ]
        },
    },
]


async def seed(db_url: str, admin_email: str = "admin@schaeffler.com", admin_password: str = "Admin1234!"):
    from app.models.template import Template
    from app.models.user import User, UserRole
    from app.utils.auth import hash_password

    engine = create_async_engine(db_url, echo=False)
    session_factory = async_sessionmaker(engine, expire_on_commit=False)

    async with session_factory() as session:
        # Seed templates
        for tpl_data in TEMPLATES:
            result = await session.execute(
                select(Template).where(Template.category_key == tpl_data["category_key"])
            )
            existing = result.scalar_one_or_none()
            if not existing:
                tpl = Template(**tpl_data)
                session.add(tpl)
                print(f"  + Template: {tpl_data['category_key']}")
            else:
                print(f"  ~ Template already exists: {tpl_data['category_key']}")

        # Seed admin user
        result = await session.execute(select(User).where(User.email == admin_email))
        if not result.scalar_one_or_none():
            admin = User(
                email=admin_email,
                password_hash=hash_password(admin_password),
                role=UserRole.global_admin,
                full_name="Schaeffler Admin",
            )
            session.add(admin)
            print(f"  + Admin user: {admin_email}")
        else:
            print(f"  ~ Admin user already exists: {admin_email}")

        await session.commit()
    await engine.dispose()
    print("Seed complete.")


if __name__ == "__main__":
    import sys
    import os

    sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
    from app.config import settings

    asyncio.run(seed(settings.database_url))