74 lines
2.8 KiB
Python
74 lines
2.8 KiB
Python
from typing import Optional
|
|
from uuid import UUID
|
|
|
|
from fastapi import APIRouter, HTTPException, Depends
|
|
from fastapi_filter import FilterDepends
|
|
from fastapi_filter.contrib.sqlalchemy import Filter
|
|
from fastapi_pagination import Page
|
|
from fastapi_pagination.ext.sqlmodel import paginate
|
|
|
|
from account.schemas import AccountCreate, AccountRead, AccountUpdate
|
|
from account.models import Account
|
|
from db import SessionDep
|
|
from user.manager import get_current_user
|
|
|
|
|
|
class AccountFilters(Filter):
|
|
name__like: Optional[str] = None
|
|
order_by: Optional[list[str]] = None
|
|
|
|
class Constants(Filter.Constants):
|
|
model = Account
|
|
search_model_fields = ["name"]
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("")
|
|
def create_account(account: AccountCreate, session: SessionDep, current_user=Depends(get_current_user)) -> AccountRead:
|
|
result = Account.create(account, session)
|
|
return result
|
|
|
|
@router.get("")
|
|
def read_accounts(session: SessionDep,
|
|
filters: AccountFilters = FilterDepends(AccountFilters),
|
|
current_user=Depends(get_current_user)) -> Page[AccountRead]:
|
|
return paginate(session, Account.list_accounts(filters))
|
|
|
|
@router.get("/assets")
|
|
def read_assets(session: SessionDep,
|
|
filters: AccountFilters = FilterDepends(AccountFilters),
|
|
current_user=Depends(get_current_user)) -> Page[AccountRead]:
|
|
return paginate(session, Account.list_assets(filters))
|
|
|
|
@router.get("/liabilities")
|
|
def read_liabilities(session: SessionDep,
|
|
filters: AccountFilters = FilterDepends(AccountFilters),
|
|
current_user=Depends(get_current_user)) -> Page[AccountRead]:
|
|
return paginate(session, Account.list_liabilities(filters))
|
|
|
|
@router.get("/{account_id}")
|
|
def read_account(account_id: UUID, session: SessionDep, current_user=Depends(get_current_user)) -> AccountRead:
|
|
account = Account.get(session, account_id)
|
|
if not account:
|
|
raise HTTPException(status_code=404, detail="Account not found")
|
|
return account
|
|
|
|
@router.put("/{account_id}")
|
|
def update_account(account_id: UUID, account: AccountUpdate, session: SessionDep, current_user=Depends(get_current_user)) -> AccountRead:
|
|
db_account = Account.get(session, account_id)
|
|
if not db_account:
|
|
raise HTTPException(status_code=404, detail="Account not found")
|
|
|
|
account_data = account.model_dump(exclude_unset=True)
|
|
account = Account.update(session, db_account, account_data)
|
|
return account
|
|
|
|
@router.delete("/{account_id}")
|
|
def delete_account(account_id: UUID, session: SessionDep, current_user=Depends(get_current_user)):
|
|
account = Account.get(session, account_id)
|
|
if not account:
|
|
raise HTTPException(status_code=404, detail="Account not found")
|
|
|
|
Account.delete(session, account)
|
|
return {"ok": True}
|