43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from typing import List
|
|
|
|
from fastapi import APIRouter, Depends, Query
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.database import get_db
|
|
from app.core.dependencies import get_current_user
|
|
from app.models import User
|
|
from app.services.wallet_service import WalletService
|
|
from app.datamodels.schemas import (
|
|
BalanceResponse, RedeemCodeRequest, TransactionResponse,
|
|
)
|
|
|
|
router = APIRouter(prefix="/wallet", tags=["wallet"])
|
|
|
|
|
|
@router.get("/balance", response_model=BalanceResponse)
|
|
async def get_balance(
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
balance = await WalletService.get_balance(db, user.id)
|
|
return {"balance": balance}
|
|
|
|
|
|
@router.post("/redeem", response_model=TransactionResponse)
|
|
async def redeem_code(
|
|
body: RedeemCodeRequest,
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
return await WalletService.redeem_code(db, user.id, body.code)
|
|
|
|
|
|
@router.get("/transactions", response_model=List[TransactionResponse])
|
|
async def list_transactions(
|
|
page: int = Query(1, ge=1),
|
|
size: int = Query(20, ge=1, le=100),
|
|
user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
return await WalletService.list_transactions(db, user.id, page, size)
|