Add views for CORE equity import batches

This commit is contained in:
Lance Edgar 2023-09-13 13:16:57 -05:00
parent a5de7a9dc1
commit 9bb46a1690

View file

@ -0,0 +1,149 @@
# -*- coding: utf-8; -*-
################################################################################
#
# Rattail -- Retail Software Framework
# Copyright © 2010-2023 Lance Edgar
#
# This file is part of Rattail.
#
# Rattail is free software: you can redistribute it and/or modify it under the
# terms of the GNU General Public License as published by the Free Software
# Foundation, either version 3 of the License, or (at your option) any later
# version.
#
# Rattail is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
# details.
#
# You should have received a copy of the GNU General Public License along with
# Rattail. If not, see <http://www.gnu.org/licenses/>.
#
################################################################################
"""
Views for CORE equity import batches
"""
from corepos.db.office_op import model as corepos
from rattail_corepos.db.model import CoreEquityImportBatch, CoreEquityImportBatchRow
from rattail_corepos.batch.equityimport import CoreEquityImportBatchHandler
from rattail_corepos.config import core_office_url, core_office_customer_account_url
from webhelpers2.html import tags
from tailbone.views.batch import BatchMasterView
from tailbone_corepos.db import CoreOfficeSession
class CoreEquityImportBatchView(BatchMasterView):
"""
Master view for CORE member batches.
"""
model_class = CoreEquityImportBatch
model_row_class = CoreEquityImportBatchRow
batch_handler_class = CoreEquityImportBatchHandler
route_prefix = 'batch.corepos.equity_import'
url_prefix = '/batch/corepos-equity-import'
rows_bulk_deletable = True
row_labels = {
'member_type_id': "Member Type",
'corepos_equity_total': "CORE-POS Equity Total",
}
row_grid_columns = [
'sequence',
'card_number',
'first_name',
'last_name',
'member_type_id',
'payment_amount',
'department_number',
'timestamp',
'status_code',
]
row_form_fields = [
'sequence',
'payment',
'card_number',
'first_name',
'last_name',
'member_type_id',
'payment_amount',
'department_number',
'timestamp',
'corepos_equity_total',
'rattail_equity_total',
'other_equity_total',
'status_code',
]
def configure_row_grid(self, g):
super().configure_row_grid(g)
app = self.get_rattail_app()
model = self.model
# card_number
g.set_link('card_number')
if 'card_number' in g.filters:
g.filters['card_number'].default_active = True
g.filters['card_number'].default_verb = 'equal'
# *_name
g.set_link('first_name')
g.set_link('last_name')
# member_type_id
query = self.Session.query(model.MembershipType)\
.order_by(model.MembershipType.name)
memtypes = app.cache_model(self.Session(), model.MembershipType, query=query,
key='number', normalizer=lambda t: str(t))
g.set_enum('member_type_id', memtypes)
# payment_amount
g.set_type('payment_amount', 'currency')
# department_number
g.set_label('department_number', "Department")
if 'department_number' in g.filters:
g.filters['department_number'].label = "Department Number"
def row_grid_extra_class(self, row, i):
if row.status_code in (row.STATUS_MEMBER_NOT_FOUND,
row.STATUS_MISSING_VALUES):
return 'warning'
if row.status_code in (row.STATUS_NEEDS_ATTENTION,):
return 'notice'
def configure_row_form(self, f):
super().configure_row_form(f)
app = self.get_rattail_app()
model = self.model
f.set_renderer('payment', self.render_payment)
query = self.Session.query(model.MembershipType)\
.order_by(model.MembershipType.name)
memtypes = app.cache_model(self.Session(), model.MembershipType, query=query,
key='number', normalizer=lambda t: str(t))
f.set_enum('member_type_id', memtypes)
f.set_type('payment_amount', 'currency')
f.set_type('corepos_equity_total', 'currency')
f.set_type('rattail_equity_total', 'currency')
f.set_type('other_equity_total', 'currency')
def render_payment(self, row, field):
payment = getattr(row, field)
if not payment:
return
text = str(payment)
url = self.request.route_url('member_equity_payments.view', uuid=payment.uuid)
return tags.link_to(text, url)
def includeme(config):
CoreEquityImportBatchView.defaults(config)