Added changes

Author: vodkar

backend/app/alembic/versions/fd8dcfe8d4fd_add_wallet_and_transaction_models.py

@@ -5,51 +5,62 @@
 Create Date: 2025-09-12 22:28:29.785616
 
 """
-from alembic import op
-import sqlalchemy as sa
-import sqlmodel.sql.sqltypes
 
+import sqlalchemy as sa
+from alembic import op
 
 # revision identifiers, used by Alembic.
-revision = 'fd8dcfe8d4fd'
-down_revision = '1a31ce608336'
+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')
+        "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')
+        "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')
-    
+    op.drop_table("transaction")
+
     # Drop wallet table
-    op.drop_table('wallet')
-    
+    op.drop_table("wallet")
+
     # Drop enums
-    op.execute('DROP TYPE IF EXISTS transactiontypeenum')
-    op.execute('DROP TYPE IF EXISTS currencyenum')
+    op.execute("DROP TYPE IF EXISTS transactiontypeenum")
+    op.execute("DROP TYPE IF EXISTS currencyenum")

backend/app/api/routes/transactions.py

@@ -7,11 +7,7 @@
 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,
-)
+from app.models import TransactionCreate, TransactionPublic, TransactionsPublic
 
 router = APIRouter(prefix="/transactions", tags=["transactions"])
 

backend/app/api/routes/wallets.py

@@ -7,11 +7,7 @@
 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,
-)
+from app.models import WalletCreate, WalletPublic, WalletsPublic
 
 router = APIRouter(prefix="/wallets", tags=["wallets"])
 
@@ -53,11 +49,11 @@ def read_wallet(
     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)