from extensions import db
from models.ambiente import Ambiente

def seed_ambiente():
    ambientes = [
        # Ambiente 1
        {
            "service_id": 4,
            "service_role_id":4,
            "name": "Entrada",
            "description": " Hall de Entrada, apresenta 3 cameras."
        },

        # Ambiente 2
        {
            "service_id": 4,
            "service_role_id":4,
            "name": "Estoque",
            "description": "Estoque, apresenta 3 cameras."
        },
    ]
    for ambiente_data in ambientes:
        # Verifica se já existe um papel para o service_id e name fornecidos
        existing_ambiente = db.session.query(Ambiente).filter_by(
            service_id=ambiente_data["service_id"],
            name=ambiente_data["name"]
        ).first()

        # Se não existe, cria o novo papel
        if not existing_ambiente:
            ambiente = Ambiente(
                service_id=ambiente_data["service_id"],
                service_role_id=ambiente_data["service_role_id"],
                name=ambiente_data["name"],
                description=ambiente_data["description"]
            )
            db.session.add(ambiente)
    
    # Commit a transação para salvar os dados no banco
    db.session.commit()