Refactor: Clean main.py

app/api/routes/schedules.py

@@ -0,0 +1,115 @@
+from datetime import datetime, timedelta
+from typing import Optional
+from fastapi import APIRouter, Depends, HTTPException
+from apscheduler.triggers.cron import CronTrigger
+from apscheduler.triggers.date import DateTrigger
+
+from app.core.state import state_manager
+from app.core.scheduler import scheduler, execute_lamp_command
+from app.drivers.wiz import WizDriver
+from app.api.deps import verify_token
+
+router = APIRouter(dependencies=[Depends(verify_token)])
+wiz = WizDriver()
+
+
+@router.post("/at")
+async def add_task_at_time(
+    target_id: str,
+    run_at: datetime,
+    is_group: bool = False,
+    state: Optional[bool] = None,
+    brightness: Optional[int] = None,
+    scene: Optional[str] = None,
+):
+    params = {}
+    if state is not None:
+        params["state"] = state
+    if brightness is not None:
+        params["dimming"] = brightness
+    if scene and scene in wiz.SCENES:
+        params["sceneId"] = wiz.SCENES[scene]
+
+    ips = state_manager.get_group_ips(target_id) if is_group else []
+    if not is_group:
+        dev = state_manager.devices.get(target_id)
+        if dev:
+            ips = [dev.ip]
+
+    if not ips:
+        raise HTTPException(status_code=404, detail="Цель не найдена")
+
+    job_ids = []
+    for ip in ips:
+        job = scheduler.add_job(
+            execute_lamp_command,
+            DateTrigger(run_date=run_at, timezone=scheduler.timezone),
+            args=[ip, params],
+            name=f"{'Group' if is_group else 'Device'}: {target_id} | {params}",
+        )
+        job_ids.append(job.id)
+
+    return {"status": "scheduled", "jobs": job_ids, "run_at": run_at}
+
+
+@router.post("/cron")
+async def add_cron_task(
+    target_id: str,
+    hour: str,
+    minute: str,
+    day_of_week: str = "*",
+    is_group: bool = True,
+    state: bool = True,
+):
+    ips = state_manager.get_group_ips(target_id) if is_group else []
+    if not is_group:
+        dev = state_manager.devices.get(target_id)
+        if dev:
+            ips = [dev.ip]
+
+    if not ips:
+        raise HTTPException(status_code=404, detail="Цель не найдена")
+
+    trigger = CronTrigger(
+        hour=hour, minute=minute, day_of_week=day_of_week, timezone=scheduler.timezone
+    )
+
+    job_ids = []
+    for ip in ips:
+        job = scheduler.add_job(
+            execute_lamp_command,
+            trigger,
+            args=[ip, {"state": state}],
+            name=f"CRON: {target_id} | {hour}:{minute}",
+        )
+        job_ids.append(job.id)
+
+    return {"status": "cron_scheduled", "jobs": job_ids}
+
+
+@router.get("/tasks")
+async def get_all_tasks():
+    jobs = []
+
+    for job in scheduler.get_jobs():
+        jobs.append(
+            {
+                "id": job.id,
+                "name": job.name,
+                "next_run": (
+                    job.next_run_time.isoformat() if job.next_run_time else None
+                ),
+                "params": str(job.args[1]) if len(job.args) > 1 else None,
+            }
+        )
+
+    return {"tasks": jobs}
+
+
+@router.delete("/{job_id}")
+async def cancel_task(job_id: str):
+    try:
+        scheduler.remove_job(job_id)
+        return {"status": "deleted"}
+    except:
+        raise HTTPException(status_code=404, detail="Задача не найдена")