from banco import SessionLocal
from models import FPSO

FPSO_PADRAO = [
    "CDA", "CDP", "CDM", "ADG",
    "ESS", "SEP", "CDI", "ATD", "CDS"
]

def main():
    db = SessionLocal()
    try:
        existentes = {f.nome for f in db.query(FPSO).all()}

        for nome in FPSO_PADRAO:
            if nome not in existentes:
                db.add(FPSO(nome=nome))

        db.commit()
        print("✅ FPSOs padrão inseridos com sucesso!")

    finally:
        db.close()


if __name__ == "__main__":
    main()