47 lines
2.0 KiB
Python
47 lines
2.0 KiB
Python
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": "Drograsil", "description": "Instância para Farmácia Drogasil"},
|
|
{"service_id": service_1.id, "name": "Fogo de Chão", "description": "Instância para Escola Fogo de Chão"},
|
|
{"service_id": service_1.id, "name": "Sabin", "description": "Instancia para Farmacia Sabin"},
|
|
]
|
|
|
|
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!")
|