47 lines
2.0 KiB
Python
47 lines
2.0 KiB
Python
from uuid import UUID
|
|
|
|
from fastapi import APIRouter, HTTPException, Depends
|
|
from fastapi_pagination import Page
|
|
from fastapi_pagination.ext.sqlmodel import paginate
|
|
|
|
from transaction.models import Transaction, TransactionCreate, TransactionRead, TransactionUpdate
|
|
from db import SessionDep
|
|
from user.manager import get_current_user
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("")
|
|
def create_transaction(transaction: TransactionCreate, session: SessionDep, current_user=Depends(get_current_user)) -> TransactionRead:
|
|
result = Transaction.create(transaction, session)
|
|
return result
|
|
|
|
@router.get("")
|
|
def read_transactions(session: SessionDep, current_user=Depends(get_current_user)) -> Page[TransactionRead]:
|
|
return paginate(session, Transaction.list())
|
|
|
|
@router.get("/{transaction_id}")
|
|
def read_transaction(transaction_id: UUID, session: SessionDep, current_user=Depends(get_current_user)) -> TransactionRead:
|
|
transaction = Transaction.get(session, transaction_id)
|
|
if not transaction:
|
|
raise HTTPException(status_code=404, detail="Transaction not found")
|
|
return transaction
|
|
|
|
@router.put("/{transaction_id}")
|
|
def update_transaction(transaction_id: UUID, transaction: TransactionUpdate, session: SessionDep, current_user=Depends(get_current_user)) -> TransactionRead:
|
|
db_transaction = Transaction.get(session, transaction_id)
|
|
if not db_transaction:
|
|
raise HTTPException(status_code=404, detail="Transaction not found")
|
|
|
|
transaction_data = transaction.model_dump(exclude_unset=True)
|
|
transaction = Transaction.update(session, db_transaction, transaction_data)
|
|
return transaction
|
|
|
|
@router.delete("/{transaction_id}")
|
|
def delete_transaction(transaction_id: UUID, session: SessionDep, current_user=Depends(get_current_user)):
|
|
transaction = Transaction.get(session, transaction_id)
|
|
if not transaction:
|
|
raise HTTPException(status_code=404, detail="Transaction not found")
|
|
|
|
Transaction.delete(session, transaction)
|
|
return {"ok": True}
|