TradeMate Dev
c6206787da
项目结构: - backend/ Python FastAPI 后端 - uni-app/ uni-app跨端前端 - docs/ 设计文档 - docker-compose.yml Docker编排 - nginx/scripts/systemd 运维配置 已完成功能: - 用户认证 (JWT) - 智能翻译 + 回复建议 - 营销素材生成 - 客户管理 + 沉默检测 - 报价单管理 - 产品库管理 - 汇率换算 - 推送通知 (uni-push) - WhatsApp Webhook框架 - Celery定时任务
147 lines
3.9 KiB
Python
147 lines
3.9 KiB
Python
from fastapi import APIRouter, Depends
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy import select
|
|
from typing import Optional, List
|
|
from pydantic import BaseModel
|
|
from app.database import get_db
|
|
from app.models.user import User
|
|
from app.core.security import decode_token
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
class DeviceRegister(BaseModel):
|
|
client_id: str
|
|
platform: Optional[str] = None
|
|
device_info: Optional[dict] = None
|
|
|
|
|
|
class PushMessage(BaseModel):
|
|
title: str
|
|
content: str
|
|
payload: Optional[dict] = None
|
|
target_type: str = "all"
|
|
target_value: Optional[str] = None
|
|
|
|
|
|
class PushResponse(BaseModel):
|
|
success: bool
|
|
message_id: Optional[str] = None
|
|
error: Optional[str] = None
|
|
|
|
|
|
# 模拟存储的设备信息（实际应存数据库）
|
|
devices_db = {}
|
|
|
|
|
|
@router.post("/register")
|
|
async def register_device(
|
|
data: DeviceRegister,
|
|
authorization: str = None,
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
if not authorization or not authorization.startswith("Bearer "):
|
|
return {"error": "Unauthorized"}, 401
|
|
|
|
payload = decode_token(authorization[7:])
|
|
if not payload:
|
|
return {"error": "Invalid token"}, 401
|
|
|
|
user_id = payload.get("sub")
|
|
|
|
if user_id not in devices_db:
|
|
devices_db[user_id] = []
|
|
|
|
existing = [d for d in devices_db[user_id] if d.get("client_id") == data.client_id]
|
|
if not existing:
|
|
devices_db[user_id].append({
|
|
"client_id": data.client_id,
|
|
"platform": data.platform,
|
|
"device_info": data.device_info,
|
|
})
|
|
|
|
return {"success": True, "message": "Device registered"}
|
|
|
|
|
|
@router.post("/send")
|
|
async def send_push(
|
|
message: PushMessage,
|
|
authorization: str = None,
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
if not authorization or not authorization.startswith("Bearer "):
|
|
return {"error": "Unauthorized"}, 401
|
|
|
|
payload = decode_token(authorization[7:])
|
|
if not payload:
|
|
return {"error": "Invalid token"}, 401
|
|
|
|
user_id = payload.get("sub")
|
|
|
|
user_devices = devices_db.get(user_id, [])
|
|
if not user_devices:
|
|
return PushResponse(success=False, error="No devices registered")
|
|
|
|
# 实际项目中这里调用 uni-push/极光等API
|
|
# 模拟返回成功
|
|
message_id = f"msg_{user_id}_{int(payload.get('iat', 0))}"
|
|
|
|
print(f"Push message to user {user_id}: {message.title} - {message.content}")
|
|
|
|
return PushResponse(success=True, message_id=message_id)
|
|
|
|
|
|
@router.post("/send-to-customer")
|
|
async def send_to_customer(
|
|
customer_id: str,
|
|
title: str,
|
|
content: str,
|
|
payload: Optional[dict] = None,
|
|
authorization: str = None,
|
|
):
|
|
"""
|
|
针对特定客户的推送通知
|
|
例如：客户沉默提醒、报价提醒等
|
|
"""
|
|
if not authorization or not authorization.startswith("Bearer "):
|
|
return {"error": "Unauthorized"}, 401
|
|
|
|
payload_data = decode_token(authorization[7:])
|
|
if not payload_data:
|
|
return {"error": "Invalid token"}, 401
|
|
|
|
user_id = payload_data.get("sub")
|
|
|
|
# 这里可以添加针对客户的特定逻辑
|
|
notification = {
|
|
"type": "customer_alert",
|
|
"customer_id": customer_id,
|
|
"title": title,
|
|
"content": content,
|
|
"payload": payload or {}
|
|
}
|
|
|
|
print(f"Customer notification for user {user_id}, customer {customer_id}: {title}")
|
|
|
|
return PushResponse(success=True, message_id=f"alert_{customer_id}")
|
|
|
|
|
|
@router.get("/devices")
|
|
async def list_devices(
|
|
authorization: str = None,
|
|
):
|
|
"""列出用户已注册的设备"""
|
|
if not authorization or not authorization.startswith("Bearer "):
|
|
return {"error": "Unauthorized"}, 401
|
|
|
|
payload = decode_token(authorization[7:])
|
|
if not payload:
|
|
return {"error": "Invalid token"}, 401
|
|
|
|
user_id = payload.get("sub")
|
|
user_devices = devices_db.get(user_id, [])
|
|
|
|
return {
|
|
"devices": user_devices,
|
|
"count": len(user_devices)
|
|
} |