69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
# -*- coding = utf-8 -*-
|
|
# @Time : 2025/2/27 上午12:15
|
|
# @File : actions.py
|
|
# @Software : PyCharm
|
|
# @Author : xingc
|
|
# @Desc :
|
|
import logging
|
|
from datetime import timedelta, datetime
|
|
|
|
from fastapi import Depends
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from core.base.schema import ProductCategory
|
|
from core.db.crud import CRUD
|
|
from core.db.engine import get_async_session
|
|
from core.db.models import User, Order, Product
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
async def actions_after_payment(
|
|
notify_params: dict,
|
|
db_session: AsyncSession = Depends(get_async_session)
|
|
) -> None:
|
|
try:
|
|
db_crud = CRUD(db_session, Order)
|
|
order_id = notify_params['out_trade_no']
|
|
order: Order = await db_crud.get(filters={'order_id': order_id})
|
|
if not order:
|
|
logger.error(f'Order not found -> {order_id}')
|
|
return
|
|
|
|
product: Product = await db_crud.get(model=Product, filters={'product_id': order.product_id})
|
|
if not product:
|
|
logger.warning(f'Product not found -> {order.product_id}')
|
|
return
|
|
|
|
user: User = await db_crud.get(model=User, filters={'user_id': order.user_id})
|
|
if not user:
|
|
logger.warning(f'User not found -> {order.order_id}')
|
|
return
|
|
|
|
if product.action == ProductCategory.VIP:
|
|
vip_expire_date = (
|
|
user.vip_expire_date or datetime.now()
|
|
) + timedelta(days=product.number)
|
|
await db_crud.update(
|
|
model=User,
|
|
data={
|
|
'vip_expire_date': vip_expire_date
|
|
},
|
|
filters={'user_id': order.user_id}
|
|
)
|
|
elif product.action == ProductCategory.POINTS:
|
|
points = (
|
|
user.points or 0
|
|
) + product.number
|
|
await db_crud.update(
|
|
model=User,
|
|
data={
|
|
'points': points
|
|
},
|
|
filters={'user_id': order.user_id}
|
|
)
|
|
else:
|
|
logger.error(f'Unknown action -> {product.action}')
|
|
except Exception as e:
|
|
logger.error(f'Payment Actions Exception occurred -> {e}')
|
|
logger.exception(e) |