feat(card): 会员功能验证

This commit is contained in:
xingc
2025-04-28 17:37:39 +08:00
parent 2fb25588c0
commit c3c2ae8dd9
3 changed files with 11 additions and 4 deletions

View File

@@ -17,7 +17,7 @@ from config import settings
from core.base import schema as base_schema
from core.db.crud import CRUD
from core.db.engine import get_async_session, get_cache
from core.db.models import Card, SearchCardCache, UserCollection
from core.db.models import Card, SearchCardCache, UserCollection, User
from core.routers.auth.schema import LoginUser
from core.routers.auth.services import get_current_user_form_http
from core.utils.helper import generate_keyname, check_picture_for_empty
@@ -267,6 +267,10 @@ async def update_card_price_info(
db_session: AsyncSession = Depends(get_async_session),
):
db_crud = CRUD(db_session, UserCollection)
user = await db_crud.get(model=User, filters={'user_id': login_user.user_id})
if user.is_vip() is False:
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail='会员未开通或者已过期')
result = await db_crud.update(
data=payload.dict(exclude_none=True),
filters={'card_id': card_id, 'user_id': login_user.user_id}