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