from models.service import Service from models.service_instance import ServiceInstance from extensions import db def seed_service_instances(): # Verifica se o serviço com os IDs 1 e 2 existem, se não, cria-os service_1 = Service.query.get(1) service_2 = Service.query.get(2) if not service_1: print("Serviço com ID 1 não encontrado. Criando serviço padrão...") service_1 = Service(name="Serviço 1", description="Descrição do serviço 1") db.session.add(service_1) if not service_2: print("Serviço com ID 2 não encontrado. Criando serviço padrão...") service_2 = Service(name="Serviço 2", description="Descrição do serviço 2") db.session.add(service_2) db.session.commit() # Confirma as inserções dos serviços, se necessário # Verifica se as instâncias de serviço já existem, caso contrário, cria instances_data = [ {"service_id": service_1.id, "name": "Farmácia 1", "description": "Instância para Farmácia 1"}, #{"service_id": service_1.id, "name": "Escola Beta", "description": "Instância para Escola Beta"}, #{"service_id": service_2.id, "name": "Campus Principal", "description": "Controle de Acesso Campus"}, ] for instance_data in instances_data: # Verifica se a instância já existe existing_instance = ServiceInstance.query.filter_by(service_id=instance_data["service_id"], name=instance_data["name"]).first() if not existing_instance: instance = ServiceInstance( service_id=instance_data["service_id"], name=instance_data["name"], description=instance_data["description"] ) db.session.add(instance) print(f"Instância '{instance_data['name']}' inserida com sucesso!") else: print(f"Instância '{instance_data['name']}' já existe, pulando inserção.") # Commit as mudanças no banco db.session.commit() print("Instâncias de serviço inseridas com sucesso!")