feat(main): initial commit
This commit is contained in:
65
api/routers/schedule.py
Normal file
65
api/routers/schedule.py
Normal file
@@ -0,0 +1,65 @@
|
||||
from ninja import Router, Schema
|
||||
from typing import List, Optional
|
||||
from core.models import ScheduleTemplate, Channel
|
||||
from django.shortcuts import get_object_or_404
|
||||
from datetime import date
|
||||
|
||||
router = Router(tags=["schedule"])
|
||||
|
||||
class ScheduleTemplateSchema(Schema):
|
||||
id: int
|
||||
name: str
|
||||
description: Optional[str] = None
|
||||
timezone_name: str
|
||||
valid_from_date: Optional[date] = None
|
||||
valid_to_date: Optional[date] = None
|
||||
priority: int
|
||||
is_active: bool
|
||||
channel_id: int
|
||||
|
||||
class ScheduleTemplateCreateSchema(Schema):
|
||||
name: str
|
||||
description: Optional[str] = None
|
||||
timezone_name: str
|
||||
valid_from_date: Optional[date] = None
|
||||
valid_to_date: Optional[date] = None
|
||||
priority: int = 0
|
||||
is_active: bool = True
|
||||
channel_id: int
|
||||
|
||||
@router.get("/template/", response=List[ScheduleTemplateSchema])
|
||||
def list_schedule_templates(request):
|
||||
return ScheduleTemplate.objects.all()
|
||||
|
||||
@router.get("/template/{template_id}", response=ScheduleTemplateSchema)
|
||||
def get_schedule_template(request, template_id: int):
|
||||
return get_object_or_404(ScheduleTemplate, id=template_id)
|
||||
|
||||
@router.post("/template/", response={201: ScheduleTemplateSchema})
|
||||
def create_schedule_template(request, payload: ScheduleTemplateCreateSchema):
|
||||
channel = get_object_or_404(Channel, id=payload.channel_id)
|
||||
|
||||
template = ScheduleTemplate.objects.create(
|
||||
channel=channel,
|
||||
name=payload.name,
|
||||
description=payload.description,
|
||||
timezone_name=payload.timezone_name,
|
||||
valid_from_date=payload.valid_from_date,
|
||||
valid_to_date=payload.valid_to_date,
|
||||
priority=payload.priority,
|
||||
is_active=payload.is_active
|
||||
)
|
||||
return 201, template
|
||||
|
||||
class GenerateScheduleSchema(Schema):
|
||||
target_date: date
|
||||
|
||||
@router.post("/generate/{channel_id}")
|
||||
def generate_schedule(request, channel_id: int, payload: GenerateScheduleSchema):
|
||||
channel = get_object_or_404(Channel, id=channel_id)
|
||||
# Inline import to prevent circular dependency issues during initialization
|
||||
from core.services.scheduler import ScheduleGenerator
|
||||
|
||||
generator = ScheduleGenerator(channel=channel)
|
||||
airings_created = generator.generate_for_date(payload.target_date)
|
||||
return {"status": "success", "airings_created": airings_created}
|
||||
Reference in New Issue
Block a user