132 lines
3.8 KiB
Python
132 lines
3.8 KiB
Python
from fastapi import APIRouter, HTTPException, Depends
|
|
from sqlalchemy.orm import Session
|
|
from sqlalchemy import and_
|
|
from typing import List
|
|
from app.models.address import AddressDB, AddressCreate, AddressUpdate, AddressInfo
|
|
from app.models.database import get_db
|
|
from app.api.deps import get_current_user
|
|
from app.models.user import UserDB
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("/", response_model=AddressInfo)
|
|
async def create_address(
|
|
address: AddressCreate,
|
|
db: Session = Depends(get_db),
|
|
current_user: UserDB = Depends(get_current_user)
|
|
):
|
|
"""创建配送地址"""
|
|
# 如果设置为默认地址，先将其他地址的默认状态取消
|
|
if address.is_default:
|
|
db.query(AddressDB).filter(
|
|
and_(
|
|
AddressDB.user_id == current_user.userid,
|
|
AddressDB.is_default == True
|
|
)
|
|
).update({"is_default": False})
|
|
|
|
db_address = AddressDB(
|
|
user_id=current_user.userid,
|
|
**address.model_dump()
|
|
)
|
|
db.add(db_address)
|
|
db.commit()
|
|
db.refresh(db_address)
|
|
return db_address
|
|
|
|
@router.get("/", response_model=List[AddressInfo])
|
|
async def get_addresses(
|
|
db: Session = Depends(get_db),
|
|
current_user: UserDB = Depends(get_current_user)
|
|
):
|
|
"""获取用户的所有配送地址"""
|
|
addresses = db.query(AddressDB).filter(
|
|
AddressDB.user_id == current_user.userid
|
|
).all()
|
|
return addresses
|
|
|
|
@router.put("/{address_id}", response_model=AddressInfo)
|
|
async def update_address(
|
|
address_id: int,
|
|
address: AddressUpdate,
|
|
db: Session = Depends(get_db),
|
|
current_user: UserDB = Depends(get_current_user)
|
|
):
|
|
"""更新配送地址"""
|
|
db_address = db.query(AddressDB).filter(
|
|
and_(
|
|
AddressDB.id == address_id,
|
|
AddressDB.user_id == current_user.userid
|
|
)
|
|
).first()
|
|
|
|
if not db_address:
|
|
raise HTTPException(status_code=404, detail="地址不存在")
|
|
|
|
# 如果设置为默认地址，先将其他地址的默认状态取消
|
|
update_data = address.model_dump(exclude_unset=True)
|
|
if update_data.get("is_default"):
|
|
db.query(AddressDB).filter(
|
|
and_(
|
|
AddressDB.user_id == current_user.userid,
|
|
AddressDB.is_default == True
|
|
)
|
|
).update({"is_default": False})
|
|
|
|
for key, value in update_data.items():
|
|
setattr(db_address, key, value)
|
|
|
|
db.commit()
|
|
db.refresh(db_address)
|
|
return db_address
|
|
|
|
@router.delete("/{address_id}")
|
|
async def delete_address(
|
|
address_id: int,
|
|
db: Session = Depends(get_db),
|
|
current_user: UserDB = Depends(get_current_user)
|
|
):
|
|
"""删除配送地址"""
|
|
result = db.query(AddressDB).filter(
|
|
and_(
|
|
AddressDB.id == address_id,
|
|
AddressDB.user_id == current_user.userid
|
|
)
|
|
).delete()
|
|
|
|
if not result:
|
|
raise HTTPException(status_code=404, detail="地址不存在")
|
|
|
|
db.commit()
|
|
return {"message": "地址已删除"}
|
|
|
|
@router.post("/{address_id}/set-default", response_model=AddressInfo)
|
|
async def set_default_address(
|
|
address_id: int,
|
|
db: Session = Depends(get_db),
|
|
current_user: UserDB = Depends(get_current_user)
|
|
):
|
|
"""设置默认地址"""
|
|
# 取消其他默认地址
|
|
db.query(AddressDB).filter(
|
|
and_(
|
|
AddressDB.user_id == current_user.userid,
|
|
AddressDB.is_default == True
|
|
)
|
|
).update({"is_default": False})
|
|
|
|
# 设置新的默认地址
|
|
db_address = db.query(AddressDB).filter(
|
|
and_(
|
|
AddressDB.id == address_id,
|
|
AddressDB.user_id == current_user.userid
|
|
)
|
|
).first()
|
|
|
|
if not db_address:
|
|
raise HTTPException(status_code=404, detail="地址不存在")
|
|
|
|
db_address.is_default = True
|
|
db.commit()
|
|
db.refresh(db_address)
|
|
return db_address |