def get_order(order_id: UUID):
try:
with UnitOfWork() as unit_of_work:
repo = OrdersRepository(unit_of_work.session)
orders_service = OrdersService(repo)
order = orders_service.get_order(order_id=order_id)
return order.dict()
except OrderNotFoundError:
raise HTTPException(status_code=404,
detail=f'Order with ID {order_id} not found')
def get_orders(request: Request,
cancelled: Optional[bool] = None,
limit: Optional[int] = None):
with UnitOfWork() as unit_of_work:
repo = OrdersRepository(unit_of_work.session)
orders_service = OrdersService(repo)
results = orders_service.list_orders(limit=limit,
cancelled=cancelled,
user_id=request.state.user_id)
return {"orders": [result.dict() for result in results]}
def create_order(payload: CreateOrderSchema):
with UnitOfWork() as unit_of_work:
repo = OrdersRepository(unit_of_work.session)
orders_service = OrdersService(repo)
order = payload.dict()['order']
for item in order:
item['size'] = item['size'].value
order = orders_service.place_order(order)
unit_of_work.commit()
return_payload = order.dict()
return return_payload
def create_order(request: Request, payload: CreateOrderSchema):
with UnitOfWork() as unit_of_work:
repo = OrdersRepository(unit_of_work.session)
orders_service = OrdersService(repo)
order = payload.dict()["order"]
for item in order:
item["size"] = item["size"].value
order = orders_service.place_order(order, request.state.user_id)
unit_of_work.commit()
return_payload = order.dict()
return return_payload
def cancel_order(request: Request, order_id: UUID):
try:
with UnitOfWork() as unit_of_work:
repo = OrdersRepository(unit_of_work.session)
orders_service = OrdersService(repo)
order = orders_service.cancel_order(order_id=order_id,
user_id=request.state.user_id)
unit_of_work.commit()
return order.dict()
except OrderNotFoundError:
raise HTTPException(status_code=404,
detail=f"Order with ID {order_id} not found")
def update_order(order_id: UUID, order_details: CreateOrderSchema):
try:
with UnitOfWork() as unit_of_work:
repo = OrdersRepository(unit_of_work.session)
orders_service = OrdersService(repo)
order = order_details.dict()['order']
for item in order:
item['size'] = item['size'].value
order = orders_service.update_order(order_id=order_id, items=order)
unit_of_work.commit()
return order.dict()
except OrderNotFoundError:
raise HTTPException(status_code=404,
detail=f'Order with ID {order_id} not found')
def get_orders(cancelled: Optional[bool] = None, limit: Optional[int] = None):
with UnitOfWork() as unit_of_work:
repo = OrdersRepository(unit_of_work.session)
orders_service = OrdersService(repo)
results = orders_service.list_orders(limit=limit, cancelled=cancelled)
return {'orders': [result.dict() for result in results]}