import os
from dotenv import load_dotenv
import requests

load_dotenv()

OKAPI_POSTE_KEY = os.getenv('OKAPI_POSTE_KEY')
DHL_KEY = os.getenv('DHL_KEY')
DHL_SECRET = os.getenv('DHL_SECRET')

POSTE_ENDPOINT = 'https://api.laposte.fr/suivi/v2/idships/'
DHL_ENDPOINT = 'https://api-eu.dhl.com/track/shipments'

class Tracking:
    def __init__(self, tracking_num):
        self.tracking_num = tracking_num
        return
    
    def track(self):
        pass

    def get_history(self):
        pass

class Poste_tracking(Tracking):
    def track(self):
        auth_headers = {'X-Okapi-Key' : OKAPI_POSTE_KEY, 'Accept' : 'application/json'}
        response = requests.get(POSTE_ENDPOINT + self.tracking_num, headers = auth_headers)
        print(response.json())

class DHL_tracking(Tracking):
    def track(self):
        auth_headers = {'DHL-API-Key' : DHL_KEY}
        query = {'trackingNumber' : self.tracking_num}
        response = requests.get(DHL_ENDPOINT, params = query, headers = auth_headers)
        print(response.json())

Ptrack = Poste_tracking("CC550445947DE")
Ptrack.track()
print(',')

Chronotrack = Poste_tracking("XA420959322FR")
Chronotrack.track()
print(',')

DHLtrack = DHL_tracking("CC550445947DE")
DHLtrack.track()