99 lines
3.4 KiB
Python
99 lines
3.4 KiB
Python
# -*- coding: utf-8; -*-
|
|
################################################################################
|
|
#
|
|
# Rattail -- Retail Software Framework
|
|
# Copyright © 2010-2020 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/>.
|
|
#
|
|
################################################################################
|
|
"""
|
|
DataSync for Rattail DB
|
|
"""
|
|
|
|
from sqlalchemy.orm.exc import NoResultFound
|
|
|
|
from corepos.api import CoreWebAPI
|
|
from corepos.db.office_op import Session as CoreSession, model as corepos
|
|
|
|
from rattail.datasync import NewDataSyncImportConsumer
|
|
|
|
|
|
class FromCOREAPIToRattail(NewDataSyncImportConsumer):
|
|
"""
|
|
Consumer for CORE POS (API) -> Rattail datasync
|
|
"""
|
|
handler_spec = 'rattail_corepos.importing.corepos.api:FromCOREPOSToRattail'
|
|
|
|
def setup(self):
|
|
super(FromCOREAPIToRattail, self).setup()
|
|
self.establish_api()
|
|
|
|
def establish_api(self):
|
|
url = self.config.require('corepos.api', 'url')
|
|
self.api = CoreWebAPI(url)
|
|
|
|
def get_host_object(self, session, change):
|
|
if change.payload_type == 'Department':
|
|
return self.api.get_department(change.payload_key)
|
|
if change.payload_type == 'Subdepartment':
|
|
return self.api.get_subdepartment(change.payload_key)
|
|
if change.payload_type == 'Vendor':
|
|
return self.api.get_vendor(change.payload_key)
|
|
if change.payload_type == 'Product':
|
|
return self.api.get_product(change.payload_key)
|
|
|
|
|
|
class FromCOREPOSToRattailBase(NewDataSyncImportConsumer):
|
|
"""
|
|
Base class for CORE POS -> Rattail data sync consumers.
|
|
"""
|
|
handler_spec = 'rattail_corepos.importing.corepos.db:FromCOREPOSToRattail'
|
|
|
|
def begin_transaction(self):
|
|
self.corepos_session = CoreSession()
|
|
|
|
def rollback_transaction(self):
|
|
self.corepos_session.rollback()
|
|
self.corepos_session.close()
|
|
|
|
def commit_transaction(self):
|
|
# always rollback here, we don't want any accidents in CORE POS
|
|
self.corepos_session.rollback()
|
|
self.corepos_session.close()
|
|
|
|
|
|
class FromCOREPOSToRattailProducts(FromCOREPOSToRattailBase):
|
|
"""
|
|
Handles CORE POS -> Rattail sync for product data.
|
|
"""
|
|
|
|
def get_host_object(self, session, change):
|
|
|
|
if change.payload_type == 'Product':
|
|
try:
|
|
return self.corepos_session.query(corepos.Product)\
|
|
.filter(corepos.Product.upc == change.payload_key)\
|
|
.one()
|
|
except NoResultFound:
|
|
pass
|
|
|
|
else:
|
|
# try to fetch CORE POS object via typical method
|
|
Model = getattr(corepos, change.payload_type)
|
|
return self.corepos_session.query(Model)\
|
|
.get(int(change.payload_key))
|