# coding=utf-8

import os
import time

CAIHUI_DATABASE = {
    'name': os.environ.get('DISH_DB_NAME', 'goods'),
    'host': os.environ.get('CARC_CRAWLER_DB_HOST', '127.0.0.1'),
    'port': int(os.environ.get('CARC_CRAWLER_DB_PORT', 3306)),
    'user': os.environ.get('CARC_CRAWLER_DB_USER', 'root'),
    'password': os.environ.get('CARC_CRAWLER_DB_PASSWORD', '123456'),
}

# SQLite 本地临时缓存（识别读此库；变更后从 MySQL 拉取刷新）
DISH_SQLITE_PATH = os.environ.get(
    'DISH_SQLITE_PATH',
    os.path.join(os.path.dirname(__file__), 'dish_templates.db'),
)

# 本机基地编码（学菜写入 source_base_id；结果 API 的 base_id 默认也用它）
# 建议填管理系统返回的 code，例如「上海路基地-1788059884828-e55vu」
DISH_BASE_CODE = os.environ.get('DISH_BASE_CODE', 'BASE_LOCAL_01')
DISH_BASE_NAME = os.environ.get('DISH_BASE_NAME', '奥体基地')
DISH_BASE_REGION = os.environ.get('DISH_BASE_REGION', '')

# 管理系统：公开接口（X-API-Key）
# 完整 Key 见 docs/第三方接口文档_mgmt.md（含 dish_ 前缀）
DISH_MGMT_API_ORIGIN = os.environ.get(
    'DISH_MGMT_API_ORIGIN',
    'https://cpsb.ruimeida.cc',
).rstrip('/')
DISH_MGMT_API_URL = os.environ.get(
    'DISH_MGMT_API_URL',
    f'{DISH_MGMT_API_ORIGIN}/api/public/bases',
)
DISH_MGMT_RESULT_API_URL = os.environ.get(
    'DISH_MGMT_RESULT_API_URL',
    f'{DISH_MGMT_API_ORIGIN}/api/public/recognition-orders',
)
DISH_MGMT_UNRECOGNIZED_API_URL = os.environ.get(
    'DISH_MGMT_UNRECOGNIZED_API_URL',
    f'{DISH_MGMT_API_ORIGIN}/api/public/unrecognized-samples',
)
DISH_MGMT_MEAL_API_URL = os.environ.get(
    'DISH_MGMT_MEAL_API_URL',
    f'{DISH_MGMT_API_ORIGIN}/api/public/meal-schedules/today',
)
DISH_MGMT_DEVICE_STATUS_URL = os.environ.get(
    'DISH_MGMT_DEVICE_STATUS_URL',
    f'{DISH_MGMT_API_ORIGIN}/api/public/devices/status',
)
DISH_MGMT_DISHES_API_URL = os.environ.get(
    'DISH_MGMT_DISHES_API_URL',
    f'{DISH_MGMT_API_ORIGIN}/api/public/dishes',
)
DISH_MGMT_DISH_LEARNING_STATUS_URL = os.environ.get(
    'DISH_MGMT_DISH_LEARNING_STATUS_URL',
    f'{DISH_MGMT_API_ORIGIN}/api/public/dishes/learning-status',
)
DISH_MGMT_API_KEY = os.environ.get(
    'DISH_MGMT_API_KEY',
    'dish_a4f7a1287198179fdc6a6571aa14064c2afc6338c6730e2b668ac5be15bdf769',
)
DISH_MGMT_API_TIMEOUT = float(os.environ.get('DISH_MGMT_API_TIMEOUT', '10'))
DISH_MGMT_SYNC_ON_START = os.environ.get('DISH_MGMT_SYNC_ON_START', '1') == '1'
DISH_MGMT_MEAL_ON_START = os.environ.get('DISH_MGMT_MEAL_ON_START', '1') == '1'

# 本机设备（上报设备状态）
DISH_DEVICE_CODE = os.environ.get('DISH_DEVICE_CODE', '') or os.environ.get('DISH_BASE_CODE', DISH_BASE_CODE)
DISH_DEVICE_NAME = os.environ.get('DISH_DEVICE_NAME', '') or os.environ.get('DISH_BASE_NAME', DISH_BASE_NAME)
DISH_DEVICE_STATUS_INTERVAL = int(os.environ.get('DISH_DEVICE_STATUS_INTERVAL', '60'))  # 秒
DISH_DEVICE_STATUS_ENABLED = os.environ.get('DISH_DEVICE_STATUS_ENABLED', '1') == '1'

# 启动时是否从 MySQL 拉取；后台定时同步间隔（秒，含上传队列 flush + 拉取）
DISH_SYNC_ON_START = os.environ.get('DISH_SYNC_ON_START', '1') == '1'
DISH_SYNC_INTERVAL = int(os.environ.get('DISH_SYNC_INTERVAL', 300))  # 默认 5 分钟

# 识别框是否显示「似xx 检xx」（默认关；需要时 set DISH_SHOW_BOX_SCORES=1）
DISH_SHOW_BOX_SCORES = os.environ.get('DISH_SHOW_BOX_SCORES', '0') == '1'

# 当前餐段匹配：有当日排餐时仅在「今日当前餐段」菜品中比对，未命中直接未知（不回退大库）
# 1) 最高分 ≥ 0.80 → 认定；2) 未达 0.80 但最高分 > 0.75 → 仍认定；否则未知
DISH_MEAL_SMALL_LIB_THRESHOLD = float(os.environ.get('DISH_MEAL_SMALL_LIB_THRESHOLD', '0.82'))
DISH_MEAL_SMALL_LIB_SOFT_THRESHOLD = float(os.environ.get('DISH_MEAL_SMALL_LIB_SOFT_THRESHOLD', '0.82'))
# 早于该时刻且尚无当日排餐 → 不做当前餐段限制，直接走大库
DISH_MEAL_CUTOFF_HOUR = int(os.environ.get('DISH_MEAL_CUTOFF_HOUR', '7'))

# 主食小库：不在每日排餐中，常驻候选；与「当前餐段菜品小库」双路匹配
# 环境变量 DISH_STAPLE_NAMES 逗号分隔覆盖；仅当本地模板库已有同名模板时才参与比对
_DISH_STAPLE_NAMES_DEFAULT = (
    '米饭,白米饭,杂粮饭,黑米饭,馒头,花卷,包子,面条,汤面,炒饭,盖浇饭,粥,白粥,小米粥,馄饨,饺子,'
    '酸奶,香蕉,苹果,梨子,葡萄,营养汤,土豆,黄瓜,山药,红薯,玉米'
)
DISH_STAPLE_NAMES = [
    x.strip() for x in os.environ.get('DISH_STAPLE_NAMES', _DISH_STAPLE_NAMES_DEFAULT).split(',')
    if x.strip()
]
DISH_STAPLE_LIB_THRESHOLD = float(os.environ.get('DISH_STAPLE_LIB_THRESHOLD', '0.80'))
DISH_STAPLE_LIB_SOFT_THRESHOLD = float(os.environ.get('DISH_STAPLE_LIB_SOFT_THRESHOLD', '0.81'))
# 主食与餐段菜都达标时：主食分 ≥ 菜分 - margin 则优先认主食（减轻主食被菜品抢走）
DISH_STAPLE_PRIORITY_MARGIN = float(os.environ.get('DISH_STAPLE_PRIORITY_MARGIN', '0.03'))
# 每餐开始前多少分钟拉取排餐（默认 60 = 开餐前 1 小时）
DISH_MEAL_PREFETCH_MINUTES = int(os.environ.get('DISH_MEAL_PREFETCH_MINUTES', '60'))
# 尚无餐段信息时的兜底拉取时刻（启动后第一次排餐未到位时用）
DISH_MEAL_REFRESH_HOUR = int(os.environ.get('DISH_MEAL_REFRESH_HOUR', '6'))
DISH_MEAL_REFRESH_MINUTE = int(os.environ.get('DISH_MEAL_REFRESH_MINUTE', '0'))

# 实时扫描：未知菜写入待学习前的最短稳定秒数（客人放盘短，默认 1 秒）
# 盘子移走时不刷入，避免移盘过程中的误检未知
DISH_LIVE_PENDING_STABLE_SECONDS = float(os.environ.get('DISH_LIVE_PENDING_STABLE_SECONDS', '1.0'))
# 运动判定：从动到静后，连续多少帧与静止锚点相似，才触发 YOLO 一次（按相机帧，不按秒）。
# 持续静止不再识别，须再经历「静→动→静」。默认 4 帧（约对应原 0.2s@20Hz 轮询）。
DISH_LIVE_STILL_HOLD_FRAMES = int(os.environ.get('DISH_LIVE_STILL_HOLD_FRAMES', '4'))
# 已有订单时，菜品区域相似度阈值（更低=更宽松，手在中间晃仍算静止，减少误触发）
DISH_LIVE_MOTION_SIM_WITH_ORDER = float(os.environ.get('DISH_LIVE_MOTION_SIM_WITH_ORDER', '0.970'))
# 订单防抖：菜品名称集合去重 + 时间间隔（秒），避免同一次晃动重复上传
# 默认 2s（建议 2~3s）
DISH_LIVE_ORDER_DEBOUNCE_SECONDS = float(os.environ.get('DISH_LIVE_ORDER_DEBOUNCE_SECONDS', '2'))
# 无餐盘持续该秒数后清场解锁，允许下一盘（含相同菜名）重新存图/上报
DISH_LIVE_ORDER_CLEAR_HOLD_SECONDS = float(os.environ.get('DISH_LIVE_ORDER_CLEAR_HOLD_SECONDS', '1.5'))
# 同场景去重：按预览帧计数（约 25fps，默认 40 帧 ≈ 1.6s）。
# 窗口内即使个别菜名不同，只要检测框布局相近也视为同一单（不重复上报）
DISH_LIVE_ORDER_SAME_SCENE_FRAMES = int(os.environ.get('DISH_LIVE_ORDER_SAME_SCENE_FRAMES', '40'))
# 框匹配 IoU 阈值；匹配对数占比阈值（0~1）
DISH_LIVE_ORDER_LAYOUT_IOU = float(os.environ.get('DISH_LIVE_ORDER_LAYOUT_IOU', '0.40'))
DISH_LIVE_ORDER_LAYOUT_MATCH_RATIO = float(os.environ.get('DISH_LIVE_ORDER_LAYOUT_MATCH_RATIO', '0.70'))

# YOLO 检测需直接丢弃的类别（不裁剪、不识别、不上报），逗号分隔
DISH_DETECT_FILTER_NAMES = [
    x.strip() for x in os.environ.get('DISH_DETECT_FILTER_NAMES', '手,空盘').split(',')
    if x.strip()
]
# 贴边不完整框过滤：托盘刚入画只露一角时 YOLO 常会检出，直接丢弃
# 框任一边缘距画面边界 ≤ margin 像素，视为贴边
DISH_DETECT_EDGE_MARGIN_PX = int(os.environ.get('DISH_DETECT_EDGE_MARGIN_PX', '8'))
# 贴边且框面积占整图比例低于此值 → 丢弃（默认 2%）
DISH_DETECT_EDGE_MIN_AREA_RATIO = float(os.environ.get('DISH_DETECT_EDGE_MIN_AREA_RATIO', '0.02'))
# 贴边且框较短边像素低于此值 → 丢弃
DISH_DETECT_EDGE_MIN_SIDE_PX = int(os.environ.get('DISH_DETECT_EDGE_MIN_SIDE_PX', '48'))
# 任意框（不论是否贴边）面积占比低于此值也丢弃；0=关闭
DISH_DETECT_MIN_AREA_RATIO = float(os.environ.get('DISH_DETECT_MIN_AREA_RATIO', '0.008'))

# 用本地视频文件夹代替相机实时流（调试/回放，主要给海康入口用）。
# 1=读 DISH_VIDEO_SOURCE_DIR，0=读相机。usb_camera_app 固定读 USB 相机，不受此项影响。
DISH_USE_VIDEO_FOLDER = os.environ.get('DISH_USE_VIDEO_FOLDER', '0') == '1'
DISH_VIDEO_SOURCE_DIR = os.environ.get('DISH_VIDEO_SOURCE_DIR', 'videos')

# 相机预览流本地录像：相机连接后自动录，只保留最近 N 分钟（滚动删除旧段）
# 视频回放模式下默认关闭，避免边读边写同一目录
_STREAM_RECORD_DEFAULT = '0' if DISH_USE_VIDEO_FOLDER else '1'
DISH_STREAM_RECORD_ENABLED = os.environ.get(
    'DISH_STREAM_RECORD_ENABLED', _STREAM_RECORD_DEFAULT,
) == '1'
DISH_STREAM_RECORD_DIR = os.environ.get('DISH_STREAM_RECORD_DIR', 'videos')
DISH_STREAM_RETENTION_MINUTES = float(os.environ.get('DISH_STREAM_RETENTION_MINUTES', '3000'))
# 单段时长（秒），默认 5 分钟一切，便于清理
DISH_STREAM_SEGMENT_SECONDS = float(os.environ.get('DISH_STREAM_SEGMENT_SECONDS', '350'))
DISH_STREAM_RECORD_FPS = float(os.environ.get('DISH_STREAM_RECORD_FPS', '10'))
DISH_STREAM_MAX_WIDTH = int(os.environ.get('DISH_STREAM_MAX_WIDTH', '1280'))


DEBUG = True


LOG_DIR = os.path.join(os.getcwd(), f'log/{time.strftime("%Y-%m-%d")}.log')
LOG_FORMAT = '<level>{level: <8}</level>  <green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> - <cyan>{name}</cyan>:<cyan>{function}</cyan> - <level>{message}</level>'