TradeMate Dev
23a31f7c00
- Add silent WeChat login for MP/browser environments - Fix Python 3.6 compatibility (remove typing.Annotated usage) - Marketing page: tab-based content generation with category support - Translate page: add auto-detect language default - Homepage: add TTS playback, announcement ticker, remove redundant quick-actions - Fix FAB button overlap with custom tabbar on customers/quotation pages - Make openai/anthropic imports lazy for Python 3.6 compat
64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from typing import Optional
|
|
from pydantic import BaseModel
|
|
from app.database import get_db
|
|
from app.services.push import PushService
|
|
from app.api.v1.deps import get_current_user_id
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
class DeviceRegisterRequest(BaseModel):
|
|
client_id: str
|
|
platform: str = "weapp"
|
|
push_token: Optional[str] = None
|
|
device_info: Optional[dict] = None
|
|
|
|
|
|
@router.post("/register")
|
|
async def register_device(
|
|
data: DeviceRegisterRequest,
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
service = PushService(db)
|
|
device = await service.register_device(
|
|
user_id=user_id,
|
|
client_id=data.client_id,
|
|
platform=data.platform,
|
|
push_token=data.push_token,
|
|
device_info=data.device_info,
|
|
)
|
|
return {
|
|
"success": True,
|
|
"device_id": str(device.id),
|
|
"message": "Device registered",
|
|
}
|
|
|
|
|
|
@router.post("/unregister")
|
|
async def unregister_device(
|
|
data: dict,
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
client_id = data.get("client_id")
|
|
if not client_id:
|
|
raise HTTPException(status_code=400, detail="client_id required")
|
|
service = PushService(db)
|
|
success = await service.unregister_device(user_id, client_id)
|
|
if not success:
|
|
raise HTTPException(status_code=404, detail="Device not found")
|
|
return {"success": True, "message": "Device unregistered"}
|
|
|
|
|
|
@router.get("/devices")
|
|
async def list_devices(
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
service = PushService(db)
|
|
devices = await service.get_user_devices(user_id)
|
|
return {"devices": devices, "count": len(devices)}
|