# -*- coding: utf-8 -*-

import os
import re
import base64
import tempfile

from lxml import objectify
from datetime import datetime

DATETIME_TYPES = ('dispatchSentDate',)
BASE64_TYPES = ('packingList', 'proformaFileData', 'returnLabelFileData',)
FLOAT_TYPES = ('totalFromOrder', 'exchangePrice', 'stockPrice', 'netPrice',)

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
}


def gsx_date(value):
    try:
        # standard GSX format: "mm/dd/yy"
        return datetime.strptime(value, "%m/%d/%y").date()
    except ValueError:
        pass

    try:
        # some dates are formatted as "yyyy-mm-dd"
        return datetime.strptime(value, "%Y-%m-%d").date()
    except (ValueError, TypeError):
        pass


def gsx_boolean(value):
    return value == 'Y' or value == 'true'


def gsx_price(value):
    return float(re.sub(r'[A-Z ,]', '', value))


def gsx_attachment(value):
    v = base64.b64decode(value)
    of = tempfile.NamedTemporaryFile(suffix=".pdf", delete=False)
    of.write(v)
    return of.name


def gsx_datetime(value):
    # 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+)$', value)
    ts, tz = m.groups()
    return datetime.strptime(ts, "%Y-%m-%d %H:%M:%S")


def gsx_timestamp(value):
    return datetime.strptime(value, "%d-%b-%y %H:%M:%S")


class GsxElement(objectify.ObjectifiedElement):
    def __getattribute__(self, name):

        result = super(GsxElement, self).__getattribute__(name)

        if isinstance(result, objectify.NumberElement):
            return result.pyval

        if isinstance(result, objectify.StringElement):

            name = result.tag
            result = result.text

            if not result:
                return

            if name in DATETIME_TYPES:
                return gsx_datetime(result)
            if name in BASE64_TYPES:
                return gsx_attachment(result)
            if name in FLOAT_TYPES:
                return gsx_price(result)
            if re.search(r'Date$', name):
                return gsx_date(result)
            if re.search(r'^[YN]$', result):
                return gsx_boolean(result)

        return result


def parse(root, response):
    """
    >>> parse('tests/fixtures/warranty_status.xml', 'warrantyDetailInfo').warrantyStatus
    'Apple Limited Warranty'
    >>> parse('tests/fixtures/warranty_status.xml', 'warrantyDetailInfo').estimatedPurchaseDate
    datetime.date(2010, 8, 25)
    >>> parse('tests/fixtures/warranty_status.xml', 'warrantyDetailInfo').limitedWarranty
    True
    >>> parse('tests/fixtures/warranty_status.xml', 'warrantyDetailInfo').isPersonalized
    """
    parser = objectify.makeparser(remove_blank_text=True)
    lookup = objectify.ObjectifyElementClassLookup(tree_class=GsxElement)
    parser.set_element_class_lookup(lookup)

    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()