Add time schedule seeding functionality to populate days of the week
This commit is contained in:
parent
1bf175d110
commit
5ff5146889
@ -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
30
seeds/time_schedule.py
Normal 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.")
|
||||
Loading…
Reference in New Issue
Block a user