added basic functionnality (last event)

master
Dorian Bourgeoisat 2024-05-18 19:45:52 +02:00
parent a9543f0ac8
commit 593dfda7c4
1 changed files with 41 additions and 4 deletions

View File

@ -1,6 +1,7 @@
import os import os
from dotenv import load_dotenv from dotenv import load_dotenv
import requests import requests
import datetime
load_dotenv() load_dotenv()
@ -11,22 +12,54 @@ DHL_SECRET = os.getenv('DHL_SECRET')
POSTE_ENDPOINT = 'https://api.laposte.fr/suivi/v2/idships/' POSTE_ENDPOINT = 'https://api.laposte.fr/suivi/v2/idships/'
DHL_ENDPOINT = 'https://api-eu.dhl.com/track/shipments' DHL_ENDPOINT = 'https://api-eu.dhl.com/track/shipments'
class Event:
def __init__(self, date, label, description="", location=""):
self.date = date
self.label = label
self.description = description
self.location = location
def __lt__(self, other):
return self.date.__lt__(other.date)
def __str__(self):
return "Event<date:" + str(self.date) + "; label:" + self.label + "; desciption:" + self.description + "; location:" + self.location + ">"
def __repr__(self):
return self.__str__()
class Tracking: class Tracking:
def __init__(self, tracking_num): def __init__(self, tracking_num):
self.tracking_num = tracking_num self.tracking_num = tracking_num
self.events = []
self.eta = None
return return
def track(self): def track(self):
pass pass
def get_history(self): def get_history(self):
pass self.events.sort()
return self.events
def get_last_event(self):
self.events.sort()
if len(self.events) == 0:
return None
return self.events[-1]
class Poste_tracking(Tracking): class Poste_tracking(Tracking):
def track(self): def track(self):
auth_headers = {'X-Okapi-Key' : OKAPI_POSTE_KEY, 'Accept' : 'application/json'} auth_headers = {'X-Okapi-Key' : OKAPI_POSTE_KEY, 'Accept' : 'application/json'}
response = requests.get(POSTE_ENDPOINT + self.tracking_num, headers = auth_headers) response = requests.get(POSTE_ENDPOINT + self.tracking_num, headers = auth_headers)
print(response.json()) js = response.json()
if "shipment" in js:
for jev in js["shipment"]["event"]:
self.events.append(Event(datetime.datetime.fromisoformat(jev["date"]), jev["label"]))
if "estimDate" in js["shipment"]:
self.eta = datetime.datetime.fromisoformat(js["shipment"]["estimDate"])
class DHL_tracking(Tracking): class DHL_tracking(Tracking):
def track(self): def track(self):
@ -35,13 +68,17 @@ class DHL_tracking(Tracking):
response = requests.get(DHL_ENDPOINT, params = query, headers = auth_headers) response = requests.get(DHL_ENDPOINT, params = query, headers = auth_headers)
print(response.json()) print(response.json())
#print('[')
Ptrack = Poste_tracking("CC550445947DE") Ptrack = Poste_tracking("CC550445947DE")
Ptrack.track() Ptrack.track()
print(Ptrack.get_last_event())
print(',') print(',')
Chronotrack = Poste_tracking("XA420959322FR") Chronotrack = Poste_tracking("XA420959322FR")
Chronotrack.track() Chronotrack.track()
print(Chronotrack.get_last_event())
print(',') print(',')
DHLtrack = DHL_tracking("CC550445947DE") #DHLtrack = DHL_tracking("CC550445947DE")
DHLtrack.track() #DHLtrack.track()
#print(']')