front_ponto_eletronico/seeds/time_schedule.py

31 lines
963 B
Python

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