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 .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:

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