finoex/dingguo/parser/kickstarter.py

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

import BeautifulSoup
import datetime
import dateutil.parser
import dingguo
import email.message
import ioex
import re

def parse_order_confirmation_mail(mail):

    assert isinstance(mail, email.message.Message)

    html = mail.get_payload()[1].get_payload(decode = True).decode('utf-8')

    if not 'kickstarter' in html:
        raise Exception()

    email_recipient_address = mail['delivered-to'].decode('utf8')
    email_date = datetime.datetime.utcfromtimestamp(
        # seconds since the Epoch in UTC
        email.utils.mktime_tz(email.utils.parsedate_tz(mail['date']))
        )
    order = dingguo.Order(
        platform = u'kickstarter',
        order_id = '%s-%s' % (email_recipient_address, email_date.isoformat()),
        order_date = email_date,
        customer_id = email_recipient_address,
        )

    doc = BeautifulSoup.BeautifulSoup(html)

    def get_text(tag):
        return u''.join([c if type(c) is BeautifulSoup.NavigableString else get_text(c) for c in tag.contents])

    shipping_attr = re.search(
            ur'^Shipping(?P<dest>.*)\((?P<price>.*)\)$',
            doc.find(text = re.compile('Shipping')).parent.parent.text,
            ).groupdict()

    shipping = dingguo.Shipping(
        destination_point = shipping_attr['dest'].strip(),
        price_brutto = dingguo.Sum.parse_text(shipping_attr['price']),
        )

    pledge_amount = dingguo.Sum.parse_text(
            doc.find(text = re.compile('Amount pledged')).parent.parent.text.replace('Amount pledged', '')
            )

    order.items.append(dingguo.Pledge(
        campaign = dingguo.Campaign(
            name = doc.find('h2').text,
            founder = doc.find('h2').findNext('p').text[len('By '):],
            end = dateutil.parser.parse(
                doc.find(text = re.compile('your card will be charged on')).findNext('time')['datetime']
                ),
            ),
        price_brutto = pledge_amount - shipping.price_brutto,
        reward = get_text(doc.find(text = re.compile('Reward')).findPrevious('td'))
            .strip()[len('Reward'):].strip().replace('\r\n', '\n'),
        ))

    order.items.append(shipping)

    return [order]