TradeMate Dev
2a107a42f3
- New DB models: credit_packages, subscription_plans, user_credits, credit_consumptions, credit_purchases - CreditService: balance, deduct, add_credits, grant_free_trial, history - User API: /api/v1/credits/* (balance/history/packages/purchase/subscribe) - Admin API: /api/v1/admin/credit-* (CRUD packages/plans, user credits, consumptions) - PaymentService.create_credit_order + handle_callback for credit purchases - Credit deduction on: discovery, translate, marketing, ai_chat, followup - Free trial 30 credits on registration - Documentation: docs/CREDIT_SYSTEM.md
127 lines
3.7 KiB
Python
127 lines
3.7 KiB
Python
from fastapi import APIRouter, HTTPException, Depends, Header
|
|
from typing import Optional
|
|
from pydantic import BaseModel
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from app.database import get_db
|
|
from app.services.marketing import MarketingService
|
|
from app.services.preference import UserPreferenceService
|
|
from app.services.credit import CreditService
|
|
from app.core.security import decode_token
|
|
from app.api.v1.deps import get_current_user_id
|
|
from app.config import settings
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
class MarketingRequest(BaseModel):
|
|
product_name: str
|
|
description: str
|
|
category: Optional[str] = None
|
|
price: Optional[str] = None
|
|
keywords: Optional[list] = None
|
|
target: str = "US importers"
|
|
style: str = "professional"
|
|
language: str = "en"
|
|
count: int = 3
|
|
|
|
|
|
class KeywordsRequest(BaseModel):
|
|
product_name: str
|
|
description: str
|
|
category: Optional[str] = None
|
|
language: str = "en"
|
|
count: int = 10
|
|
|
|
|
|
class CompetitorRequest(BaseModel):
|
|
product_name: str
|
|
description: str
|
|
category: Optional[str] = None
|
|
market: str = "US"
|
|
|
|
|
|
@router.post("/generate")
|
|
async def generate_marketing(
|
|
data: MarketingRequest,
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
credit_svc = CreditService(db)
|
|
ok, balance = await credit_svc.deduct(user_id, "marketing_content")
|
|
if not ok:
|
|
raise HTTPException(
|
|
status_code=402,
|
|
detail=f"次数不足 (剩余 {balance:.1f}, 需要 5)"
|
|
)
|
|
|
|
service = MarketingService()
|
|
pref_service = UserPreferenceService(db)
|
|
pref_context = await pref_service.get_preference_context(user_id, "marketing")
|
|
|
|
product_info = {
|
|
"name": data.product_name,
|
|
"description": data.description,
|
|
"category": data.category,
|
|
"price": data.price,
|
|
"keywords": data.keywords,
|
|
}
|
|
results = await service.generate(product_info, data.target, data.style, data.language, data.count, pref_context)
|
|
|
|
return {
|
|
"results": results,
|
|
"product": data.product_name,
|
|
"target": data.target,
|
|
"count": len(results),
|
|
"credits_remaining": balance - 5,
|
|
}
|
|
|
|
|
|
@router.post("/keywords")
|
|
async def generate_keywords(
|
|
data: KeywordsRequest,
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
credit_svc = CreditService(db)
|
|
ok, balance = await credit_svc.deduct(user_id, "marketing_content")
|
|
if not ok:
|
|
raise HTTPException(
|
|
status_code=402,
|
|
detail=f"次数不足 (剩余 {balance:.1f}, 需要 5)"
|
|
)
|
|
|
|
service = MarketingService()
|
|
product_info = {
|
|
"name": data.product_name,
|
|
"description": data.description,
|
|
"category": data.category,
|
|
}
|
|
keywords = await service.generate_keywords(product_info, data.language, data.count)
|
|
|
|
return {"keywords": keywords, "product": data.product_name, "credits_remaining": balance - 5}
|
|
|
|
|
|
@router.post("/competitor-analysis")
|
|
async def competitor_analysis(
|
|
data: CompetitorRequest,
|
|
user_id: str = Depends(get_current_user_id),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
credit_svc = CreditService(db)
|
|
ok, balance = await credit_svc.deduct(user_id, "competitor_analysis")
|
|
if not ok:
|
|
raise HTTPException(
|
|
status_code=402,
|
|
detail=f"次数不足 (剩余 {balance:.1f}, 需要 10)"
|
|
)
|
|
|
|
service = MarketingService()
|
|
product_info = {
|
|
"name": data.product_name,
|
|
"description": data.description,
|
|
"category": data.category,
|
|
}
|
|
analysis = await service.analyze_competitors(product_info, data.market)
|
|
|
|
return {"analysis": analysis, "product": data.product_name, "market": data.market, "credits_remaining": balance - 10}
|