76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
from fastapi import APIRouter, Depends, Response
|
|
from sqlalchemy.orm import Session
|
|
from app.models.database import get_db
|
|
from app.models.user import UserInfo,UserDB, PhoneLoginRequest, generate_user_code
|
|
from app.core.response import success_response, error_response, ResponseModel
|
|
from app.core.wechat import WeChatClient
|
|
from app.core.security import create_access_token, set_jwt_cookie
|
|
from pydantic import BaseModel
|
|
|
|
router = APIRouter()
|
|
|
|
class PhoneNumberRequest(BaseModel):
|
|
code: str # 手机号获取凭证
|
|
referral_code: str = None # 推荐码（可选）
|
|
|
|
@router.post("/phone-login", response_model=ResponseModel)
|
|
async def wechat_phone_login(
|
|
request: PhoneNumberRequest,
|
|
db: Session = Depends(get_db),
|
|
response: Response = None
|
|
):
|
|
"""通过微信手机号登录/注册"""
|
|
try:
|
|
# 初始化微信客户端
|
|
wechat = WeChatClient()
|
|
|
|
# 获取用户手机号
|
|
phone_info = await wechat.get_phone_number(request.code)
|
|
|
|
if not phone_info or not phone_info.get('phone_number'):
|
|
return error_response(code=400, message="获取手机号失败")
|
|
|
|
phone = phone_info['phone_number']
|
|
|
|
# 查找或创建用户
|
|
user = db.query(UserDB).filter(UserDB.phone == phone).first()
|
|
if not user:
|
|
# 生成用户编码
|
|
user_code = generate_user_code(db)
|
|
|
|
user = UserDB(
|
|
username=f"user_{phone[-4:]}",
|
|
phone=phone,
|
|
user_code=user_code,
|
|
referral_code=request.referral_code
|
|
)
|
|
db.add(user)
|
|
db.flush() # 获取用户ID
|
|
|
|
# 发放优惠券
|
|
from app.api.endpoints.user import issue_register_coupons
|
|
issue_register_coupons(db, user.userid)
|
|
|
|
db.commit()
|
|
db.refresh(user)
|
|
|
|
# 创建访问令牌
|
|
access_token = create_access_token(
|
|
data={"phone": user.phone, "userid": user.userid}
|
|
)
|
|
|
|
# 设置JWT cookie
|
|
if response:
|
|
set_jwt_cookie(response, access_token)
|
|
|
|
return success_response(
|
|
message="登录成功",
|
|
data={
|
|
"user": UserInfo.model_validate(user),
|
|
"access_token": access_token,
|
|
"token_type": "bearer"
|
|
}
|
|
)
|
|
except Exception as e:
|
|
db.rollback()
|
|
return error_response(code=500, message=f"登录失败: {str(e)}") |