You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
from datetime import datetime
|
|
from fastapi import FastAPI
|
|
from pydantic import BaseModel
|
|
from loguru import logger
|
|
|
|
app = FastAPI(title="直播状态控制接口")
|
|
app.shared_config = None
|
|
|
|
class SpeakingStatus(BaseModel):
|
|
is_speaking: bool
|
|
|
|
@app.post("/speaking-status", summary="更新说话状态")
|
|
def update_speaking_status(status: SpeakingStatus):
|
|
if app.shared_config is None:
|
|
return {"error": "服务未初始化"}
|
|
|
|
prev = app.shared_config.get("is_speaking", False)
|
|
app.shared_config["is_speaking"] = status.is_speaking
|
|
|
|
# 当状态从 True 变为 False 的瞬间，取出预测队列并计算差值
|
|
if (not status.is_speaking) and prev:
|
|
now_dt = datetime.now()
|
|
now_str = now_dt.strftime("%Y-%m-%d %H:%M:%S")
|
|
app.shared_config["speaking_false_time"] = now_str
|
|
|
|
predicted_time_str = None
|
|
try:
|
|
q = app.shared_config.get("prediction_queue")
|
|
if q is not None and len(q) > 0:
|
|
predicted_time_str = q.pop(0)
|
|
except Exception:
|
|
predicted_time_str = None
|
|
|
|
app.shared_config["predicted_time"] = predicted_time_str
|
|
|
|
# 打印日志
|
|
if predicted_time_str:
|
|
try:
|
|
predicted_dt = datetime.strptime(predicted_time_str, "%Y-%m-%d %H:%M:%S")
|
|
delta_seconds = (now_dt - predicted_dt).total_seconds()
|
|
app.shared_config["time_diff_seconds"] = float(delta_seconds)
|
|
logger.info(
|
|
f"[计算差值] 队列剩余: {list(app.shared_config.get('prediction_queue', []))} | "
|
|
f"真实时间: {now_str} | 使用预测: {predicted_time_str} | 差值(秒): {delta_seconds}"
|
|
)
|
|
except Exception:
|
|
app.shared_config["time_diff_seconds"] = None
|
|
else:
|
|
app.shared_config["time_diff_seconds"] = None
|
|
logger.warning(
|
|
f"[计算差值] 队列为空或解析失败 | 真实时间: {now_str}"
|
|
)
|
|
|
|
return {"message": "状态更新成功"}
|