diff options
Diffstat (limited to 'gsxws/core.py')
| -rw-r--r-- | gsxws/core.py | 40 |
1 files changed, 26 insertions, 14 deletions
diff --git a/gsxws/core.py b/gsxws/core.py index 4733001..da82314 100644 --- a/gsxws/core.py +++ b/gsxws/core.py @@ -394,34 +394,46 @@ class GsxObject(object): v = unicode(v) # "must be unicode, not str" - # strip currency prefix and munge into float - if re.search(r'Price$', k): - v = float(re.sub(r'[A-Z ,]', '', v)) - # convert Y and N to boolean - if re.search(r'^[YN]$', k): + if re.search(r'^[YN]$', v): v = (v == 'Y') - # convert dates to native Python type ("mm/dd/yy") - if re.search(r'^\d{2}/\d{2}/\d{2}$', v): - m, d, y = v.split('/') - v = date(2000+int(y), int(m), int(d)) - - # seems that some dates are in the format "yyyy-mm-dd" - if re.search(r'^\d{4}-\d{2}-\d{2}$', v): - y, m, d = v.split('-') - v = date(int(y), int(m), int(d)) + # strip currency prefix and munge into float + if re.search(r'Price$', k): + v = float(re.sub(r'[A-Z ,]', '', v)) # Convert timestamps to native Python type # 18-Jan-13 14:38:04 if re.search(r'TimeStamp$', k): v = datetime.strptime(v, '%d-%b-%y %H:%M:%S') + if re.search(r'Date$', k): + # looks like some sort of date, let's try to convert + # @TODO: return actual native dates, not isoformat() + try: + # standard GSX format: "mm/dd/yy" + dt = datetime.strptime(v, "%m/%d/%y") + v = dt.date().isoformat() + except ValueError: + pass + + try: + # some dates are formatted as "yyyy-mm-dd" + dt = datetime.strptime(v, "%Y-%m-%d") + v = dt.date().isoformat() + except (ValueError, TypeError): + pass + setattr(obj, k, v) return obj +class GsxRequestObject(GsxObject): + "The GSX-friendly representation of this GsxObject" + pass + + class GsxSession(GsxObject): userId = "" |