Upload routers/payments.py

routers/payments.py

@@ -0,0 +1,68 @@
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.future import select
+from typing import List
+from database.session import get_db
+from database.models import Transaction as TransactionModel, User as UserModel
+from auth import get_current_user
+import schemas
+
+router = APIRouter(
+    prefix="/payments",
+    tags=["payments"]
+)
+
+@router.post("/deposit", response_model=schemas.Transaction, status_code=status.HTTP_201_CREATED)
+async def create_deposit(
+    transaction: schemas.TransactionCreate, 
+    current_user: UserModel = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db)
+):
+    """
+    Create a deposit transaction (simulate payment for now).
+    """
+    if transaction.amount <= 0:
+        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="Amount must be positive")
+        
+    balance_before = current_user.balance
+    balance_after = balance_before + transaction.amount
+    
+    # Update user balance
+    current_user.balance = balance_after
+    
+    # Create transaction record
+    new_txn = TransactionModel(
+        user_id=current_user.id,
+        amount=transaction.amount,
+        type="Deposit",
+        payment_method=transaction.payment_method,
+        status="Completed",
+        balance_before=balance_before,
+        balance_after=balance_after,
+        note=transaction.note
+    )
+    
+    db.add(new_txn)
+    await db.commit()
+    await db.refresh(new_txn)
+    
+    return new_txn
+
+@router.get("/transactions", response_model=List[schemas.Transaction])
+async def get_transactions(
+    skip: int = 0, 
+    limit: int = 100, 
+    current_user: UserModel = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db)
+):
+    """
+    Get user's transactions.
+    """
+    result = await db.execute(
+        select(TransactionModel)
+        .where(TransactionModel.user_id == current_user.id)
+        .order_by(TransactionModel.created_at.desc())
+        .offset(skip)
+        .limit(limit)
+    )
+    return result.scalars().all()