116 lines
3.2 KiB
Python
116 lines
3.2 KiB
Python
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="Задача не найдена")
|