feat: transactions router

Dockerfile

@@ -25,4 +25,7 @@ COPY . .
 # Make port 8000 available to the world outside this container
 EXPOSE 8000
 
+# Run migrations
+RUN alembic upgrade head
+
 CMD ["python", "main.py"]

app/api/routers/transaction.py

@@ -0,0 +1,28 @@
+from typing import List
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlalchemy.ext.asyncio import AsyncSession
+from app.model.transaction import Transaction as TransactionModel
+from app.schema.index import TransactionOutput
+from app.engine.postgresdb import get_db_session
+
+transaction_router = r = APIRouter(prefix="/api/v1/transactions", tags=["transactions"])
+
+
+@r.get(
+    "/{user_id}",
+    response_model=List[TransactionOutput],
+    responses={
+        200: {"description": "New user created"},
+        400: {"description": "Bad request"},
+        204: {"description": "No content"},
+        500: {"description": "Internal server error"},
+    },
+)
+async def get_transactions(user_id: int, db: AsyncSession = Depends(get_db_session)):
+    """
+    Retrieve all transactions.
+    """
+    result = await TransactionModel.get_by_user(db, user_id)
+    if len(result) == 0:
+        raise HTTPException(status_code=status.HTTP_204_NO_CONTENT)
+    return result.scalars().all()

app/model/transaction.py

@@ -1,6 +1,9 @@
 from datetime import datetime
+from typing import List
 from sqlalchemy import ForeignKey
 from sqlalchemy.orm import relationship, Mapped, mapped_column
+from sqlalchemy.sql import expression as sql
+from sqlalchemy.ext.asyncio import AsyncSession
 
 from app.model.base import BaseModel
 from app.engine.postgresdb import Base
@@ -16,3 +19,29 @@ class Transaction(Base, BaseModel):
 
     user_id = mapped_column(ForeignKey("users.id"))
     user = relationship("User", back_populates="transactions")
+
+    @classmethod
+    async def create(cls: "type[Transaction]", db: AsyncSession, **kwargs) -> "Transaction":
+        query = sql.insert(cls).values(**kwargs).returning(cls.id)
+        transactions = await db.execute(query)
+        await db.commit()
+        return transactions.first()
+
+    @classmethod
+    async def update(cls: "type[Transaction]", db: AsyncSession, id: int, **kwargs) -> "Transaction":
+        query = (
+            sql.update(cls)
+            .where(cls.id == id)
+            .values(**kwargs)
+            .execution_options(synchronize_session="fetch")
+            .returning(cls.id)
+        )
+        transactions = await db.execute(query)
+        await db.commit()
+        return transactions.first()
+
+    @classmethod
+    async def get_by_user(cls: "type[Transaction]", db: AsyncSession, user_id: int) -> "List[Transaction]":
+        query = sql.select(cls).where(cls.user_id == user_id)
+        transactions = await db.execute(query)
+        return transactions

app/schema/index.py

@@ -24,10 +24,13 @@ class User(BaseModel):
     transactions: "List[Transaction]" = []
 
 
-class Transaction(BaseModel):
+class TransactionOutput(BaseModel):
     transaction_date: datetime
     category: str
     name_description: str
     amount: float
     type: TransactionType
+
+
+class Transaction(TransactionOutput):
     user: User

main.py

@@ -7,6 +7,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import RedirectResponse
 from app.api.routers.chat import chat_router
 from app.api.routers.user import user_router
+from app.api.routers.transaction import transaction_router
 from app.settings import init_settings
 from fastapi.staticfiles import StaticFiles
 from alembic.config import Config
@@ -61,6 +62,7 @@ def init_app():
         app.mount("/api/data", StaticFiles(directory="data"), name="static")
     app.include_router(chat_router, prefix="/api/chat")
     app.include_router(user_router)
+    app.include_router(transaction_router)
 
     return app
 

migration/env.py

@@ -22,7 +22,8 @@ if config.config_file_name is not None:
     fileConfig(config.config_file_name)
 
 # Import all models so they're registered with SQLAlchemy.
-from app.model import *
+import app.model.transaction
+import app.model.user
 
 # add your model's MetaData object here
 # for 'autogenerate' support

migration/versions/8feaedca36f9_users_datatype_updates.py

@@ -0,0 +1,74 @@
+"""Remove_Transactions_From_Users
+
+Revision ID: 8feaedca36f9
+Revises: cd515c44401d
+Create Date: 2024-06-02 01:26:54.731002
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision: str = '8feaedca36f9'
+down_revision: Union[str, None] = 'cd515c44401d'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('transactions', 'transaction_date',
+               existing_type=postgresql.TIMESTAMP(),
+               nullable=False)
+    op.alter_column('transactions', 'category',
+               existing_type=sa.VARCHAR(),
+               nullable=False)
+    op.alter_column('transactions', 'name_description',
+               existing_type=sa.VARCHAR(),
+               nullable=False)
+    op.alter_column('transactions', 'amount',
+               existing_type=sa.DOUBLE_PRECISION(precision=53),
+               nullable=False)
+    op.alter_column('transactions', 'type',
+               existing_type=sa.VARCHAR(),
+               nullable=False)
+    op.add_column('users', sa.Column('hashed_password', sa.String(), nullable=False))
+    op.add_column('users', sa.Column('is_deleted', sa.Boolean(), nullable=False))
+    op.alter_column('users', 'name',
+               existing_type=sa.VARCHAR(),
+               nullable=False)
+    op.alter_column('users', 'email',
+               existing_type=sa.VARCHAR(),
+               nullable=False)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('users', 'email',
+               existing_type=sa.VARCHAR(),
+               nullable=True)
+    op.alter_column('users', 'name',
+               existing_type=sa.VARCHAR(),
+               nullable=True)
+    op.drop_column('users', 'is_deleted')
+    op.drop_column('users', 'hashed_password')
+    op.alter_column('transactions', 'type',
+               existing_type=sa.VARCHAR(),
+               nullable=True)
+    op.alter_column('transactions', 'amount',
+               existing_type=sa.DOUBLE_PRECISION(precision=53),
+               nullable=True)
+    op.alter_column('transactions', 'name_description',
+               existing_type=sa.VARCHAR(),
+               nullable=True)
+    op.alter_column('transactions', 'category',
+               existing_type=sa.VARCHAR(),
+               nullable=True)
+    op.alter_column('transactions', 'transaction_date',
+               existing_type=postgresql.TIMESTAMP(),
+               nullable=True)
+    # ### end Alembic commands ###