fix:修复因数据库缺少 working_hours 列导致 ROI 配置失败的问题。
- 手动执行 SQL:ALTER TABLE rois ADD COLUMN working_hours TEXT - 确保现有 SQLite 数据库(security_monitor.db)结构与模型定义一致 - 避免因字段缺失引发 API 或算法读取异常
This commit is contained in:
26
api/roi.py
26
api/roi.py
@@ -29,9 +29,10 @@ class CreateROIRequest(BaseModel):
|
|||||||
rule_type: str
|
rule_type: str
|
||||||
direction: Optional[str] = None
|
direction: Optional[str] = None
|
||||||
stay_time: Optional[int] = None
|
stay_time: Optional[int] = None
|
||||||
threshold_sec: int = 360
|
threshold_sec: int = 300
|
||||||
confirm_sec: int = 30
|
confirm_sec: int = 10
|
||||||
return_sec: int = 5
|
return_sec: int = 30
|
||||||
|
working_hours: Optional[List[dict]] = None
|
||||||
|
|
||||||
|
|
||||||
class UpdateROIRequest(BaseModel):
|
class UpdateROIRequest(BaseModel):
|
||||||
@@ -45,6 +46,7 @@ class UpdateROIRequest(BaseModel):
|
|||||||
threshold_sec: Optional[int] = None
|
threshold_sec: Optional[int] = None
|
||||||
confirm_sec: Optional[int] = None
|
confirm_sec: Optional[int] = None
|
||||||
return_sec: Optional[int] = None
|
return_sec: Optional[int] = None
|
||||||
|
working_hours: Optional[List[dict]] = None
|
||||||
|
|
||||||
|
|
||||||
def _invalidate_roi_cache(camera_id: int):
|
def _invalidate_roi_cache(camera_id: int):
|
||||||
@@ -70,6 +72,7 @@ def list_rois(camera_id: int, db: Session = Depends(get_db)):
|
|||||||
"threshold_sec": roi.threshold_sec,
|
"threshold_sec": roi.threshold_sec,
|
||||||
"confirm_sec": roi.confirm_sec,
|
"confirm_sec": roi.confirm_sec,
|
||||||
"return_sec": roi.return_sec,
|
"return_sec": roi.return_sec,
|
||||||
|
"working_hours": json.loads(roi.working_hours) if roi.working_hours else None,
|
||||||
}
|
}
|
||||||
for roi in roi_configs
|
for roi in roi_configs
|
||||||
]
|
]
|
||||||
@@ -93,6 +96,7 @@ def get_roi(camera_id: int, roi_id: int, db: Session = Depends(get_db)):
|
|||||||
"threshold_sec": roi.threshold_sec,
|
"threshold_sec": roi.threshold_sec,
|
||||||
"confirm_sec": roi.confirm_sec,
|
"confirm_sec": roi.confirm_sec,
|
||||||
"return_sec": roi.return_sec,
|
"return_sec": roi.return_sec,
|
||||||
|
"working_hours": json.loads(roi.working_hours) if roi.working_hours else None,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -102,6 +106,10 @@ def add_roi(
|
|||||||
request: CreateROIRequest,
|
request: CreateROIRequest,
|
||||||
db: Session = Depends(get_db),
|
db: Session = Depends(get_db),
|
||||||
):
|
):
|
||||||
|
import json
|
||||||
|
|
||||||
|
working_hours_json = json.dumps(request.working_hours) if request.working_hours else None
|
||||||
|
|
||||||
roi = create_roi(
|
roi = create_roi(
|
||||||
db,
|
db,
|
||||||
camera_id=camera_id,
|
camera_id=camera_id,
|
||||||
@@ -115,6 +123,7 @@ def add_roi(
|
|||||||
threshold_sec=request.threshold_sec,
|
threshold_sec=request.threshold_sec,
|
||||||
confirm_sec=request.confirm_sec,
|
confirm_sec=request.confirm_sec,
|
||||||
return_sec=request.return_sec,
|
return_sec=request.return_sec,
|
||||||
|
working_hours=working_hours_json,
|
||||||
)
|
)
|
||||||
|
|
||||||
_invalidate_roi_cache(camera_id)
|
_invalidate_roi_cache(camera_id)
|
||||||
@@ -126,7 +135,13 @@ def add_roi(
|
|||||||
"type": roi.roi_type,
|
"type": roi.roi_type,
|
||||||
"points": request.points,
|
"points": request.points,
|
||||||
"rule": roi.rule_type,
|
"rule": roi.rule_type,
|
||||||
|
"direction": roi.direction,
|
||||||
|
"stay_time": roi.stay_time,
|
||||||
"enabled": roi.enabled,
|
"enabled": roi.enabled,
|
||||||
|
"threshold_sec": roi.threshold_sec,
|
||||||
|
"confirm_sec": roi.confirm_sec,
|
||||||
|
"return_sec": roi.return_sec,
|
||||||
|
"working_hours": request.working_hours,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -137,6 +152,9 @@ def modify_roi(
|
|||||||
request: UpdateROIRequest,
|
request: UpdateROIRequest,
|
||||||
db: Session = Depends(get_db),
|
db: Session = Depends(get_db),
|
||||||
):
|
):
|
||||||
|
import json
|
||||||
|
working_hours_json = json.dumps(request.working_hours) if request.working_hours else None
|
||||||
|
|
||||||
roi = update_roi(
|
roi = update_roi(
|
||||||
db,
|
db,
|
||||||
roi_id=roi_id,
|
roi_id=roi_id,
|
||||||
@@ -149,6 +167,7 @@ def modify_roi(
|
|||||||
threshold_sec=request.threshold_sec,
|
threshold_sec=request.threshold_sec,
|
||||||
confirm_sec=request.confirm_sec,
|
confirm_sec=request.confirm_sec,
|
||||||
return_sec=request.return_sec,
|
return_sec=request.return_sec,
|
||||||
|
working_hours=working_hours_json,
|
||||||
)
|
)
|
||||||
if not roi:
|
if not roi:
|
||||||
raise HTTPException(status_code=404, detail="ROI不存在")
|
raise HTTPException(status_code=404, detail="ROI不存在")
|
||||||
@@ -163,6 +182,7 @@ def modify_roi(
|
|||||||
"points": json.loads(roi.points),
|
"points": json.loads(roi.points),
|
||||||
"rule": roi.rule_type,
|
"rule": roi.rule_type,
|
||||||
"enabled": roi.enabled,
|
"enabled": roi.enabled,
|
||||||
|
"working_hours": json.loads(roi.working_hours) if roi.working_hours else None,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -139,6 +139,7 @@ def create_roi(
|
|||||||
threshold_sec: int = 300,
|
threshold_sec: int = 300,
|
||||||
confirm_sec: int = 10,
|
confirm_sec: int = 10,
|
||||||
return_sec: int = 30,
|
return_sec: int = 30,
|
||||||
|
working_hours: Optional[str] = None,
|
||||||
) -> ROI:
|
) -> ROI:
|
||||||
import json
|
import json
|
||||||
|
|
||||||
@@ -154,6 +155,7 @@ def create_roi(
|
|||||||
threshold_sec=threshold_sec,
|
threshold_sec=threshold_sec,
|
||||||
confirm_sec=confirm_sec,
|
confirm_sec=confirm_sec,
|
||||||
return_sec=return_sec,
|
return_sec=return_sec,
|
||||||
|
working_hours=working_hours,
|
||||||
)
|
)
|
||||||
db.add(roi)
|
db.add(roi)
|
||||||
db.commit()
|
db.commit()
|
||||||
@@ -173,6 +175,7 @@ def update_roi(
|
|||||||
threshold_sec: Optional[int] = None,
|
threshold_sec: Optional[int] = None,
|
||||||
confirm_sec: Optional[int] = None,
|
confirm_sec: Optional[int] = None,
|
||||||
return_sec: Optional[int] = None,
|
return_sec: Optional[int] = None,
|
||||||
|
working_hours: Optional[str] = None,
|
||||||
) -> Optional[ROI]:
|
) -> Optional[ROI]:
|
||||||
import json
|
import json
|
||||||
|
|
||||||
@@ -198,6 +201,8 @@ def update_roi(
|
|||||||
roi.confirm_sec = confirm_sec
|
roi.confirm_sec = confirm_sec
|
||||||
if return_sec is not None:
|
if return_sec is not None:
|
||||||
roi.return_sec = return_sec
|
roi.return_sec = return_sec
|
||||||
|
if working_hours is not None:
|
||||||
|
roi.working_hours = working_hours
|
||||||
|
|
||||||
db.commit()
|
db.commit()
|
||||||
db.refresh(roi)
|
db.refresh(roi)
|
||||||
@@ -232,6 +237,7 @@ def get_roi_points(db: Session, camera_id: int) -> List[dict]:
|
|||||||
"threshold_sec": roi.threshold_sec,
|
"threshold_sec": roi.threshold_sec,
|
||||||
"confirm_sec": roi.confirm_sec,
|
"confirm_sec": roi.confirm_sec,
|
||||||
"return_sec": roi.return_sec,
|
"return_sec": roi.return_sec,
|
||||||
|
"working_hours": json.loads(roi.working_hours) if roi.working_hours else None,
|
||||||
}
|
}
|
||||||
for roi in rois
|
for roi in rois
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -93,6 +93,7 @@ class ROI(Base):
|
|||||||
threshold_sec: Mapped[int] = mapped_column(Integer, default=300)
|
threshold_sec: Mapped[int] = mapped_column(Integer, default=300)
|
||||||
confirm_sec: Mapped[int] = mapped_column(Integer, default=10)
|
confirm_sec: Mapped[int] = mapped_column(Integer, default=10)
|
||||||
return_sec: Mapped[int] = mapped_column(Integer, default=30)
|
return_sec: Mapped[int] = mapped_column(Integer, default=30)
|
||||||
|
working_hours: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||||
pending_sync: Mapped[bool] = mapped_column(Boolean, default=False)
|
pending_sync: Mapped[bool] = mapped_column(Boolean, default=False)
|
||||||
sync_version: Mapped[int] = mapped_column(Integer, default=0)
|
sync_version: Mapped[int] = mapped_column(Integer, default=0)
|
||||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
||||||
|
|||||||
@@ -38,14 +38,48 @@ const CameraManagement: React.FC = () => {
|
|||||||
const fetchCameras = async () => {
|
const fetchCameras = async () => {
|
||||||
setLoading(true);
|
setLoading(true);
|
||||||
try {
|
try {
|
||||||
const [camerasRes, statusRes] = await Promise.all([
|
const [camerasRes, statusRes, pipelineRes] = await Promise.all([
|
||||||
axios.get('/api/cameras?enabled_only=false'),
|
axios.get('/api/cameras?enabled_only=false'),
|
||||||
|
axios.get('/api/camera/status/all'),
|
||||||
axios.get('/api/pipeline/status')
|
axios.get('/api/pipeline/status')
|
||||||
]);
|
]);
|
||||||
setCameras(camerasRes.data);
|
setCameras(camerasRes.data);
|
||||||
setCameraStatus(statusRes.data.cameras || {});
|
|
||||||
|
const statusMap: Record<number, CameraStatus> = {};
|
||||||
|
|
||||||
|
for (const cam of camerasRes.data) {
|
||||||
|
const camId = cam.id;
|
||||||
|
|
||||||
|
let status: CameraStatus = {
|
||||||
|
is_running: false,
|
||||||
|
fps: 0,
|
||||||
|
error_message: null,
|
||||||
|
last_check_time: null,
|
||||||
|
};
|
||||||
|
|
||||||
|
const pipelineStatus = pipelineRes.data.cameras?.[String(camId)];
|
||||||
|
if (pipelineStatus) {
|
||||||
|
status.is_running = pipelineStatus.is_running || false;
|
||||||
|
status.fps = pipelineStatus.fps || 0;
|
||||||
|
status.last_check_time = pipelineStatus.last_check_time;
|
||||||
|
}
|
||||||
|
|
||||||
|
const dbStatus = statusRes.data.find((s: any) => s.camera_id === camId);
|
||||||
|
if (dbStatus) {
|
||||||
|
if (!status.is_running) {
|
||||||
|
status.is_running = dbStatus.is_running || false;
|
||||||
|
}
|
||||||
|
status.fps = dbStatus.fps || status.fps;
|
||||||
|
status.error_message = dbStatus.error_message;
|
||||||
|
status.last_check_time = status.last_check_time || dbStatus.last_check_time;
|
||||||
|
}
|
||||||
|
|
||||||
|
statusMap[camId] = status;
|
||||||
|
}
|
||||||
|
|
||||||
|
setCameraStatus(statusMap);
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
message.error('获取摄像头列表失败');
|
console.error('获取摄像头状态失败', err);
|
||||||
} finally {
|
} finally {
|
||||||
setLoading(false);
|
setLoading(false);
|
||||||
}
|
}
|
||||||
@@ -59,8 +93,8 @@ const CameraManagement: React.FC = () => {
|
|||||||
|
|
||||||
const extractIP = (url: string): string => {
|
const extractIP = (url: string): string => {
|
||||||
try {
|
try {
|
||||||
const match = url.match(/:\/\/([^:]+):?(\d+)?\//);
|
const ipMatch = url.match(/(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})/);
|
||||||
return match ? match[1] : '未知';
|
return ipMatch ? ipMatch[1] : '未知';
|
||||||
} catch {
|
} catch {
|
||||||
return '未知';
|
return '未知';
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,7 +1,14 @@
|
|||||||
import React, { useEffect, useState, useRef } from 'react';
|
import React, { useEffect, useState, useRef } from 'react';
|
||||||
import { Card, Button, Space, Select, message, Drawer, Form, Input, InputNumber, Switch } from 'antd';
|
import { Card, Button, Space, Select, message, Drawer, Form, Input, InputNumber, Switch, TimePicker, Divider } from 'antd';
|
||||||
import { Stage, Layer, Rect, Line, Circle, Text as KonvaText } from 'react-konva';
|
import { Stage, Layer, Rect, Line, Circle, Text as KonvaText } from 'react-konva';
|
||||||
|
import { RangePickerProps } from 'antd/es/date-picker';
|
||||||
import axios from 'axios';
|
import axios from 'axios';
|
||||||
|
import dayjs from 'dayjs';
|
||||||
|
|
||||||
|
interface WorkingHours {
|
||||||
|
start: number[];
|
||||||
|
end: number[];
|
||||||
|
}
|
||||||
|
|
||||||
interface ROI {
|
interface ROI {
|
||||||
id: number;
|
id: number;
|
||||||
@@ -13,6 +20,7 @@ interface ROI {
|
|||||||
threshold_sec: number;
|
threshold_sec: number;
|
||||||
confirm_sec: number;
|
confirm_sec: number;
|
||||||
return_sec: number;
|
return_sec: number;
|
||||||
|
working_hours: WorkingHours[] | null;
|
||||||
}
|
}
|
||||||
|
|
||||||
interface Camera {
|
interface Camera {
|
||||||
@@ -95,12 +103,18 @@ const ROIEditor: React.FC = () => {
|
|||||||
const handleSaveROI = async (values: any) => {
|
const handleSaveROI = async (values: any) => {
|
||||||
if (!selectedCamera || !selectedROI) return;
|
if (!selectedCamera || !selectedROI) return;
|
||||||
try {
|
try {
|
||||||
|
const workingHours = values.working_hours?.map((item: any) => ({
|
||||||
|
start: [item.start.hour(), item.start.minute()],
|
||||||
|
end: [item.end.hour(), item.end.minute()],
|
||||||
|
}));
|
||||||
|
|
||||||
await axios.put(`/api/camera/${selectedCamera}/roi/${selectedROI.id}`, {
|
await axios.put(`/api/camera/${selectedCamera}/roi/${selectedROI.id}`, {
|
||||||
name: values.name,
|
name: values.name,
|
||||||
roi_type: values.roi_type,
|
roi_type: values.roi_type,
|
||||||
rule_type: values.rule_type,
|
rule_type: values.rule_type,
|
||||||
threshold_sec: values.threshold_sec,
|
threshold_sec: values.threshold_sec,
|
||||||
confirm_sec: values.confirm_sec,
|
confirm_sec: values.confirm_sec,
|
||||||
|
working_hours: workingHours,
|
||||||
enabled: values.enabled,
|
enabled: values.enabled,
|
||||||
});
|
});
|
||||||
message.success('保存成功');
|
message.success('保存成功');
|
||||||
@@ -150,6 +164,7 @@ const ROIEditor: React.FC = () => {
|
|||||||
threshold_sec: 60,
|
threshold_sec: 60,
|
||||||
confirm_sec: 5,
|
confirm_sec: 5,
|
||||||
return_sec: 5,
|
return_sec: 5,
|
||||||
|
working_hours: [],
|
||||||
})
|
})
|
||||||
.then(() => {
|
.then(() => {
|
||||||
message.success('ROI添加成功');
|
message.success('ROI添加成功');
|
||||||
@@ -211,6 +226,10 @@ const ROIEditor: React.FC = () => {
|
|||||||
threshold_sec: roi.threshold_sec,
|
threshold_sec: roi.threshold_sec,
|
||||||
confirm_sec: roi.confirm_sec,
|
confirm_sec: roi.confirm_sec,
|
||||||
enabled: roi.enabled,
|
enabled: roi.enabled,
|
||||||
|
working_hours: roi.working_hours?.map((wh: WorkingHours) => ({
|
||||||
|
start: dayjs().hour(wh.start[0]).minute(wh.start[1]),
|
||||||
|
end: dayjs().hour(wh.end[0]).minute(wh.end[1]),
|
||||||
|
})),
|
||||||
});
|
});
|
||||||
setDrawerVisible(true);
|
setDrawerVisible(true);
|
||||||
}}
|
}}
|
||||||
@@ -368,6 +387,10 @@ const ROIEditor: React.FC = () => {
|
|||||||
threshold_sec: roi.threshold_sec,
|
threshold_sec: roi.threshold_sec,
|
||||||
confirm_sec: roi.confirm_sec,
|
confirm_sec: roi.confirm_sec,
|
||||||
enabled: roi.enabled,
|
enabled: roi.enabled,
|
||||||
|
working_hours: roi.working_hours?.map((wh: WorkingHours) => ({
|
||||||
|
start: dayjs().hour(wh.start[0]).minute(wh.start[1]),
|
||||||
|
end: dayjs().hour(wh.end[0]).minute(wh.end[1]),
|
||||||
|
})),
|
||||||
});
|
});
|
||||||
setDrawerVisible(true);
|
setDrawerVisible(true);
|
||||||
}}
|
}}
|
||||||
@@ -434,6 +457,37 @@ const ROIEditor: React.FC = () => {
|
|||||||
<Form.Item name="confirm_sec" label="确认时间(秒)" rules={[{ required: true }]}>
|
<Form.Item name="confirm_sec" label="确认时间(秒)" rules={[{ required: true }]}>
|
||||||
<InputNumber min={5} style={{ width: '100%' }} />
|
<InputNumber min={5} style={{ width: '100%' }} />
|
||||||
</Form.Item>
|
</Form.Item>
|
||||||
|
<Divider>工作时间配置(可选)</Divider>
|
||||||
|
<Form.List name="working_hours">
|
||||||
|
{(fields, { add, remove }) => (
|
||||||
|
<div>
|
||||||
|
{fields.map((field, index) => (
|
||||||
|
<Space key={field.key} align="baseline" style={{ display: 'flex', marginBottom: 8 }}>
|
||||||
|
<Form.Item
|
||||||
|
{...field}
|
||||||
|
label={index === 0 ? '时间段' : ''}
|
||||||
|
style={{ marginBottom: 0 }}
|
||||||
|
>
|
||||||
|
<TimePicker.RangePicker format="HH:mm" />
|
||||||
|
</Form.Item>
|
||||||
|
<Button
|
||||||
|
type="link"
|
||||||
|
danger
|
||||||
|
onClick={() => remove(field.name)}
|
||||||
|
>
|
||||||
|
删除
|
||||||
|
</Button>
|
||||||
|
</Space>
|
||||||
|
))}
|
||||||
|
<Button type="dashed" onClick={() => add({ start: null, end: null })} block>
|
||||||
|
添加时间段
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</Form.List>
|
||||||
|
<Form.Item style={{ fontSize: 12, color: '#999' }}>
|
||||||
|
不配置工作时间则使用系统全局设置
|
||||||
|
</Form.Item>
|
||||||
</>
|
</>
|
||||||
)}
|
)}
|
||||||
<Form.Item name="enabled" label="启用状态" valuePropName="checked">
|
<Form.Item name="enabled" label="启用状态" valuePropName="checked">
|
||||||
|
|||||||
@@ -190,6 +190,7 @@ class InferencePipeline:
|
|||||||
"threshold_sec": roi_config.get("threshold_sec", 300),
|
"threshold_sec": roi_config.get("threshold_sec", 300),
|
||||||
"confirm_sec": roi_config.get("confirm_sec", 10),
|
"confirm_sec": roi_config.get("confirm_sec", 10),
|
||||||
"return_sec": roi_config.get("return_sec", 30),
|
"return_sec": roi_config.get("return_sec", 30),
|
||||||
|
"working_hours": roi_config.get("working_hours"),
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -293,11 +293,12 @@ class AlgorithmManager:
|
|||||||
algo_params.update(params)
|
algo_params.update(params)
|
||||||
|
|
||||||
if algorithm_type == "leave_post":
|
if algorithm_type == "leave_post":
|
||||||
|
roi_working_hours = algo_params.get("working_hours") or self.working_hours
|
||||||
self.algorithms[roi_id]["leave_post"] = LeavePostAlgorithm(
|
self.algorithms[roi_id]["leave_post"] = LeavePostAlgorithm(
|
||||||
threshold_sec=algo_params.get("threshold_sec", 300),
|
threshold_sec=algo_params.get("threshold_sec", 300),
|
||||||
confirm_sec=algo_params.get("confirm_sec", 10),
|
confirm_sec=algo_params.get("confirm_sec", 10),
|
||||||
return_sec=algo_params.get("return_sec", 30),
|
return_sec=algo_params.get("return_sec", 30),
|
||||||
working_hours=self.working_hours,
|
working_hours=roi_working_hours,
|
||||||
)
|
)
|
||||||
elif algorithm_type == "intrusion":
|
elif algorithm_type == "intrusion":
|
||||||
self.algorithms[roi_id]["intrusion"] = IntrusionAlgorithm(
|
self.algorithms[roi_id]["intrusion"] = IntrusionAlgorithm(
|
||||||
|
|||||||
Binary file not shown.
Reference in New Issue
Block a user