Jmaa/package-tracking
1
0
Fork 0
Code Actions Packages

Ruff
Some checks failed
Verify Python project can be installed, loaded and have version checked / Test (push) Failing after 22s
Details
Run Python tests (through Pytest) / Test (push) Failing after 24s
Details

Browse Source
This commit is contained in:
Jon Michael Aanes 2025-01-22 17:31:00 +01:00
parent 809e75ad27
commit 13658958e4
4 changed files with 43 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
package_tracking/__main__.py
View File

@ -1,16 +1,17 @@
import logging import logging
from . import parcelsapp from . import http, parcelsapp, secrets
from . import secrets
from . import http
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
def main(): def main():
logging.basicConfig() logging.basicConfig()
logger.setLevel('INFO') logger.setLevel('INFO')
parcelsapp_client: parcelsapp.ParcelsAppClient = parcelsapp.ParcelsAppClient(secrets.PARCELS_API_KEY) parcelsapp_client: parcelsapp.ParcelsAppClient = parcelsapp.ParcelsAppClient(
secrets.PARCELS_API_KEY,
)
http.initialize_server(parcelsapp_client) http.initialize_server(parcelsapp_client)

1
package_tracking/database.py
View File

@ -1,5 +1,6 @@
import dataclasses import dataclasses
@dataclasses.dataclass @dataclasses.dataclass
class TrackingNumberEntry: class TrackingNumberEntry:
number: str number: str

14
package_tracking/http.py
View File

@ -1,10 +1,10 @@
from bottle import route, run, template from bottle import route, run, template
from . import parcelsapp
from . import database from . import database, parcelsapp
PARCELSAPP_CLIENT: parcelsapp.ParcelsAppClient | None = None PARCELSAPP_CLIENT: parcelsapp.ParcelsAppClient | None = None
TEMPLATE = ''' TEMPLATE = """
<doctype HTML> <doctype HTML>
<html> <html>
<head> <head>
@ -96,7 +96,8 @@ a {
</body> </body>
</html> </html>
''' """
@route('/') @route('/')
def index(): def index():
@ -105,13 +106,16 @@ def index():
tracking_results = PARCELSAPP_CLIENT.get_tracking_status(tracking_numbers) tracking_results = PARCELSAPP_CLIENT.get_tracking_status(tracking_numbers)
tracking_results_by_id = {result.tracking_number: result for result in tracking_results} tracking_results_by_id = {
result.tracking_number: result for result in tracking_results
}
derps = [(e, tracking_results_by_id.get(e.number)) for e in tracking_entries] derps = [(e, tracking_results_by_id.get(e.number)) for e in tracking_entries]
derps.sort(key=lambda x: x[1].latest_state().date, reverse=True) derps.sort(key=lambda x: x[1].latest_state().date, reverse=True)
return template(TEMPLATE, tracking_results=derps) return template(TEMPLATE, tracking_results=derps)
def initialize_server(parcelsapp_client: parcelsapp.ParcelsAppClient): def initialize_server(parcelsapp_client: parcelsapp.ParcelsAppClient):
global PARCELSAPP_CLIENT global PARCELSAPP_CLIENT
PARCELSAPP_CLIENT = parcelsapp_client PARCELSAPP_CLIENT = parcelsapp_client

30
package_tracking/parcelsapp.py
View File

@ -1,11 +1,11 @@
import requests
import time
import datetime
import yaml
import dataclasses import dataclasses
import datetime
import logging import logging
import time
from collections.abc import Iterator from collections.abc import Iterator
import requests
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
URL_TRACKING = 'https://parcelsapp.com/api/v3/shipments/tracking' URL_TRACKING = 'https://parcelsapp.com/api/v3/shipments/tracking'
@ -14,6 +14,7 @@ target_country = 'Denmark'
TRACKING_STATUS_CHECKING_INTERVAL = 1 TRACKING_STATUS_CHECKING_INTERVAL = 1
@dataclasses.dataclass(frozen=True) @dataclasses.dataclass(frozen=True)
class ParcelState: class ParcelState:
date: datetime.datetime date: datetime.datetime
@ -34,14 +35,16 @@ class ParcelInfo:
def latest_state(self) -> ParcelState: def latest_state(self) -> ParcelState:
return max(self.states, key=lambda state: state.date) return max(self.states, key=lambda state: state.date)
class ParcelsAppClient:
class ParcelsAppClient:
def __init__(self, api_key: str): def __init__(self, api_key: str):
self.api_key = api_key self.api_key = api_key
def _request_json(self, method: str, url: str, **kwargs) -> dict: def _request_json(self, method: str, url: str, **kwargs) -> dict:
request_json_data = {'apiKey': self.api_key, **kwargs} request_json_data = {'apiKey': self.api_key, **kwargs}
response = requests.request(method=method,url=URL_TRACKING, json=request_json_data) response = requests.request(
method=method, url=URL_TRACKING, json=request_json_data,
)
response.raise_for_status() response.raise_for_status()
json_data = response.json() json_data = response.json()
if 'error' in json_data: if 'error' in json_data:
@ -61,7 +64,10 @@ class ParcelsAppClient:
return self.check_tracking_status(uuid) return self.check_tracking_status(uuid)
def _get_tracking_status_to_json(self, tracking_ids: list[str]) -> dict: def _get_tracking_status_to_json(self, tracking_ids: list[str]) -> dict:
shipments = [{'trackingId': id, 'language': 'en', 'country': target_country} for id in tracking_ids] shipments = [
{'trackingId': id, 'language': 'en', 'country': target_country}
for id in tracking_ids
]
# Initiate tracking request # Initiate tracking request
@ -71,7 +77,6 @@ class ParcelsAppClient:
return self.check_tracking_status(json_data['uuid']) return self.check_tracking_status(json_data['uuid'])
def get_tracking_status(self, tracking_ids: list[str]) -> Iterator[ParcelInfo]: def get_tracking_status(self, tracking_ids: list[str]) -> Iterator[ParcelInfo]:
if len(tracking_ids) == 0: if len(tracking_ids) == 0:
return return
@ -83,7 +88,12 @@ class ParcelsAppClient:
status=parcel_json['status'], status=parcel_json['status'],
destination=parcel_json.get('destination'), destination=parcel_json.get('destination'),
origin=parcel_json.get('origin'), origin=parcel_json.get('origin'),
states = [ParcelState(status=s['status'], states=[
ParcelState(
status=s['status'],
date=datetime.datetime.fromisoformat(s['date']), date=datetime.datetime.fromisoformat(s['date']),
carrier=s['carrier']) for s in parcel_json['states']], carrier=s['carrier'],
)
for s in parcel_json['states']
],
) )