Moving requests to resource
This commit is contained in:
@@ -9,6 +9,8 @@ from fastapi_pagination.ext.sqlmodel import paginate
|
||||
|
||||
from account.schemas import AccountCreate, AccountRead, AccountUpdate
|
||||
from account.models import Account
|
||||
from account.resource import AccountResource
|
||||
|
||||
from db import SessionDep
|
||||
from user.manager import get_current_user
|
||||
|
||||
@@ -25,49 +27,49 @@ router = APIRouter()
|
||||
|
||||
@router.post("")
|
||||
def create_account(account: AccountCreate, session: SessionDep, current_user=Depends(get_current_user)) -> AccountRead:
|
||||
result = Account.create(account, session)
|
||||
result = AccountResource.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))
|
||||
return paginate(session, AccountResource.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))
|
||||
return paginate(session, AccountResource.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))
|
||||
return paginate(session, AccountResource.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)
|
||||
account = AccountResource.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)
|
||||
db_account = AccountResource.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)
|
||||
account = AccountResource.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)
|
||||
account = AccountResource.get(session, account_id)
|
||||
if not account:
|
||||
raise HTTPException(status_code=404, detail="Account not found")
|
||||
|
||||
Account.delete(session, account)
|
||||
AccountResource.delete(session, account)
|
||||
return {"ok": True}
|
||||
|
||||
Reference in New Issue
Block a user