chore: Run black on code
This commit is contained in:
parent
9374b1284e
commit
1d90de9205
4 changed files with 29 additions and 20 deletions
|
@ -141,7 +141,6 @@ class Tracker:
|
|||
self.db.write_event(event)
|
||||
self.notify_event(shipment, event, event == events[0])
|
||||
|
||||
|
||||
def start_loop(self) -> Never:
|
||||
logging.debug("Starting loop")
|
||||
|
||||
|
@ -159,7 +158,10 @@ class Tracker:
|
|||
while True:
|
||||
tasks = []
|
||||
for shipment in self.db.get_shipments():
|
||||
task = asyncio.wait_for(asyncio.to_thread(self.process_shipment, shipment), timeout=self.loop_timeout)
|
||||
task = asyncio.wait_for(
|
||||
asyncio.to_thread(self.process_shipment, shipment),
|
||||
timeout=self.loop_timeout,
|
||||
)
|
||||
tasks.append(task)
|
||||
|
||||
try:
|
||||
|
|
|
@ -15,19 +15,23 @@ class KeyDelivery(BaseTracker):
|
|||
all_events = self.api.realtime(carrier, tracking_number)
|
||||
|
||||
try:
|
||||
logging.debug(f"Got events for {tracking_number}: {len(all_events['data']['items'])}")
|
||||
logging.debug(
|
||||
f"Got events for {tracking_number}: {len(all_events['data']['items'])}"
|
||||
)
|
||||
except KeyError:
|
||||
print(f"Error getting events for {tracking_number}: {all_events}")
|
||||
return
|
||||
|
||||
events = sorted(all_events["data"]["items"], key=lambda x: x["time"], reverse=True)
|
||||
events = sorted(
|
||||
all_events["data"]["items"], key=lambda x: x["time"], reverse=True
|
||||
)
|
||||
|
||||
for event in events:
|
||||
yield Event(
|
||||
shipment_id=0,
|
||||
event_time=event["time"],
|
||||
event_description=event["context"],
|
||||
raw_event = json.dumps(event)
|
||||
raw_event=json.dumps(event),
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
|
@ -36,4 +40,5 @@ class KeyDelivery(BaseTracker):
|
|||
("*", 1),
|
||||
]
|
||||
|
||||
|
||||
tracker = KeyDelivery
|
|
@ -6,6 +6,7 @@ import json
|
|||
from dateutil.parser import parse
|
||||
from postat.classes.api import PostAPI
|
||||
|
||||
|
||||
class PostAT(BaseTracker):
|
||||
def __init__(self, *args, **kwargs):
|
||||
pass
|
||||
|
@ -24,7 +25,7 @@ class PostAT(BaseTracker):
|
|||
shipment_id=0,
|
||||
event_time=event_time,
|
||||
event_description=event["text"],
|
||||
raw_event = json.dumps(event)
|
||||
raw_event=json.dumps(event),
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
|
@ -33,4 +34,5 @@ class PostAT(BaseTracker):
|
|||
("austrian_post", 100),
|
||||
]
|
||||
|
||||
|
||||
tracker = PostAT
|
Loading…
Reference in a new issue