Add time schedule seeding functionality to populate days of the week

This commit is contained in:
flavia-vic 2025-04-30 14:46:40 -04:00
parent 1bf175d110
commit 5ff5146889
2 changed files with 49 additions and 17 deletions

View File

@ -16,30 +16,32 @@ from .reports_seed import reports_seed
from .shifts_times_seed import seed_shifts_and_schedules from .shifts_times_seed import seed_shifts_and_schedules
from .holiday_seed import seed_holiday from .holiday_seed import seed_holiday
from .company_seed import seed_company from .company_seed import seed_company
from .time_schedule import seed_time_schedules
from extensions import db from extensions import db
def run_all_seeds(): def run_all_seeds():
try: try:
print("Iniciando o processo de seed...") print("Iniciando o processo de seed...")
seed_services() # seed_services()
seed_holiday() # seed_holiday()
seed_service_instances() # seed_service_instances()
seed_company() # seed_company()
seed_permissions() # seed_permissions()
seed_service_roles() # seed_service_roles()
seed_service_role_permissions() # seed_service_role_permissions()
seed_users() # seed_users()
seed_address_types() # seed_address_types()
seed_address() # seed_address()
#seed_contacts() #seed_contacts()
seed_user_service_roles() # seed_user_service_roles()
seed_licenses() # seed_licenses()
seed_ambiente() # seed_ambiente()
reports_seed() # reports_seed()
seed_cameras() # seed_cameras()
seed_shifts_and_schedules() # seed_shifts_and_schedules()
seed_holiday() # seed_holiday()
seed_time_schedules()
print("Seeds executados com sucesso!") print("Seeds executados com sucesso!")
except Exception as e: except Exception as e:

30
seeds/time_schedule.py Normal file
View File

@ -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.")