debconf-data/penta/frontaccounting_sync.py

import logging
import mechanize
import random
import re
import sys
import urllib
import dc_objs
from BeautifulSoup import BeautifulSoup

FRONTACCOUNTING_URL = "https://debconf10.kaplowitz.org/account"
HTTP_USERNAME = "XXXXXX"
HTTP_PASSWORD = "XXXXXX"
APP_USERNAME = "XXXXXX"
APP_PASSWORD = "XXXXXX"

CAT_MAPPINGS = { }
CAT_MAPPINGS.update(dict.fromkeys(dc_objs.CAT_B, 'Pays Own Bed & Food'))
CAT_MAPPINGS.update(dict.fromkeys(dc_objs.CAT_BA, 'Sponsored Bed Only'))
CAT_MAPPINGS.update(dict.fromkeys(dc_objs.CAT_BF, 'Sponsored Food Only'))
CAT_MAPPINGS.update(dict.fromkeys(dc_objs.CAT_BAF, 'Sponsored Bed & Food'))
CAT_MAPPINGS.update(dict.fromkeys(dc_objs.CAT_PRO, 'Professional'))
CAT_MAPPINGS.update(dict.fromkeys(dc_objs.CAT_COR, 'Corporate'))

def clean_html(br):
        data = br.response().get_data()
        soup = BeautifulSoup(data)
        response = mechanize.make_response(str(soup), 
          [("Content-Type", "text/html")], br.geturl(), 200, "OK")
        br.set_response(response)

def ajax_submit(br, name=None):
        q = urllib.quote
        # Yay adapting JavaScript weirdness
        br.form.action += "?JsHttpRequest=0-xml"
        req = mechanize.Request(br.form.action)
        req.add_header('Content-Type', 'application/octet-stream; charset=UTF-8')
        pairs = br.form.click_pairs()
        body = "&".join(["%s=%s" % (q(k),q(v)) for (k,v) in br.form.click_pairs(name=name)])
        #body = re.sub('&_[a-z_]+_update=%20', '', body)
        if not "&submit=" in body:
                body += "&submit=1"
        body += "&_random=%f" % (random.random() * 1234567)
        req.add_header('Content-Length', len(body))
        req.add_data(body)
        br.open(req)

def ajax_select(br, label, name):
        br.set_value_by_label(label, name=name) #XXX NAME
        update_name = "_%s_update" % name
        br.submit(name=update_name)
        clean_html(br)
        br.select_form(nr=0)

br = mechanize.Browser()
br.add_password(FRONTACCOUNTING_URL, HTTP_USERNAME, HTTP_PASSWORD)
br.open(FRONTACCOUNTING_URL)

br.select_form(name="loginform")
br["user_name_entry_field"] = APP_USERNAME
br["password"] = APP_PASSWORD
br.submit()

People = dc_objs.get_people(where="dcvn.attend and dcvn.reconfirmed")

for P in people:
        br.open(FRONTACCOUNTING_URL)
        br.follow_link(text="Add and Manage Customers")

        br.select_form(nr=0)
        br["CustName"] = br["cust_ref"] = P.name
        br["email"] = P.email
        br["notes"] = "person_id: %d" % P.person_id
        br.set_value_by_label([CAT_MAPPINGS[P.participant_category]], name="sales_type")
        br.set_value_by_label(["Net 30"], name="payment_terms")
        br.set_value_by_label(["Conference Registration"], name="dimension_id")

        ajax_submit(br)

        br.open(FRONTACCOUNTING_URL)
        br.follow_link(text="Customer Branches")
        clean_html(br)
        br.select_form(nr=0)
        ajax_select(br, [P.name], name="customer_id") #XXX NAME
        ajax_submit(br, name="ADD_ITEM")

# This part isn't working
#br.open(FRONTACCOUNTING_URL)
#br.follow_link(text="Direct Invoice")
#clean_html(br)
#br.select_form(nr=0)
#ajax_select(br, ["Test Monkey"], name="customer_id") #XXX NAME
##ajax_select(br, ["Residence Hall Bed"], name="stock_id") #XXX line_item
##br["qty"] = 6 #XXX line_item
1	jimmy	1439	import logging
2			import mechanize
3			import random
4			import re
5			import sys
6			import urllib
7	jimmy	1444	import dc_objs
8	jimmy	1439	from BeautifulSoup import BeautifulSoup
9
10			FRONTACCOUNTING_URL = "https://debconf10.kaplowitz.org/account"
11			HTTP_USERNAME = "XXXXXX"
12			HTTP_PASSWORD = "XXXXXX"
13			APP_USERNAME = "XXXXXX"
14			APP_PASSWORD = "XXXXXX"
15
16	jimmy	1444	CAT_MAPPINGS = { }
17			CAT_MAPPINGS.update(dict.fromkeys(dc_objs.CAT_B, 'Pays Own Bed & Food'))
18			CAT_MAPPINGS.update(dict.fromkeys(dc_objs.CAT_BA, 'Sponsored Bed Only'))
19			CAT_MAPPINGS.update(dict.fromkeys(dc_objs.CAT_BF, 'Sponsored Food Only'))
20			CAT_MAPPINGS.update(dict.fromkeys(dc_objs.CAT_BAF, 'Sponsored Bed & Food'))
21			CAT_MAPPINGS.update(dict.fromkeys(dc_objs.CAT_PRO, 'Professional'))
22			CAT_MAPPINGS.update(dict.fromkeys(dc_objs.CAT_COR, 'Corporate'))
23
24	jimmy	1439	def clean_html(br):
25			data = br.response().get_data()
26			soup = BeautifulSoup(data)
27			response = mechanize.make_response(str(soup),
28			[("Content-Type", "text/html")], br.geturl(), 200, "OK")
29			br.set_response(response)
30
31			def ajax_submit(br, name=None):
32			q = urllib.quote
33			# Yay adapting JavaScript weirdness
34			br.form.action += "?JsHttpRequest=0-xml"
35			req = mechanize.Request(br.form.action)
36			req.add_header('Content-Type', 'application/octet-stream; charset=UTF-8')
37			pairs = br.form.click_pairs()
38			body = "&".join(["%s=%s" % (q(k),q(v)) for (k,v) in br.form.click_pairs(name=name)])
39			#body = re.sub('&_[a-z_]+_update=%20', '', body)
40			if not "&submit=" in body:
41			body += "&submit=1"
42			body += "&_random=%f" % (random.random() * 1234567)
43			req.add_header('Content-Length', len(body))
44			req.add_data(body)
45			br.open(req)
46
47			def ajax_select(br, label, name):
48			br.set_value_by_label(label, name=name) #XXX NAME
49			update_name = "_%s_update" % name
50			br.submit(name=update_name)
51			clean_html(br)
52			br.select_form(nr=0)
53
54			br = mechanize.Browser()
55			br.add_password(FRONTACCOUNTING_URL, HTTP_USERNAME, HTTP_PASSWORD)
56			br.open(FRONTACCOUNTING_URL)
57
58			br.select_form(name="loginform")
59			br["user_name_entry_field"] = APP_USERNAME
60			br["password"] = APP_PASSWORD
61			br.submit()
62
63	jimmy	1444	People = dc_objs.get_people(where="dcvn.attend and dcvn.reconfirmed")
64	jimmy	1439
65	jimmy	1444	for P in people:
66			br.open(FRONTACCOUNTING_URL)
67			br.follow_link(text="Add and Manage Customers")
68	jimmy	1439
69	jimmy	1444	br.select_form(nr=0)
70			br["CustName"] = br["cust_ref"] = P.name
71			br["email"] = P.email
72			br["notes"] = "person_id: %d" % P.person_id
73			br.set_value_by_label([CAT_MAPPINGS[P.participant_category]], name="sales_type")
74			br.set_value_by_label(["Net 30"], name="payment_terms")
75			br.set_value_by_label(["Conference Registration"], name="dimension_id")
76	jimmy	1439
77	jimmy	1444	ajax_submit(br)
78	jimmy	1439
79	jimmy	1444	br.open(FRONTACCOUNTING_URL)
80			br.follow_link(text="Customer Branches")
81			clean_html(br)
82			br.select_form(nr=0)
83			ajax_select(br, [P.name], name="customer_id") #XXX NAME
84			ajax_submit(br, name="ADD_ITEM")
85
86			# This part isn't working
87			#br.open(FRONTACCOUNTING_URL)
88			#br.follow_link(text="Direct Invoice")
89			#clean_html(br)
90			#br.select_form(nr=0)
91			#ajax_select(br, ["Test Monkey"], name="customer_id") #XXX NAME
92			##ajax_select(br, ["Residence Hall Bed"], name="stock_id") #XXX line_item
93			##br["qty"] = 6 #XXX line_item