Moving to lxml

1 files changed, 133 insertions, 0 deletions

diff --git a/gsxws/objectify.py b/gsxws/objectify.py
new file mode 100644
index 0000000..d001a15
--- /dev/null
+++ b/gsxws/objectify.py
@@ -0,0 +1,133 @@
+# -*- coding: utf-8 -*-
+
+import os
+import re
+import base64
+import tempfile
+from lxml import objectify
+from lxml.objectify import StringElement
+
+from datetime import datetime
+
+BASE64_TYPES = ('packingList', 'proformaFileData', 'returnLabelFileData',)
+FLOAT_TYPES = ('totalFromOrder', 'exchangePrice', 'stockPrice', 'netPrice',)
+BOOLEAN_TYPES = ('isSerialized', 'popMandatory', 'limitedWarranty', 'partCovered',)
+
+TZMAP = {
+    'GMT': '',        # Greenwich Mean Time
+    'PDT': '-0700',   # Pacific Daylight Time
+    'PST': '-0800',   # Pacific Standard Time
+    'CDT': '-0700',   # Central Daylight Time
+    'CST': '-0600',   # Central Standard Time
+    'EST': '-0500',   # Eastern Standard Time
+    'EDT': '-0400',   # Eastern Daylight Time
+    'CET': '+0100',   # Central European Time
+    'CEST': '+0200',  # Central European Summer Time
+    'IST': '+0530',   # Indian Standard Time
+    'CCT': '+0800',   # Chinese Coast Time
+    'JST': '+0900',   # Japan Standard Time
+    'ACST': '+0930',  # Austrailian Central Standard Time
+    'AEST': '+1000',  # Australian Eastern Standard Time
+    'ACDT': '+1030',  # Australian Central Daylight Time
+    'AEDT': '+1100',  # Australian Eastern Daylight Time
+    'NZST': '+1200',  # New Zealand Standard Time
+}
+
+
+class GsxElement(StringElement):
+    def __str__(self):
+        return str(self.pyval)
+
+
+class GsxDateElement(GsxElement):
+    @property
+    def pyval(self):
+        # looks like some sort of date, let's try to convert
+        try:
+            # standard GSX format: "mm/dd/yy"
+            return datetime.strptime(self.text, "%m/%d/%y").date()
+        except ValueError:
+            pass
+
+        try:
+            # some dates are formatted as "yyyy-mm-dd"
+            return datetime.strptime(self.text, "%Y-%m-%d").date()
+        except (ValueError, TypeError):
+            pass
+
+
+class GsxBooleanElement(GsxElement):
+    @property
+    def pyval(self):
+        return self.text == 'Y'
+
+
+class GsxPriceElement(GsxElement):
+    @property
+    def pyval(self):
+        return float(re.sub(r'[A-Z ,]', '', self.text))
+
+
+class GsxAttachment(GsxElement):
+    @property
+    def pyval(self):
+        v = base64.b64decode(self.text)
+        of = tempfile.NamedTemporaryFile(suffix=".pdf", delete=False)
+        of.write(v)
+        return of.name
+
+
+class GsxDatetimeElement(GsxElement):
+    @property
+    def pyval(self):
+        #2011-01-27 11:45:01 PST
+        # Unfortunately we have to chomp off the TZ info...
+        m = re.search(r'^(\d+\-\d+\-\d+ \d+:\d+:\d+) (\w+)$', self.text)
+        ts, tz = m.groups()
+        return datetime.strptime(ts, "%Y-%m-%d %H:%M:%S")
+
+
+class GsxTimestampElement(GsxElement):
+    @property
+    def pyval(self):
+        return datetime.strptime(self.text, "%d-%b-%y %H:%M:%S")
+
+
+class GsxClassLookup(objectify.ObjectifyElementClassLookup):
+    def lookup(self, node_type, document, namespace, name):
+        if name == 'dispatchSentDate':
+            return GsxDatetimeElement
+        if name == 'acPlusFlag':
+            return GsxBooleanElement
+        if name in BOOLEAN_TYPES:
+            return GsxBooleanElement
+        if name in BASE64_TYPES:
+            return GsxAttachment
+        if name in FLOAT_TYPES:
+            return GsxPriceElement
+        if re.search(r'Date$', name):
+            return GsxDateElement
+
+        return objectify.ObjectifiedElement
+
+
+def parse(root, response):
+    """
+    >>> parse('/tmp/authenticate.xml', 'AuthenticateResponse').userSessionId
+    Sdt7tXp2XytTEVwHBeDx6lHTXI3w9s+M
+    """
+    parser = objectify.makeparser(remove_blank_text=True)
+    parser.set_element_class_lookup(GsxClassLookup())
+
+    if isinstance(root, basestring) and os.path.exists(root):
+        root = objectify.parse(root, parser)
+    else:
+        root = objectify.fromstring(root, parser)
+
+    return root.find('*//%s' % response)
+
+if __name__ == '__main__':
+    import doctest
+    import logging
+    logging.basicConfig(level=logging.DEBUG)
+    doctest.testmod()