Update app.py

app.py

@@ -3,9 +3,11 @@ import logging
 import json
 import base64
 import uuid
+import os
 from datetime import datetime
 from flask import Flask, jsonify, request
 from threading import Thread
+import time
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -18,6 +20,9 @@ UPS_CLIENT_SECRET = 'DBAV77mzTWgI5oqWwXlA12hi62hgyUkesoGsREZsmjd1WHMSugpDUuBGTjy
 FEDEX_API_KEY = 'l705da56bf34254c3495f351772a6ea2d3'
 FEDEX_SECRET_KEY = 'ae27688af74341f094228fc67a36a8cc'
 
+TRACKING_FILE = 'trackings.json'
+UPDATE_INTERVAL = 60 * 1  # 10 minutes
+
 def get_fedex_access_token():
     url = "https://apis.fedex.com/oauth/token"
     headers = {
@@ -233,6 +238,43 @@ def fetch_fedex_tracking_info(tracking_number):
         logger.error(f"Response content: {data}")
         raise
 
+def load_trackings():
+    if os.path.exists(TRACKING_FILE):
+        with open(TRACKING_FILE, 'r') as file:
+            return json.load(file)
+    return {}
+
+def save_trackings(trackings):
+    with open(TRACKING_FILE, 'w') as file:
+        json.dump(trackings, file, indent=4)
+
+def update_tracking_info():
+    while True:
+        try:
+            logger.info("Starting periodic tracking update")
+            trackings = load_trackings()
+            updated_trackings = {}
+
+            for tracking_number, tracking_info in trackings.items():
+                logger.info(f"Updating tracking info for {tracking_number}")
+                if tracking_info['courier'] == 'UPS':
+                    updated_info = fetch_ups_tracking_info(tracking_number)
+                elif tracking_info['courier'] == 'FedEx':
+                    updated_info = fetch_fedex_tracking_info(tracking_number)
+                else:
+                    logger.warning(f"Unknown courier for tracking number {tracking_number}: {tracking_info['courier']}")
+                    continue
+
+                updated_info['tracking'] = tracking_number
+                updated_trackings[tracking_number] = updated_info
+
+            save_trackings(updated_trackings)
+            logger.info("Tracking update completed")
+        except Exception as e:
+            logger.error(f"Error during periodic tracking update: {e}")
+
+        time.sleep(UPDATE_INTERVAL)
+
 @app.route('/')
 def home():
     return "Tracking update service is running."
@@ -262,4 +304,8 @@ def get_fedex_tracking():
         return jsonify({"status": "error", "message": str(e)}), 500
 
 if __name__ == '__main__':
+    # Start the background thread for periodic updates
+    update_thread = Thread(target=update_tracking_info, daemon=True)
+    update_thread.start()
+
     app.run(host='0.0.0.0', port=7860)