45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
from extensions import db
|
|
from models.holiday import Holiday
|
|
from datetime import datetime
|
|
|
|
def seed_holiday():
|
|
date= "01/05/2025"
|
|
data_iso = datetime.strptime(date, "%d/%m/%Y").date()
|
|
date2= "12/10/2025"
|
|
data_iso2 = datetime.strptime(date2, "%d/%m/%Y").date()
|
|
holidays = [
|
|
# Feriado 1
|
|
{
|
|
"parent_id": 1,
|
|
"service_instance_id":4,
|
|
"name": "Dia do trabalhador",
|
|
"date":data_iso,
|
|
"estado":"Rio de Janeiro",
|
|
"municipio":"Rio de Janeiro",
|
|
|
|
}
|
|
|
|
]
|
|
for holiday_data in holidays:
|
|
# Verifica se já existe um papel para o service_id e name fornecidos
|
|
existing_holiday = db.session.query(Holiday).filter_by(
|
|
date=holiday_data["date"],
|
|
name=holiday_data["name"]
|
|
).first()
|
|
|
|
# Se não existe, cria o novo papel
|
|
if not existing_holiday:
|
|
holiday = Holiday(
|
|
date=holiday_data["date"],
|
|
service_instance_id=holiday_data["service_instance_id"],
|
|
name=holiday_data["name"],
|
|
parent_id=holiday_data["parent_id"],
|
|
estado=holiday_data["estado"],
|
|
municipio=holiday_data["municipio"]
|
|
)
|
|
print("antes do banco")
|
|
db.session.add(holiday)
|
|
print("adiconei no banco")
|
|
|
|
# Commit a transação para salvar os dados no banco
|
|
db.session.commit() |