fix(card): 非会员限制存卡数 非会员限制存卡数量，默认18张

core/routers/card/routes.py

@@ -24,6 +24,7 @@ from core.utils.helper import generate_keyname, check_picture_for_empty
 
 from . import schema
 from .scraper import card_scraper
+from .services import check_user_card_created_permission
 
 logger = logging.getLogger(__name__)
 card_router = APIRouter(prefix='/card', tags=['卡册模块'])
@@ -190,6 +191,14 @@ async def created_card(
         db_session: AsyncSession = Depends(get_async_session)
 ):
     card_crud = CRUD(db_session, Card)
+    # 确认用户存卡权限
+    if await check_user_card_created_permission(
+            db_session, login_user=login_user
+    ) is False:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST, detail=f'非会员仅支持存入{settings.USER_MAX_CREATED_CARD_NUM}张卡片'
+        )
+
     # 卡片库查询
     card: Card = await card_crud.get(filters={'cert_number': payload.cert_number})
     if payload.source.value != 'other':

core/routers/card/services.py

@@ -7,7 +7,11 @@
 from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from core.db.models import SearchCardCache
+from core.db.crud import CRUD
+from core.db.models import SearchCardCache, User, UserCollection
+from core.routers.auth.schema import LoginUser
+
+from config import settings
 
 
 async def get_cache_card_by_number(db_session: AsyncSession, *, cert_number: str) -> SearchCardCache | None:
@@ -16,3 +20,13 @@ async def get_cache_card_by_number(db_session: AsyncSession, *, cert_number: str
     card: SearchCardCache | None = result.scalar_one_or_none()
 
     return card
+
+
+async def check_user_card_created_permission(db_session: AsyncSession, *, login_user: LoginUser) -> bool:
+    db_crud = CRUD(db_session, User)
+    user = await db_crud.get(filters={'user_id': login_user.user_id})
+    if user.is_vip():
+        return True
+
+    total = await db_crud.total(model=UserCollection, filters={'user_id': login_user.user_id})
+    return settings.USER_MAX_CREATED_CARD_NUM > total