Add support for Vendor -> Quickbooks Bank Accounts field

this feels a bit hacky yet, but had to introduce some new mechanisms
to allow for extra template stuff while avoiding adding a new
vendors/edit (etc.) template in this project, since we are using a
view supplement..ugh
This commit is contained in:
Lance Edgar 2024-04-16 18:23:49 -05:00
parent ddeb4545a6
commit 7e08dd0f89
3 changed files with 277 additions and 1 deletions
tailbone_quickbooks/views

View file

@ -2,7 +2,7 @@
################################################################################
#
# Rattail -- Retail Software Framework
# Copyright © 2010-2022 Lance Edgar
# Copyright © 2010-2024 Lance Edgar
#
# This file is part of Rattail.
#
@ -24,6 +24,14 @@
Vendor views, w/ Quickbooks integration
"""
import json
import colander
from deform import widget as dfwidget
from pyramid.renderers import render
from webhelpers2.html import HTML, tags
from tailbone import grids
from tailbone.views import ViewSupplement
@ -44,14 +52,138 @@ class VendorViewSupplement(ViewSupplement):
g.set_filter('quickbooks_terms', model.QuickbooksVendor.quickbooks_terms)
def configure_form(self, f):
# quickbooks_name
f.append('quickbooks_name')
# quickbooks_bank_account
f.append('quickbooks_bank_account')
# quickbooks_bank_accounts
f.append('quickbooks_bank_accounts_')
f.set_renderer('quickbooks_bank_accounts_', self.render_quickbooks_bank_accounts)
f.set_node('quickbooks_bank_accounts_', BankAccounts())
f.set_widget('quickbooks_bank_accounts_', BankAccountsWidget())
# quickbooks_terms
f.append('quickbooks_terms')
def render_quickbooks_bank_accounts(self, vendor, field):
accounts = getattr(vendor, 'quickbooks_bank_accounts')
if accounts:
g = make_accounts_grid()
return HTML.literal(g.render_table_element(data_prop='quickbooksBankAccountsData'))
def objectify(self, vendor, form, data):
model = self.model
old_accounts = vendor.quickbooks_bank_accounts
new_accounts = data['quickbooks_bank_accounts_']
for new_account in new_accounts:
old_account = old_accounts.get(new_account['store_uuid'])
if old_account:
if old_account.account_number != new_account['account_number']:
old_account.account_number = new_account['account_number']
else:
account = model.QuickbooksVendorBankAccount()
account.store_uuid = new_account['store_uuid']
account.account_number = new_account['account_number']
vendor.quickbooks_bank_accounts[account.store_uuid] = account
self.Session.add(account)
final_store_uuids = set([a['store_uuid'] for a in new_accounts])
for old_account in list(vendor.quickbooks_bank_accounts.values()):
if old_account.store_uuid not in final_store_uuids:
self.Session.delete(old_account)
return vendor
def template_kwargs(self, **kwargs):
app = self.get_rattail_app()
form = kwargs.get('form')
if form:
# quickbooks bank accounts
vendor = kwargs['instance']
accounts = []
for account in vendor.quickbooks_bank_accounts.values():
store = account.store
accounts.append({
'uuid': account.uuid,
'store': f'{store.id} - {store.name}',
'store_uuid': store.uuid,
'store_id': store.id,
'store_name': store.name,
'account_number': account.account_number,
})
accounts.sort(key=lambda a: a['store_id'])
# nb. this is needed for widget *and* readonly template
form.set_json_data('quickbooksBankAccountsData', accounts)
# TODO: these are needed by widget
stores = []
for store in app.get_active_stores(self.Session()):
stores.append({
'uuid': store.uuid,
'display': f'{store.id} - {store.name}',
})
form.include_template('/vendors/quickbooks_bank_accounts_js.mako', {
'store_options': stores,
})
return kwargs
def get_version_child_classes(self):
model = self.model
return [model.QuickbooksVendor]
def make_accounts_grid():
g = grids.Grid('quickbooks_bank_accounts',
[], # empty data
columns=[
'store',
'account_number',
])
return g
class BankAccount(colander.MappingSchema):
store_uuid = colander.SchemaNode(colander.String())
account_number = colander.SchemaNode(colander.String())
class BankAccounts(colander.SequenceSchema):
account = BankAccount()
class BankAccountsWidget(dfwidget.Widget):
def serialize(self, field, cstruct, **kw):
g = make_accounts_grid()
g.main_actions.append(
grids.GridAction('edit', icon='edit',
click_handler='quickbooksBankAccountEdit(props.row)'))
g.main_actions.append(
grids.GridAction('delete', icon='trash',
click_handler='quickbooksBankAccountDelete(props.row)'))
widget = render('/vendors/quickbooks_bank_accounts_widget.mako', {
'grid': g,
})
return HTML.tag('div', c=[
HTML.literal(widget),
tags.hidden(field.name, **{':value': "quickbooksBankAccountsFinal"}),
])
def deserialize(self, field, pstruct):
return json.loads(pstruct)
def includeme(config):
VendorViewSupplement.defaults(config)