92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
from redis import Redis
|
|
from datetime import timedelta
|
|
from openai import OpenAI
|
|
|
|
# Redis配置
|
|
REDIS_CONFIG = {
|
|
"host": "222.186.10.253",
|
|
"port": 6379,
|
|
"password": "Obscura@2024",
|
|
"decode_responses": True
|
|
}
|
|
|
|
# 摄像头数据库映射
|
|
CAMERA_DB_MAPPING = {
|
|
"A01": 210,
|
|
"B02": 211,
|
|
"C03": 212,
|
|
"report": 213 # 分析报告使用213数据库
|
|
}
|
|
|
|
# 创建Redis连接池
|
|
def create_redis_connections():
|
|
redis_connections = {}
|
|
for camera_id, db in CAMERA_DB_MAPPING.items():
|
|
redis_connections[camera_id] = Redis(
|
|
**REDIS_CONFIG,
|
|
db=db
|
|
)
|
|
return redis_connections
|
|
|
|
# SiliconFlow API配置
|
|
SILICON_FLOW_CONFIG = {
|
|
"base_url": "https://api.deepseek.com/v1",
|
|
"api_key": "sk-3027fb3c810b4e17985fa397d41250b9"
|
|
}
|
|
|
|
# 初始化OpenAI客户端
|
|
ai_client = OpenAI(
|
|
base_url=SILICON_FLOW_CONFIG["base_url"],
|
|
api_key=SILICON_FLOW_CONFIG["api_key"]
|
|
)
|
|
|
|
# 行为类别配置
|
|
BEHAVIOR_CATEGORIES = {
|
|
"基础动作": [
|
|
"站", "站立", "站着",
|
|
"走", "走路", "散步", "行走", "徒步",
|
|
"跑", "奔跑", "慢跑",
|
|
"坐", "坐下", "坐着",
|
|
"蹲", "蹲下", "蹲着",
|
|
"转", "转身", "转头", "回头", "旋转", "转向", "转弯",
|
|
"看", "闻", "嗅", "听"
|
|
],
|
|
"日常生活": [
|
|
"吃", "食用", "吃饭", "吃零食", "吃东西", "用餐", "咀嚼", "嚼",
|
|
"喝水", "喝牛奶", "喝茶", "饮用", "喝咖啡", "喝", "饮水",
|
|
"穿衣服", "穿裤子", "穿鞋", "戴帽子", "戴口罩", "戴围巾",
|
|
"躺", "睡", "睡觉", "休息", "打哈欠",
|
|
"洗澡", "刷牙", "洗手", "洗涤", "清洁", "擦洗",
|
|
"吃药", "喝药", "服药"
|
|
],
|
|
"社交活动": [
|
|
"说话", "交流", "演讲", "谈话", "聊天", "采访", "社交",
|
|
"打麻将", "打牌", "玩手机", "玩电脑", "玩游戏", "赌博",
|
|
"笑", "大笑", "微笑", "哭泣", "咯咯笑", "皱眉"
|
|
],
|
|
"工作学习": [
|
|
"读书", "阅读", "看书",
|
|
"写作", "写字", "写",
|
|
"工作", "学习", "使用电脑", "使用笔记本电脑", "使用手机", "开会", "打字",
|
|
"画画", "绘画", "摄影", "素描"
|
|
],
|
|
"运动娱乐": [
|
|
"跳", "跳跃", "跳舞", "游泳", "运动", "健身", "锻炼"
|
|
],
|
|
"异常行为": [
|
|
'打架', '斗殴', '摔倒', '晕倒', '昏倒', '跌倒', '滑倒',
|
|
'摔', '踢', '受伤', '暴力', '攻击', '威胁', '破坏',
|
|
'偷窃', '抢夺', '游荡', '徘徊', '尾随', '骚扰'
|
|
],
|
|
"其他": ["其他"]
|
|
}
|
|
|
|
# 异常行为列表
|
|
ABNORMAL_BEHAVIORS = BEHAVIOR_CATEGORIES["异常行为"]
|
|
|
|
# Redis缓存配置
|
|
REDIS_CACHE_CONFIG = {
|
|
"report_expiry": timedelta(days=30), # 报告缓存30天
|
|
"task_status_expiry": timedelta(hours=1) # 任务状态缓存1小时
|
|
}
|