Parcels prototype

package_tracking/__main__.py

@@ -0,0 +1,58 @@
+import requests
+import time
+import logging
+import yaml
+from . import secrets
+
+logger = logging.getLogger(__name__)
+
+URL_TRACKING = 'https://parcelsapp.com/api/v3/shipments/tracking'
+
+target_country = 'Denmark'
+
+TRACKING_STATUS_CHECKING_INTERVAL = 1
+
+def request_json(method: str, url: str, **kwargs) -> dict:
+    response = requests.request(method=method,url=URL_TRACKING, json={'apiKey': secrets.PARCELS_API_KEY, **kwargs})
+    print(response)
+    response.raise_for_status()
+    json_data = response.json()
+    if 'error' in json_data:
+        msg = 'Error from endpoint: {}'.format(json_data['error'])
+        raise RuntimeError(msg)
+    return json_data
+
+def check_tracking_status(uuid: str) -> dict:
+    """ Function to check tracking status with UUID"""
+    json_data = request_json('GET', URL_TRACKING, uuid=uuid)
+    if json_data['done']:
+        logger.info('Tracking complete')
+        return json_data
+    else:
+        logger.info('Tracking in progress...')
+        time.sleep(TRACKING_STATUS_CHECKING_INTERVAL)
+        return check_tracking_status(uuid)
+
+def get_tracking_status(tracking_ids: list[str]):
+    shipments = [{'trackingId': id, 'language': 'en', 'country': target_country} for id in tracking_ids]
+
+    # Initiate tracking request
+
+    json_data = request_json('POST', URL_TRACKING, shipments=shipments)
+    if json_data.get('done'):
+        return json_data
+    return check_tracking_status(json_data['uuid'])
+
+def main():
+    logging.basicConfig()
+    logger.setLevel('INFO')
+
+    tracking_ids = [
+            '00157128965207138207',
+            '00057151273127784840',
+    ]
+    shipment_statuses = get_tracking_status(tracking_ids)
+    print(yaml.dump(shipment_statuses['shipments']))
+
+if __name__ == '__main__':
+    main()