68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import date, time
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from .config import settings
|
|
from .models import Schedule, Team, User
|
|
from .security import hash_password
|
|
|
|
|
|
def ensure_initial_data(db: Session) -> None:
|
|
team_count = db.scalar(select(Team.id).limit(1))
|
|
if team_count is None:
|
|
db.add(
|
|
Team(
|
|
name=settings.initial_admin_name,
|
|
email=settings.initial_admin_email or None,
|
|
phone=None,
|
|
user=User(
|
|
name=settings.initial_admin_name,
|
|
username=settings.initial_admin_username,
|
|
password_hash=hash_password(settings.initial_admin_password),
|
|
is_admin=True,
|
|
is_active=True,
|
|
),
|
|
is_active=True,
|
|
)
|
|
)
|
|
db.commit()
|
|
|
|
defaults = [
|
|
{
|
|
"name": "Noite",
|
|
"slug": "night",
|
|
"schedule_type": "fixed",
|
|
"weekdays": "0,1,2,3,4,5,6",
|
|
"start_time": time(20, 30),
|
|
"end_time": time(0, 0),
|
|
"rotation_anchor_date": None,
|
|
},
|
|
{
|
|
"name": "Almoço - Sábados",
|
|
"slug": "saturday-lunch",
|
|
"schedule_type": "rotation",
|
|
"weekdays": "5",
|
|
"start_time": time(13, 0),
|
|
"end_time": time(16, 0),
|
|
"rotation_anchor_date": date(2024, 1, 6),
|
|
},
|
|
{
|
|
"name": "Almoço - Domingo",
|
|
"slug": "sunday-lunch",
|
|
"schedule_type": "rotation",
|
|
"weekdays": "6",
|
|
"start_time": time(13, 0),
|
|
"end_time": time(16, 0),
|
|
"rotation_anchor_date": date(2024, 1, 7),
|
|
},
|
|
]
|
|
|
|
for item in defaults:
|
|
exists = db.scalar(select(Schedule.id).where(Schedule.slug == item["slug"]))
|
|
if exists is None:
|
|
db.add(Schedule(**item, requires_manager_approval=True, is_active=True))
|
|
db.commit()
|