summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xmodules/tracking.py38
1 files changed, 38 insertions, 0 deletions
diff --git a/modules/tracking.py b/modules/tracking.py
index 0905c7c..9696afe 100755
--- a/modules/tracking.py
+++ b/modules/tracking.py
@@ -16,6 +16,7 @@ from sqlalchemy.orm import sessionmaker, relation, backref
from sqlalchemy.orm.exc import NoResultFound
from sqlalchemy.exc import IntegrityError
from sqlalchemy.sql import or_, and_
+import easypost
engine = Session = None
Base = declarative_base()
@@ -315,6 +316,42 @@ class UpsModule(TrackingModule):
results.append(TrackingResult(code, isodate, desc, statuscode == 'D'))
return results
+class EasypostModule:
+ def track(self, code):
+ code = self.splitcode(code)
+
+ try:
+ tracker = (t for t in easypost.Tracker.all() if t.tracking_code == code).next()
+ except StopIteration:
+ tracker = easypost.Tracker.create(tracking_code = code, carrier = self.easypost_carrier)
+
+ results = []
+
+ if tracker.tracking_details:
+ delivered = tracker.status == 'delivered'
+ lastevent = tracker.tracking_details[-1]
+ date = datetime.datetime.strptime(lastevent.datetime, '%Y-%m-%dT%H:%M:%SZ')
+ status = lastevent.message
+ location = []
+ for f in ('city', 'state', 'country'):
+ if lastevent['tracking_location'][f]:
+ location.append(lastevent['tracking_location'][f])
+ if location:
+ status += ' (%s)' % ', '.join(location)
+ results.append(TrackingResult(code, date, status.encode('utf-8'), delivered))
+
+ return results
+
+class DHLModule(TrackingModule, EasypostModule):
+ name = 'dhl'
+ easypost_carrier = 'DHLExpress'
+
+ def get_url(self, code = None):
+ url = 'http://www.dhl.no/no/express/sporing.html?AWB='
+ if code:
+ url += code
+ return url
+
def code_split(code):
if ':' in code:
return code.split(':')
@@ -331,6 +368,7 @@ def get_tracking_module(arg):
class Module:
def __init__(self, bot):
+ easypost.api_key = config.get(cfg_section, 'easypost_api_key')
self.setup_db()
self.irc = bot
self.tracking = []