diff options
Diffstat (limited to 'modules/tracking.py')
| -rw-r--r-- | modules/tracking.py | 44 |
1 files changed, 41 insertions, 3 deletions
diff --git a/modules/tracking.py b/modules/tracking.py index 1d84fce..20b6ae5 100644 --- a/modules/tracking.py +++ b/modules/tracking.py @@ -6,7 +6,7 @@ info = { cfg_section = 'module/tracking' -import urllib2, datetime, re +import urllib2, datetime, re, pytz, json, urllib from xml.etree import ElementTree as ET from twisted.internet import reactor from twisted.internet.task import LoopingCall @@ -100,13 +100,16 @@ class TrackingModuleMeta(type): class TrackingModule(object): __metaclass__ = TrackingModuleMeta + def splitcode(self, code): + return code_split(code)[-1] + class PostenModule(TrackingModule): name = 'posten' url = 'http://sporing.posten.no/sporing.xml?q=%s' def get_xml(self, url): try: - u = urllib2.urlopen(url, timeout = config.getfloat('module/tracking', 'timeout')) + u = urllib2.urlopen(url, timeout = config.getfloat(cfg_section, 'timeout')) except Exception as e: raise PackageError(str(e)) if u.headers['content-type'].startswith('application/xml'): @@ -123,6 +126,7 @@ class PostenModule(TrackingModule): return url def track(self, code): + code = self.splitcode(code) xml = self.get_xml(self.url % code) if not xml: return @@ -155,6 +159,40 @@ class PostenModule(TrackingModule): results.append(TrackingResult(code, isodate, desc, status == 'DELIVERED', previous_code = previous_code)) return results +class FedexModule(TrackingModule): + name = 'fedex' + + def get_url(self, code = None): + url = 'https://www.fedex.com/fedextrack/index.html' + if code: + url += '?tracknumbers=' + code + return url + + def track(self, code): + code = self.splitcode(code) + data = {'TrackPackagesRequest': {'appType': 'wtrk', 'uniqueKey': '', 'processingParameters': {'anonymousTransaction': True, 'clientId': 'WTRK', 'returnDetailedErrors': True, 'returnLocalizedDateTime': False}, 'trackingInfoList': [{'trackNumberInfo': {'trackingNumber': code, 'trackingQualifier': '', 'trackingCarrier': ''}}]}} + url = 'https://www.fedex.com/trackingCal/track?' + urllib.urlencode({'data': json.dumps(data), 'action': 'trackpackages', 'locale': 'en_US', 'format': 'json', 'version': 99}) + u = urllib2.urlopen(url) + data = u.read() + u.close() + for m in re.findall(r'(\\x[a-fA-F\d]{2})', data): + data = data.replace(m, chr(int(m[-2:], 16))) + data = json.loads(data) + response = data['TrackPackagesResponse'] + packages = response['packageList'] + results = [] + for package in packages: + lastevent = package['scanEventList'][0] + date = datetime.datetime.strptime('%s %s' % (lastevent['date'], lastevent['time']), '%Y-%m-%d %H:%M:%S') + if lastevent['gmtOffset']: + tz = pytz.timezone('Etc/GMT' + ('+' if lastevent['gmtOffset'][0] == '-' else '-') + str(int(lastevent['gmtOffset'][1:3]))) + date = tz.localize(date).astimezone(pytz.timezone(config.get(cfg_section, 'local_timezone'))).replace(tzinfo = None) + status = lastevent['status'].encode('utf-8') + if lastevent['scanLocation']: + status += ' (%s)' % lastevent['scanLocation'].encode('utf-8') + results.append(TrackingResult(code, date, status, lastevent['isDelivered'])) + return results + def code_split(code): if ':' in code: return code.split(':') @@ -343,7 +381,7 @@ class Module: results = [] for consignment in consignments: for package in consignment.packages: - m = get_tracking_module(package.code) + m = get_tracking_module('%s:%s' % (consignment.type, package.code)) msg.append(m.get_url(package.code)) if not len(msg): # Rreturn an error only if we got a code to filter with. |