Claude implementaton

Duration: 7.30
Iterations count: 99

Author: vodkar

backend/app/alembic/versions/fd8dcfe8d4fd_add_wallet_and_transaction_models.py

@@ -0,0 +1,55 @@
+"""add_wallet_and_transaction_models
+
+Revision ID: fd8dcfe8d4fd
+Revises: 1a31ce608336
+Create Date: 2025-09-12 22:28:29.785616
+
+"""
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel.sql.sqltypes
+
+
+# revision identifiers, used by Alembic.
+revision = 'fd8dcfe8d4fd'
+down_revision = '1a31ce608336'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # Create wallet table
+    op.create_table(
+        'wallet',
+        sa.Column('id', sa.UUID(), nullable=False),
+        sa.Column('user_id', sa.UUID(), nullable=False),
+        sa.Column('currency', sa.Enum('USD', 'EUR', 'RUB', name='currencyenum'), nullable=False),
+        sa.Column('balance', sa.Numeric(precision=10, scale=2), nullable=False),
+        sa.ForeignKeyConstraint(['user_id'], ['user.id'], ondelete='CASCADE'),
+        sa.PrimaryKeyConstraint('id')
+    )
+    
+    # Create transaction table
+    op.create_table(
+        'transaction',
+        sa.Column('id', sa.UUID(), nullable=False),
+        sa.Column('wallet_id', sa.UUID(), nullable=False),
+        sa.Column('amount', sa.Numeric(precision=10, scale=2), nullable=False),
+        sa.Column('type', sa.Enum('credit', 'debit', name='transactiontypeenum'), nullable=False),
+        sa.Column('currency', sa.Enum('USD', 'EUR', 'RUB', name='currencyenum'), nullable=False),
+        sa.Column('timestamp', sa.DateTime(), nullable=False),
+        sa.ForeignKeyConstraint(['wallet_id'], ['wallet.id'], ondelete='CASCADE'),
+        sa.PrimaryKeyConstraint('id')
+    )
+
+
+def downgrade():
+    # Drop transaction table
+    op.drop_table('transaction')
+    
+    # Drop wallet table
+    op.drop_table('wallet')
+    
+    # Drop enums
+    op.execute('DROP TYPE IF EXISTS transactiontypeenum')
+    op.execute('DROP TYPE IF EXISTS currencyenum')

backend/app/api/main.py

@@ -2,14 +2,16 @@
 
 from fastapi import APIRouter
 
-from app.api.routes import items, login, misc, private, users
+from app.api.routes import items, login, misc, private, transactions, users, wallets
 from app.core.config import settings
 
 api_router = APIRouter()
 api_router.include_router(login.router)
 api_router.include_router(users.router)
 api_router.include_router(misc.router)
 api_router.include_router(items.router)
+api_router.include_router(wallets.router)
+api_router.include_router(transactions.router)
 
 
 if settings.ENVIRONMENT == "local":

backend/app/api/routes/transactions.py

@@ -0,0 +1,67 @@
+"""Transaction management API endpoints."""
+
+import uuid
+
+from fastapi import APIRouter, HTTPException
+
+from app.api.deps import CurrentUser, SessionDep
+from app.constants import NOT_FOUND_CODE
+from app.crud import create_transaction, get_wallet_by_id, get_wallet_transactions
+from app.models import (
+    TransactionCreate,
+    TransactionPublic,
+    TransactionsPublic,
+)
+
+router = APIRouter(prefix="/transactions", tags=["transactions"])
+
+
+@router.post("/")
+def create_wallet_transaction(
+    *,
+    session: SessionDep,
+    current_user: CurrentUser,
+    transaction_in: TransactionCreate,
+) -> TransactionPublic:
+    """Create a new transaction for a wallet."""
+    transaction = create_transaction(
+        session=session,
+        transaction_in=transaction_in,
+        user_id=current_user.id,
+    )
+    return TransactionPublic.model_validate(transaction)
+
+
+@router.get("/wallet/{wallet_id}")
+def read_wallet_transactions(
+    session: SessionDep,
+    current_user: CurrentUser,
+    wallet_id: uuid.UUID,
+    skip: int = 0,
+    limit: int = 100,
+) -> TransactionsPublic:
+    """Get transactions for a specific wallet."""
+    # Verify wallet belongs to user (this is also checked in get_wallet_transactions)
+    wallet = get_wallet_by_id(session=session, wallet_id=wallet_id)
+    if not wallet:
+        raise HTTPException(status_code=NOT_FOUND_CODE, detail="Wallet not found")
+
+    if wallet.user_id != current_user.id and not current_user.is_superuser:
+        raise HTTPException(
+            status_code=403,
+            detail="Not authorized to access this wallet",
+        )
+
+    transactions = get_wallet_transactions(
+        session=session,
+        wallet_id=wallet_id,
+        skip=skip,
+        limit=limit,
+    )
+    transaction_data = [
+        TransactionPublic.model_validate(transaction) for transaction in transactions
+    ]
+    return TransactionsPublic(
+        transaction_data=transaction_data,
+        count=len(transaction_data),
+    )

backend/app/api/routes/wallets.py

@@ -0,0 +1,63 @@
+"""Wallet management API endpoints."""
+
+import uuid
+
+from fastapi import APIRouter, HTTPException
+
+from app.api.deps import CurrentUser, SessionDep
+from app.constants import BAD_REQUEST_CODE, NOT_FOUND_CODE
+from app.crud import create_wallet, get_user_wallets, get_wallet_by_id
+from app.models import (
+    WalletCreate,
+    WalletPublic,
+    WalletsPublic,
+)
+
+router = APIRouter(prefix="/wallets", tags=["wallets"])
+
+
+@router.post("/")
+def create_user_wallet(
+    *,
+    session: SessionDep,
+    current_user: CurrentUser,
+    wallet_in: WalletCreate,
+) -> WalletPublic:
+    """Create a new wallet for the current user."""
+    wallet = create_wallet(
+        session=session,
+        wallet_in=wallet_in,
+        user_id=current_user.id,
+    )
+    return WalletPublic.model_validate(wallet)
+
+
+@router.get("/")
+def read_user_wallets(
+    session: SessionDep,
+    current_user: CurrentUser,
+) -> WalletsPublic:
+    """Get all wallets for the current user."""
+    wallets = get_user_wallets(session=session, user_id=current_user.id)
+    wallet_data = [WalletPublic.model_validate(wallet) for wallet in wallets]
+    return WalletsPublic(wallet_data=wallet_data, count=len(wallet_data))
+
+
+@router.get("/{wallet_id}")
+def read_wallet(
+    session: SessionDep,
+    current_user: CurrentUser,
+    wallet_id: uuid.UUID,
+) -> WalletPublic:
+    """Get wallet details by ID."""
+    wallet = get_wallet_by_id(session=session, wallet_id=wallet_id)
+    if not wallet:
+        raise HTTPException(status_code=NOT_FOUND_CODE, detail="Wallet not found")
+    
+    if wallet.user_id != current_user.id and not current_user.is_superuser:
+        raise HTTPException(
+            status_code=BAD_REQUEST_CODE,
+            detail="Not enough permissions",
+        )
+    
+    return WalletPublic.model_validate(wallet)