FreeBSDAdmin/Reminder/shellmc.py

# Shell MasterCard
#
# Douglas Thrift
#
# $Id$

import bank
from ClientForm import FormParser
from datetime import datetime
from mechanize import LinkNotFoundError
import re
import website

class _FormParser(FormParser):
        def __init__(self, *args, **kwargs):
                FormParser.__init__(self, *args, **kwargs)

        def do_input(self, attributes):
                self.__enable(attributes)
                FormParser.do_input(self, attributes)

        def start_select(self, attributes):
                self.__name = None

                for name, value in attributes:
                        if name == 'name':
                                self.__name = value

                self.__enable(attributes)
                FormParser.start_select(self, attributes)

        def end_select(self):
                if self.__name == 'download_date':
                        self.do_option([('value', 'Activity Since Last Statement')])
                elif self.__name == 'download_format':
                        self.do_option([('value', 'QIF')])

                FormParser.end_select(self)

        def __enable(self, attributes):
                try:
                        attributes.remove(('disabled', 'disabled'))
                except ValueError:
                        pass

class Bank(website.Website, bank.Bank):
        PAYMENT_DUE = re.compile(r'([A-Z][a-z]{2})\.?( \d{1,2}, \d{4})')

        def __init__(self, config, debug):
                website.Website.__init__(self, debug, factory = website.Factory(_FormParser))
                bank.Bank.__init__(self, config)

                self.browser.open('http://www.shellmc.accountonline.com/')
                self.browser.select_form(name = 'LOGIN')

                self.browser['USERNAME'] = self._username()
                self.browser['PASSWORD'] = self._password()

                self.browser.submit()

        def __del__(self):
                try:
                        self.browser.follow_link(text_regex = 'Log Out')
                except (AttributeError, LinkNotFoundError):
                        pass

        def download(self, account):
                try:
                        balance = self.Soup(self._follow_link(text_regex = account)).find(attrs = {'class': 'curr_balance'}).string

                        self._follow_link(text_regex = 'View/Download Your Statement')
                        self.browser.select_form(name = 'download_statement')

                        self.browser.form.action = 'https://www.accountonline.com/cards/svc/StatementDownload.do?dateRange=unbilled&viewType=qif'
                        self.browser['download_date'] = ['Activity Since Last Statement']
                        self.browser['download_format'] = ['QIF']

                        return 'unbilled.qif', self._submit().read(), balance
                finally:
                        self._back()

        def due(self, account):
                try:
                        return datetime.strptime(''.join(self.PAYMENT_DUE.search(self.Soup(self._follow_link(text_regex = account)).find(text = re.compile('Payment Due')).findNext(text = self.PAYMENT_DUE)).group(1, 2)), '%b %d, %Y').date()
                finally:
                        self._back()
Revision:	1179
Committed:	2009-05-19T16:53:29-07:00 (16 years, 1 month ago) by douglas
Content type:	text/x-python
File size:	2463 byte(s)
Log Message:	Working somewhat again.
#	Content
1	# Shell MasterCard
2	#
3	# Douglas Thrift
4	#
5	# $Id$
6
7	import bank
8	from ClientForm import FormParser
9	from datetime import datetime
10	from mechanize import LinkNotFoundError
11	import re
12	import website
13
14	class _FormParser(FormParser):
15	def __init__(self, args, *kwargs):
16	FormParser.__init__(self, args, *kwargs)
17
18	def do_input(self, attributes):
19	self.__enable(attributes)
20	FormParser.do_input(self, attributes)
21
22	def start_select(self, attributes):
23	self.__name = None
24
25	for name, value in attributes:
26	if name == 'name':
27	self.__name = value
28
29	self.__enable(attributes)
30	FormParser.start_select(self, attributes)
31
32	def end_select(self):
33	if self.__name == 'download_date':
34	self.do_option([('value', 'Activity Since Last Statement')])
35	elif self.__name == 'download_format':
36	self.do_option([('value', 'QIF')])
37
38	FormParser.end_select(self)
39
40	def __enable(self, attributes):
41	try:
42	attributes.remove(('disabled', 'disabled'))
43	except ValueError:
44	pass
45
46	class Bank(website.Website, bank.Bank):
47	PAYMENT_DUE = re.compile(r'([A-Z][a-z]{2})\.?( \d{1,2}, \d{4})')
48
49	def __init__(self, config, debug):
50	website.Website.__init__(self, debug, factory = website.Factory(_FormParser))
51	bank.Bank.__init__(self, config)
52
53	self.browser.open('http://www.shellmc.accountonline.com/')
54	self.browser.select_form(name = 'LOGIN')
55
56	self.browser['USERNAME'] = self._username()
57	self.browser['PASSWORD'] = self._password()
58
59	self.browser.submit()
60
61	def __del__(self):
62	try:
63	self.browser.follow_link(text_regex = 'Log Out')
64	except (AttributeError, LinkNotFoundError):
65	pass
66
67	def download(self, account):
68	try:
69	balance = self.Soup(self._follow_link(text_regex = account)).find(attrs = {'class': 'curr_balance'}).string
70
71	self._follow_link(text_regex = 'View/Download Your Statement')
72	self.browser.select_form(name = 'download_statement')
73
74	self.browser.form.action = 'https://www.accountonline.com/cards/svc/StatementDownload.do?dateRange=unbilled&viewType=qif'
75	self.browser['download_date'] = ['Activity Since Last Statement']
76	self.browser['download_format'] = ['QIF']
77
78	return 'unbilled.qif', self._submit().read(), balance
79	finally:
80	self._back()
81
82	def due(self, account):
83	try:
84	return datetime.strptime(''.join(self.PAYMENT_DUE.search(self.Soup(self._follow_link(text_regex = account)).find(text = re.compile('Payment Due')).findNext(text = self.PAYMENT_DUE)).group(1, 2)), '%b %d, %Y').date()
85	finally:
86	self._back()