diff --git a/rattail_corepos/datasync/rattail.py b/rattail_corepos/datasync/rattail.py index a87573f..867397c 100644 --- a/rattail_corepos/datasync/rattail.py +++ b/rattail_corepos/datasync/rattail.py @@ -47,6 +47,8 @@ class FromCOREAPIToRattail(NewDataSyncImportConsumer): self.api = CoreWebAPI(url) def get_host_object(self, session, change): + if change.payload_type == 'Customer': + return self.api.get_customer(change.payload_key) if change.payload_type == 'Department': return self.api.get_department(change.payload_key) if change.payload_type == 'Subdepartment': diff --git a/rattail_corepos/importing/corepos/api.py b/rattail_corepos/importing/corepos/api.py index 5a63d86..29e1ec7 100644 --- a/rattail_corepos/importing/corepos/api.py +++ b/rattail_corepos/importing/corepos/api.py @@ -28,10 +28,12 @@ import decimal import logging from corepos.api import CoreWebAPI +from corepos.db.office_op import Session as CoreSession, model as corepos from rattail import importing from rattail.gpc import GPC from rattail.util import OrderedDict +from rattail.db.util import normalize_full_name, short_session from rattail_corepos import importing as corepos_importing @@ -46,6 +48,7 @@ class FromCOREPOSToRattail(importing.ToRattailHandler): def get_importers(self): importers = OrderedDict() + importers['Customer'] = CustomerImporter importers['Department'] = DepartmentImporter importers['Subdepartment'] = SubdepartmentImporter importers['Vendor'] = VendorImporter @@ -67,6 +70,69 @@ class FromCOREPOSAPI(importing.Importer): self.api = CoreWebAPI(url) +class CustomerImporter(FromCOREPOSAPI, importing.model.CustomerImporter): + """ + Importer for customer data from CORE POS API. + """ + key = 'number' + supported_fields = [ + 'id', + 'number', + 'name', + 'first_name', + 'last_name', + ] + + def get_host_objects(self): + # TODO: ideally could do this, but API doesn't let us fetch "all" + # return self.api.get_customers() + + # first we fetch all customer records from CORE DB + with short_session(Session=CoreSession) as s: + db_customers = s.query(corepos.Customer).all() + s.expunge_all() + + # now we must fetch each customer account individually from API + customers = OrderedDict() + + def fetch(dbcust, i): + if dbcust.card_number in customers: + return # already fetched this one + customer = self.api.get_customer(dbcust.card_number) + if customer: + customers[dbcust.card_number] = customer + else: + logger = log.warning if dbcust.person_number == 1 else log.debug + logger("could not fetch customer from CORE API: %s", + dbcust.card_number) + + self.progress_loop(fetch, db_customers, + message="Fetching Customer data from CORE-POS API") + return list(customers.values()) + + def normalize_host_object(self, customer): + + # figure out the "account holder" person for the customer + people = customer['customers'] + account_holders = [person for person in people + if person['accountHolder']] + if len(account_holders) > 1: + log.warning("customer %s has %s account holders in CORE: %s", + customer['cardNo'], len(account_holders), customer) + elif not account_holders: + raise NotImplementedError("TODO: how to handle customer with no account holders?") + person = account_holders[0] + + return { + 'id': customer['customerAccountID'], + 'number': int(customer['cardNo']), + 'name': normalize_full_name(person['firstName'], + person['lastName']), + 'first_name': person['firstName'], + 'last_name': person['lastName'], + } + + class DepartmentImporter(FromCOREPOSAPI, importing.model.DepartmentImporter): """ Importer for department data from CORE POS API.