# seed.py
from sqlalchemy.orm import Session
from database import SessionLocal, init_db
from models import Equipment

# List of equipment data
equipmentData = [
    "aut.",
    "aut.gear/tiptronic",
    "ratgearskifte",
    "alu.",
    "15\" Alufælge",
    "16\" Alufælge",
    "17\" Alufælge",
    "18\" Alufælge",
    "19\" Alufælge",
    "20\" Alufælge",
    "21\" Alufælge",
    "22\" Alufælge",
    "vinterhjul",
    "varme i rat",
    "airc.",
    "fuldaut. klima",
    "2 zone klima",
    "3 zone klima",
    "4 zone klima",
    "køl i handskerum",
    "elektrisk",
    "kabinevarmer",
    "motorkabinevarmer",
    "alarm",
    "el-klapbare sidespejle",
    "el-klapbare sidespejle m. varme",
    "el-spejle m/varme",
    "automatisk parkerings system",
    "360° kamera",
    "bakkamera",
    "parkeringssensor (bag)",
    "parkeringssensor (for)",
    "adaptiv fartpilot",
    "automatisk",
    "start/stop",
    "el betjent bagklap",
    "dæktryksmåler",
    "adaptiv undervogn",
    "elektrisk parkeringsbremse",
    "træthedsregistrering",
    "skiltegenkendelse",
    "CD",
    "CD/radio",
    "radio med CDboks",
    "el komfortsæder",
    "sportssæder",
    "integrerede børnesæder",
    "3 individuelle sæder i bag",
    "lygtevasker",
    "tågelygter",
    "bi-xenon",
    "xenonlys",
    "automatisk lys",
    "fjernlysassistent",
    "kurvelys",
    "LED kørelys",
    "fuld LED forlygter",
    "airbag",
    "db. airbags",
    "4 airbags",
    "6 airbags",
    "7 airbags",
    "8 airbags",
    "9 airbags",
    "10 airbags",
    "ABS",
    "antispin",
    "c.lås",
    "fjernb. c.lås",
    "nøglefri betjening",
    "fartpilot",
    "kørecomputer",
    "infocenter",
    "startspærre",
    "varme i forrude",
    "auto. nedbl. Bakspejl",
    "udv. temp. måler",
    "regnsensor",
    "sædevarme",
    "højdejust. forsæder",
    "højdejust. førersæde",
    "el indst. forsæder",
    "el indst. førersæde m. memory",
    "soltag",
    "el-soltag",
    "glastag",
    "el-ruder",
    "4x el-ruder",
    "el-spejle",
    "DAB radio",
    "DAB+ radio",
    "navigation",
    "multifunktionsrat",
    "håndfrit til mobil",
    "bluetooth",
    "musikstreaming via bluetooth",
    "nightvision",
    "digitalt cockpit",
    "headup display",
    "Android Auto",
    "Apple CarPlay",
    "Internet",
    "trådløs",
    "mobilopladning",
    "SD kortlæser",
    "USB tilslutning",
    "AUX tilslutning",
    "armlæn",
    "isofix",
    "bagagerumsdækken",
    "kopholder",
    "stofindtræk",
    "dellæder",
    "læderindtræk",
    "kunstlæder",
    "splitbagsæde",
    "læderrat",
    "ESP",
    "servo",
    "vognbaneassistent",
    "blindvinkelsassistent",
    "automatisk nødbremsesystem",
    "sænket",
    "tagræling",
    "tonede ruder",
    "mørktonede ruder i bag",
    "1 ejer",
    "ikke ryger",
    "service ok",
    "brugtbilsattest",
    "træk",
    "aftag. træk",
    "svingbart træk (manuel)",
    "svingbart træk (elektrisk)",
    "diesel partikel filter",
    "undervognsbehandlet"
]

def seed_equipment_data(db: Session):
    for equipment in equipmentData:
        db.add(Equipment(name=equipment))
    db.commit()

def main():
    init_db()  # Ensure all tables are created
    db = SessionLocal()
    try:
        seed_equipment_data(db)
        print("Equipment data seeded.")
    finally:
        db.close()

if __name__ == "__main__":
    main()