Skip to content

Commit

Permalink
add exception handling for order creation
Browse files Browse the repository at this point in the history
  • Loading branch information
Der-Henning committed Apr 2, 2023
1 parent a9ec2da commit 2f4834a
Showing 1 changed file with 15 additions and 8 deletions.
23 changes: 15 additions & 8 deletions src/models/reservations.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,12 @@
import logging
from dataclasses import dataclass
from typing import Callable, Dict, List

from models.item import Item
from tgtg import TgtgClient

log = logging.getLogger("tgtg")


@dataclass
class Order():
Expand Down Expand Up @@ -33,14 +36,18 @@ def reserve(self, item_id: str,
Reservation(item_id, amount, display_name))

def _create_order(self, reservation: Reservation) -> None:
res = self.client.create_order(reservation.item_id, reservation.amount)
order_id = res.get("id")
order = Order(order_id,
reservation.item_id,
reservation.amount,
reservation.display_name)
self.active_orders.append(order)
self.reservation_query.remove(reservation)
try:
res = self.client.create_order(
reservation.item_id, reservation.amount)
order_id = res.get("id")
order = Order(order_id,
reservation.item_id,
reservation.amount,
reservation.display_name)
self.active_orders.append(order)
self.reservation_query.remove(reservation)
except Exception as exc:
log.error("Create Order Error: %s", exc)

def make_orders(self, state: Dict[str, Item],
callback: Callable[[Reservation], None]) -> None:
Expand Down

0 comments on commit 2f4834a

Please sign in to comment.