rattail/rattail-corepos
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Add 2-way sync for basic Member data, CORE <-> Rattail

Browse source
This commit is contained in:
Lance Edgar 2020-03-18 11:30:18 -05:00
parent 9dbdb81f07
commit 8917316a21
5 changed files with 158 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
rattail_corepos/corepos/importing/model.py
View file

@ -92,8 +92,8 @@ class MemberImporter(ToCoreAPI):
# 'chargeBalance',
# 'chargeLimit',
# 'idCardUPC',
# 'startDate',
# 'endDate',
'startDate',
'endDate',
'addressFirstLine',
'addressSecondLine',
'city',
@ -123,6 +123,8 @@ class MemberImporter(ToCoreAPI):
# 'modified',
]
empty_date_value = '0000-00-00 00:00:00'
def get_local_objects(self, host_data=None):
return get_core_members(self.api, progress=self.progress)
@ -144,6 +146,15 @@ class MemberImporter(ToCoreAPI):
if not self.customer_data_differs(local_data, host_data):
diffs.remove('customers')
# also the start/end dates should be looked at more closely. if they
# contain the special '__omit__' value then we won't ever count as diff
if 'startDate' in self.fields and 'startDate' in diffs:
if host_data['startDate'] == '__omit__':
diffs.remove('startDate')
if 'endDate' in self.fields and 'endDate' in diffs:
if host_data['endDate'] == '__omit__':
diffs.remove('endDate')
return diffs
def customer_data_differs(self, local_data, host_data):
@ -211,6 +222,11 @@ class MemberImporter(ToCoreAPI):
return data
cardNo = data.pop('cardNo')
data = dict(data)
if data.get('startDate') == '__omit__':
data.pop('startDate')
if data.get('endDate') == '__omit__':
data.pop('endDate')
member = self.api.set_member(cardNo, **data)
return member

18
rattail_corepos/corepos/importing/rattail.py
View file

@ -78,6 +78,8 @@ class MemberImporter(FromRattail, corepos_importing.model.MemberImporter):
'city',
'state',
'zip',
'startDate',
'endDate',
]
supported_customer_fields = [
'customerID',
@ -120,6 +122,20 @@ class MemberImporter(FromRattail, corepos_importing.model.MemberImporter):
'email': email.address if email else '',
})
member = customer.only_member(require=False)
if member:
if member.joined:
start_date = member.joined.strftime('%Y-%m-%d 00:00:00')
else:
start_date = self.empty_date_value
if member.withdrew:
end_date = member.withdrew.strftime('%Y-%m-%d 00:00:00')
else:
end_date = self.empty_date_value
else:
start_date = '__omit__'
end_date = '__omit__'
return {
'cardNo': customer.number,
'customerAccountID': customer.id,
@ -128,6 +144,8 @@ class MemberImporter(FromRattail, corepos_importing.model.MemberImporter):
'city': address.city if address else '',
'state': address.state if address else '',
'zip': address.zipcode if address else '',
'startDate': start_date,
'endDate': end_date,
'customers': people,
}

6
rattail_corepos/datasync/corepos.py
View file

@ -160,6 +160,7 @@ class FromRattailToCore(NewDataSyncImportConsumer):
'CustomerMailingAddress',
'PersonPhoneNumber',
'PersonEmailAddress',
'Member',
]
for change in [c for c in changes if c.payload_type in types]:
if change.payload_type == 'Customer' and change.deletion:
@ -252,6 +253,11 @@ class FromRattailToCore(NewDataSyncImportConsumer):
if email:
return email.person.customers
if change.payload_type == 'Member':
member = session.query(model.Member).get(change.payload_key)
if member:
return [member.customer]
return []
def get_vendor(self, session, change):

2
rattail_corepos/datasync/rattail.py
View file

@ -73,6 +73,8 @@ class FromCOREAPIToRattail(NewDataSyncImportConsumer):
for person in people:
self.process_change(session, self.importers['Person'],
host_object=person)
self.process_change(session, self.importers['Member'],
host_object=member)
# process all remaining supported models with typical logic
types = [

126
rattail_corepos/importing/corepos/api.py
View file

@ -24,6 +24,7 @@
CORE POS (API) -> Rattail data importing
"""
import datetime
import decimal
import logging
@ -55,6 +56,7 @@ class FromCOREPOSToRattail(importing.ToRattailHandler):
importers['Customer'] = CustomerImporter
importers['Person'] = PersonImporter
importers['CustomerPerson'] = CustomerPersonImporter
importers['Member'] = MemberImporter
importers['Department'] = DepartmentImporter
importers['Subdepartment'] = SubdepartmentImporter
importers['Vendor'] = VendorImporter
@ -100,20 +102,32 @@ class CustomerImporter(FromCOREPOSAPI, importing.model.CustomerImporter):
def normalize_host_object(self, member):
# figure out the "account holder" customer for the member
if member['customerAccountID'] == 0:
log.debug("member %s has customerAccountID of 0: %s",
member['cardNo'], member)
return
# figure out the "account holder" customer for the member. note that
# we only use this to determine the `Customer.name` in Rattail
customers = member['customers']
account_holders = [customer for customer in customers
if customer['accountHolder']]
if len(account_holders) > 1:
log.warning("member %s has %s account holders in CORE: %s",
member['cardNo'], len(account_holders), member)
elif not account_holders:
raise NotImplementedError("TODO: how to handle member with no account holders?")
customer = account_holders[0]
if account_holders:
if len(account_holders) > 1:
log.warning("member %s has %s account holders in CORE: %s",
member['cardNo'], len(account_holders), member)
customer = account_holders[0]
elif customers:
if len(customers) > 1:
log.warning("member %s has %s customers but no account holders: %s",
member['cardNo'], len(customers), member)
customer = customers[0]
else:
raise NotImplementedError("TODO: how to handle member with no customers?")
return {
'id': member['customerAccountID'],
'number': member['cardNo'],
'number': int(member['cardNo']),
'name': normalize_full_name(customer['firstName'],
customer['lastName']),
@ -167,12 +181,22 @@ class PersonImporter(FromCOREPOSAPI, corepos_importing.model.PersonImporter):
Return a list of Person data objects for the given Member. This
logic is split out separately so that datasync can leverage it too.
"""
customers = member['customers']
people = []
# make sure we put the account holder first in the list!
customers = sorted(member['customers'],
key=lambda cust: 1 if cust['accountHolder'] else 0,
reverse=True)
# make sure account holder is listed first
account_holder = None
secondary = False
mixedup = False
for customer in customers:
if customer['accountHolder'] and not secondary:
account_holder = customer
elif not customer['accountHolder']:
secondary = True
elif customer['accountHolder'] and secondary:
mixedup = True
if mixedup:
raise NotImplementedError("TODO: should re-sort the customers list for member {}".format(member['cardNo']))
for i, customer in enumerate(customers, 1):
person = dict(customer)
@ -450,3 +474,81 @@ class ProductImporter(FromCOREPOSAPI, importing.model.ProductImporter):
'regular_price_multiple': 1 if price is not None else None,
'regular_price_type': self.enum.PRICE_TYPE_REGULAR if price is not None else None,
}
class MemberImporter(FromCOREPOSAPI, importing.model.MemberImporter):
"""
Importer for member data from CORE POS API.
"""
key = 'id'
supported_fields = [
'id',
'customer_uuid',
'joined',
'withdrew',
]
# TODO: should make this configurable
member_status_codes = [
'PC',
'TERM',
]
# TODO: should make this configurable
non_member_status_codes = [
'REG',
'INACT',
]
def setup(self):
super(MemberImporter, self).setup()
self.customers = self.cache_model(model.Customer, key='number')
def get_host_objects(self):
return self.get_core_members()
def get_customer(self, number):
if hasattr(self, 'customers'):
return self.customers.get(number)
try:
return self.session.query(model.Customer)\
.filter(model.Customer.number == number)\
.one()
except NoResultFound:
pass
def normalize_host_object(self, member):
customer = self.get_customer(member['cardNo'])
if not customer:
log.warning("Rattail customer not found for cardNo %s: %s",
member['cardNo'], member)
return
if member['memberStatus'] in self.non_member_status_codes:
log.debug("skipping non-member %s with status '%s': %s",
member['memberStatus'], member['cardNo'], member)
return
if member['memberStatus'] not in self.member_status_codes:
# note that we will still import this one! we don't skip it
log.warning("unexpected status '%s' for member %s: %s",
member['memberStatus'], member['cardNo'], member)
joined = None
if member['startDate'] and member['startDate'] != '0000-00-00 00:00:00':
joined = datetime.datetime.strptime(member['startDate'],
'%Y-%m-%d %H:%M:%S')
joined = joined.date()
withdrew = None
if member['endDate'] and member['endDate'] != '0000-00-00 00:00:00':
withdrew = datetime.datetime.strptime(member['endDate'],
'%Y-%m-%d %H:%M:%S')
withdrew = withdrew.date()
return {
'id': str(member['cardNo']),
'customer_uuid': customer.uuid,
'joined': joined,
'withdrew': withdrew,
}