From ab325de5a6391c63f86343b4f4eab21c655a4616 Mon Sep 17 00:00:00 2001
From: Kumi <git@kumi.email>
Date: Thu, 17 Aug 2023 12:50:49 +0200
Subject: [PATCH] Current status

---
 .gitignore       |   5 ++
 requirements.txt |   1 +
 trackbert.py     | 121 +++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 127 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 requirements.txt
 create mode 100644 trackbert.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..f7df281
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,5 @@
+*.pyc
+__pycache__/
+venv/
+config.ini
+trackbert.db
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..74b4fa0
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+git+https://kumig.it/kumitterer/pykeydelivery.git
\ No newline at end of file
diff --git a/trackbert.py b/trackbert.py
new file mode 100644
index 0000000..c0c9dc5
--- /dev/null
+++ b/trackbert.py
@@ -0,0 +1,121 @@
+from pykeydelivery import KeyDelivery
+
+import sqlite3
+import json
+import time
+import subprocess
+
+
+def notify(title, message):
+    subprocess.run(
+        [
+            "notify-send",
+            title,
+            message,
+        ]
+    )
+
+
+def create_shipment(db, tracking_number: str, carrier: str, description=""):
+    db.execute(
+        "INSERT INTO shipments (tracking_number, carrier, description) VALUES (?, ?, ?)",
+        (tracking_number, carrier, description),
+    )
+    db.commit()
+
+
+def get_shipment(db: sqlite3.Connection, tracking_number: str):
+    cur = db.cursor()
+    cur.execute("SELECT * FROM shipments WHERE tracking_number = ?", (tracking_number,))
+    return cur.fetchone()
+
+
+def get_shipments(db: sqlite3.Connection):
+    cur = db.cursor()
+    cur.execute("SELECT * FROM shipments")
+    return cur.fetchall()
+
+
+def get_shipment_events(db, shipment_id):
+    cur = db.cursor()
+    cur.execute("SELECT * FROM events WHERE shipment_id = ?", (shipment_id,))
+    return cur.fetchall()
+
+
+def create_event(
+    db,
+    shipment_id,
+    event_time,
+    event_description,
+    raw_event,
+):
+    db.execute(
+        "INSERT INTO events (shipment_id, event_time, event_description, raw_event) VALUES (?, ?, ?, ?)",
+        (
+            shipment_id,
+            event_time,
+            event_description,
+            json.dumps(raw_event),
+        ),
+    )
+    db.commit()
+
+
+def get_latest_event(db, shipment_id):
+    cur = db.cursor()
+    cur.execute(
+        "SELECT * FROM events WHERE shipment_id = ? ORDER BY event_time DESC LIMIT 1",
+        (shipment_id,),
+    )
+    return cur.fetchone()
+
+
+def initialize_db(db):
+    db.execute(
+        "CREATE TABLE IF NOT EXISTS shipments (id INTEGER PRIMARY KEY AUTOINCREMENT, tracking_number TEXT, carrier TEXT, description TEXT)"
+    )
+    db.execute(
+        "CREATE TABLE IF NOT EXISTS events (id INTEGER PRIMARY KEY AUTOINCREMENT, shipment_id INTEGER, event_time TEXT, event_description TEXT, raw_event TEXT, FOREIGN KEY(shipment_id) REFERENCES shipments(id))"
+    )
+    db.commit()
+
+
+def get_db():
+    db = sqlite3.connect("trackbert.db")
+    initialize_db(db)
+    return db
+
+
+def start_loop(db, api: KeyDelivery):
+    while True:
+        for shipment in get_shipments(db):
+            shipment_id = shipment[0]
+            tracking_number = shipment[1]
+            carrier = shipment[2]
+            description = shipment[3]
+            latest_known_event = get_latest_event(db, shipment_id)
+            all_events = api.realtime(carrier, tracking_number)
+            for event in all_events["data"]["items"]:
+                if latest_known_event is None or event["time"] > latest_known_event[3]:
+                    create_event(
+                        db,
+                        shipment_id,
+                        event["time"],
+                        event["context"],
+                        event,
+                    )
+                    print(f"New event for {tracking_number}: {event['context']}")
+                    notify(f"New event for {description}", event["context"])
+
+        time.sleep(300)
+
+
+def main():
+    db = get_db()
+    api = KeyDelivery.from_config("config.ini")
+    notify("Trackbert", "Starting up")
+    start_loop(db, api)
+
+
+if __name__ == "__main__":
+    main()