Add scheduler to track fall down status + Line notify

This commit is contained in:
sosokker 2024-05-10 22:30:01 +07:00
parent 67bae60e3a
commit e2c1097ba5
2 changed files with 14 additions and 4 deletions

View File

@ -27,4 +27,5 @@ TEMP_VIDEO_FILE = config('TEMP_VIDEO_FILE')
CONFIG_FILE = config('CONFIG_FILE')
YOLO_WEIGHT_FILE = config('YOLO_WEIGHT_FILE')
SPPE_WEIGHT_FILE = config('SPPE_WEIGHT_FILE')
TSSTG_WEIGHT_FILE = config('TSSTG_WEIGHT_FILE')
TSSTG_WEIGHT_FILE = config('TSSTG_WEIGHT_FILE')
LINE_NOTIFY_TOKEN = config('LINE_NOTIFY_TOKEN')

View File

@ -1,5 +1,6 @@
import cv2
import time
import httpx
from cv2 import VideoCapture, imencode
from contextlib import asynccontextmanager
@ -12,7 +13,7 @@ from fastapi.responses import StreamingResponse
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from apscheduler.jobstores.memory import MemoryJobStore
from database import minio_client
from config import TEMP_VIDEO_FILE, VIDEO_BUCKET
from config import TEMP_VIDEO_FILE, VIDEO_BUCKET, LINE_NOTIFY_TOKEN
from scheme import Camera
from utils import save_to_config, read_cameras_from_config
@ -24,8 +25,6 @@ jobstores = {
}
scheduler = AsyncIOScheduler(jobstores=jobstores, timezone='Asia/Bangkok')
action_model = ActionModel()
@asynccontextmanager
async def lifespan(application: FastAPI):
scheduler.start()
@ -36,6 +35,8 @@ router = APIRouter()
cameras: list[Camera] = read_cameras_from_config('config.json')
action_model = ActionModel()
# --- UTILITY FUNCTIONS ---
def generate_camera_id() -> int:
@ -71,6 +72,14 @@ def check_camera_status():
save_to_config(key="cameras", value=cameras)
@scheduler.scheduled_job('interval', seconds=10)
def check_falldown_action():
"""If action Fall Down is detected, notify the user"""
if "Fall Down" in action_model.ACTION_LIST:
# send post to https://notify-api.line.me/api/notify
# Authorization: Bearer use LINE_NOTIFY_TOKEN (ChannelAccessToken)
httpx.post("https://notify-api.line.me/api/notify", headers={"Authorization": f"Bearer {LINE_NOTIFY_TOKEN}"}, data={"message": "Fall Down detected!"})
# --- ROUTER ENDPOINTS ---
@router.post("/add", response_model=dict)