1717from kayman .logics .account import update_balances_with_transactions
1818from kayman .logics .payment import validate_total
1919from kayman .schemas .api_models import PaymentCreateDetailed , PaymentReadDetailed
20- from kayman .schemas .payment import (
21- Payment ,
22- PaymentBase ,
20+ from kayman .schemas .event import (
21+ Event ,
22+ EventBase ,
23+ EventRead ,
2324 PaymentEntryBase ,
24- PaymentRead ,
2525)
2626from kayman .schemas .transaction import TransactionBase
2727
@@ -257,7 +257,7 @@ def create(
257257 * ,
258258 session : Session = Depends (get_session ),
259259 body : PaymentCreateDetailed = Body (openapi_examples = EXAMPLES ["create" ]),
260- ) -> PaymentBase :
260+ ) -> EventBase :
261261 # Validate total
262262 try :
263263 validate_total (body )
@@ -266,7 +266,7 @@ def create(
266266
267267 # Store payment
268268 db_payment = create_payment (session , body .payment , commit = False )
269- payment_id = PaymentRead .model_validate (db_payment ).id
269+ payment_id = EventRead .model_validate (db_payment ).id
270270
271271 # Store entries
272272 entries = []
@@ -313,7 +313,7 @@ def create(
313313@payment_router .get (
314314 "/{payment_id}" , name = "Read Payment" , response_model = PaymentReadDetailed
315315)
316- def read (* , session : Session = Depends (get_session ), payment_id : int ) -> PaymentBase :
316+ def read (* , session : Session = Depends (get_session ), payment_id : int ) -> EventBase :
317317 payment = read_payment (session , payment_id )
318318 if payment is None :
319319 raise HTTPException (status_code = 404 , detail = "Payment not found" )
@@ -326,12 +326,12 @@ def reads(
326326 session : Session = Depends (get_session ),
327327 payment_date : date | None = None ,
328328 category_id : int | None = None ,
329- ) -> Sequence [PaymentBase ]:
329+ ) -> Sequence [EventBase ]:
330330 return read_payments (session , payment_date , category_id )
331331
332332
333- @payment_router .patch ("" , name = "Update Payment" , response_model = PaymentRead )
334- def update (* , session : Session = Depends (get_session ), payment : Payment ) -> PaymentBase :
333+ @payment_router .patch ("" , name = "Update Payment" , response_model = EventRead )
334+ def update (* , session : Session = Depends (get_session ), payment : Event ) -> EventBase :
335335 session .merge (payment )
336336 session .commit ()
337337 session .refresh (payment )
@@ -340,7 +340,7 @@ def update(*, session: Session = Depends(get_session), payment: Payment) -> Paym
340340
341341@payment_router .delete ("/{id}" , name = "Delete Payment" )
342342def delete (* , session : Session = Depends (get_session ), id : int ) -> None :
343- payment = session .get (Payment , id )
343+ payment = session .get (Event , id )
344344 if payment is None :
345345 raise HTTPException (status_code = 404 , detail = "Payment not found" )
346346 session .delete (payment )
0 commit comments