diff --git a/seeds/run_seed.py b/seeds/run_seed.py index a1fdfd8..fe26a91 100644 --- a/seeds/run_seed.py +++ b/seeds/run_seed.py @@ -16,30 +16,32 @@ from .reports_seed import reports_seed from .shifts_times_seed import seed_shifts_and_schedules from .holiday_seed import seed_holiday from .company_seed import seed_company +from .time_schedule import seed_time_schedules from extensions import db def run_all_seeds(): try: print("Iniciando o processo de seed...") - seed_services() - seed_holiday() - seed_service_instances() - seed_company() - seed_permissions() - seed_service_roles() - seed_service_role_permissions() - seed_users() - seed_address_types() - seed_address() + # seed_services() + # seed_holiday() + # seed_service_instances() + # seed_company() + # seed_permissions() + # seed_service_roles() + # seed_service_role_permissions() + # seed_users() + # seed_address_types() + # seed_address() #seed_contacts() - seed_user_service_roles() - seed_licenses() - seed_ambiente() - reports_seed() - seed_cameras() - seed_shifts_and_schedules() - seed_holiday() + # seed_user_service_roles() + # seed_licenses() + # seed_ambiente() + # reports_seed() + # seed_cameras() + # seed_shifts_and_schedules() + # seed_holiday() + seed_time_schedules() print("Seeds executados com sucesso!") except Exception as e: diff --git a/seeds/time_schedule.py b/seeds/time_schedule.py new file mode 100644 index 0000000..4d621aa --- /dev/null +++ b/seeds/time_schedule.py @@ -0,0 +1,30 @@ +from extensions import db +from models.time_schedule import TimeSchedule + +def seed_time_schedules(): + # Lista de dias da semana + days = [ + {"id": 1, "name": "Domingo"}, + {"id": 2, "name": "Segunda-feira"}, + {"id": 3, "name": "Terça-feira"}, + {"id": 4, "name": "Quarta-feira"}, + {"id": 5, "name": "Quinta-feira"}, + {"id": 6, "name": "Sexta-feira"}, + {"id": 7, "name": "Sábado"}, + ] + + # Verifica se já existe algum dia na tabela + existing_days = TimeSchedule.query.all() + existing_names = [day.name for day in existing_days] + + for day in days: + if day["name"] in existing_names: + print(f"Dia '{day['name']}' já existe. Pulando...") + continue + + # Cria e adiciona o dia + new_day = TimeSchedule(id=day["id"], name=day["name"]) + db.session.add(new_day) + + db.session.commit() + print("Dias da semana criados com sucesso, se não existirem.")