forked from CMS/memberPlumbing
89 lines
3.1 KiB
Python
89 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
|
|
from datetime import datetime
|
|
import sqlite3
|
|
import yaml
|
|
|
|
#from common import membershipworks
|
|
|
|
# TODO: TEMP. remove later:
|
|
#from lib.MembershipWorks import MembershipWorks
|
|
from passwords import MEMBERSHIPWORKS_USERNAME, MEMBERSHIPWORKS_PASSWORD
|
|
membershipworks = MembershipWorks()
|
|
membershipworks.login(MEMBERSHIPWORKS_USERNAME, MEMBERSHIPWORKS_PASSWORD)
|
|
|
|
def insertFromTableMap(table, data, tableMap):
|
|
def formatRows(data):
|
|
for d in data:
|
|
yield [
|
|
d.get(v.get('source', k) if v is not None else k) for k, v in tableMap.items()
|
|
]
|
|
|
|
c.execute('BEGIN')
|
|
# NOTE: this only works if in python >= 3.7 where `dict` is ordered
|
|
c.executemany('INSERT OR REPLACE INTO ' + table + ' (' +
|
|
','.join('"' + k + '"' for k in tableMap.keys()) +
|
|
') values (' +
|
|
','.join(len(tableMap) * '?') +
|
|
');',
|
|
list(formatRows(data)))
|
|
c.execute('END')
|
|
|
|
def insertLabels(members):
|
|
def formatLabels(labels):
|
|
for member in members:
|
|
for label in membershipworks._parse_flags()['labels']:
|
|
if member[label]:
|
|
yield member['Account ID'], label
|
|
|
|
c.execute('BEGIN')
|
|
c.executemany('INSERT OR REPLACE INTO labels (uid, label) values (?, ?);',
|
|
list(formatLabels(members)))
|
|
c.execute('END')
|
|
|
|
with open('tableMapping.yaml') as f:
|
|
tableMapping = yaml.load(f, yaml.SafeLoader)
|
|
|
|
with sqlite3.connect('membershipworks.db') as conn:
|
|
c = conn.cursor()
|
|
|
|
c.execute('CREATE TABLE IF NOT EXISTS members (' +
|
|
', '.join([f'"{k}" {v.get("type", "") if v is not None else ""}'
|
|
for k, v in tableMapping['members'].items()]) +
|
|
');')
|
|
|
|
c.execute("CREATE TABLE IF NOT EXISTS transactions (" +
|
|
', '.join([f'"{k}" {v.get("type", "") if v is not None else ""}'
|
|
for k, v in tableMapping['transactions'].items()]) +
|
|
""", FOREIGN KEY(uid) REFERENCES member(uid)
|
|
-- FOREIGN KEY(event_id) REFERENCES event(eid)
|
|
);
|
|
""")
|
|
|
|
c.execute("""CREATE TABLE IF NOT EXISTS labels (
|
|
uid TEXT, label Text,
|
|
PRIMARY KEY(uid, label),
|
|
FOREIGN KEY(uid) REFERENCES member(uid)
|
|
);""")
|
|
|
|
|
|
members = membershipworks.get_all_members()
|
|
for m in members:
|
|
# replace flags by booleans
|
|
for flag in [dek['lbl'] for dek in membershipworks.org_info['dek']]:
|
|
if flag in m:
|
|
m[flag] = m[flag] == flag
|
|
|
|
for field_id, field in membershipworks._all_fields().items():
|
|
# convert checkboxes to real booleans
|
|
if field.get('typ') == 8 and field['lbl'] in m: # check box
|
|
m[field['lbl']] = True if m[field['lbl']] == 'Y' else False
|
|
|
|
insertFromTableMap('members', members, tableMapping['members'])
|
|
insertLabels(members)
|
|
|
|
transactions = membershipworks.get_transactions(datetime(2020, 1, 1), datetime.now())
|
|
insertFromTableMap('transactions', transactions, tableMapping['transactions'])
|
|
|
|
# TODO: folders, levels, addons
|