Add schema extension for Stores, importing from CORE API

rattail_corepos/db/alembic/versions/7fea5aebddfb_add_store_extension.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8; -*-
+"""add store extension
+
+Revision ID: 7fea5aebddfb
+Revises: 130e4632a28a
+Create Date: 2021-01-27 21:01:59.767884
+
+"""
+
+from __future__ import unicode_literals
+
+# revision identifiers, used by Alembic.
+revision = '7fea5aebddfb'
+down_revision = '130e4632a28a'
+branch_labels = None
+depends_on = None
+
+from alembic import op
+import sqlalchemy as sa
+import rattail.db.types
+
+
+
+def upgrade():
+
+    # corepos_store
+    op.create_table('corepos_store',
+                    sa.Column('uuid', sa.String(length=32), nullable=False),
+                    sa.Column('corepos_id', sa.Integer(), nullable=False),
+                    sa.ForeignKeyConstraint(['uuid'], ['store.uuid'], name='corepos_store_fk_store'),
+                    sa.PrimaryKeyConstraint('uuid')
+    )
+    op.create_table('corepos_store_version',
+                    sa.Column('uuid', sa.String(length=32), autoincrement=False, nullable=False),
+                    sa.Column('corepos_id', sa.Integer(), autoincrement=False, nullable=True),
+                    sa.Column('transaction_id', sa.BigInteger(), autoincrement=False, nullable=False),
+                    sa.Column('end_transaction_id', sa.BigInteger(), nullable=True),
+                    sa.Column('operation_type', sa.SmallInteger(), nullable=False),
+                    sa.PrimaryKeyConstraint('uuid', 'transaction_id')
+    )
+    op.create_index(op.f('ix_corepos_store_version_end_transaction_id'), 'corepos_store_version', ['end_transaction_id'], unique=False)
+    op.create_index(op.f('ix_corepos_store_version_operation_type'), 'corepos_store_version', ['operation_type'], unique=False)
+    op.create_index(op.f('ix_corepos_store_version_transaction_id'), 'corepos_store_version', ['transaction_id'], unique=False)
+
+
+def downgrade():
+
+    # corepos_store
+    op.drop_index(op.f('ix_corepos_store_version_transaction_id'), table_name='corepos_store_version')
+    op.drop_index(op.f('ix_corepos_store_version_operation_type'), table_name='corepos_store_version')
+    op.drop_index(op.f('ix_corepos_store_version_end_transaction_id'), table_name='corepos_store_version')
+    op.drop_table('corepos_store_version')
+    op.drop_table('corepos_store')

rattail_corepos/db/model/__init__.py

@@ -2,7 +2,7 @@
 ################################################################################
 #
 #  Rattail -- Retail Software Framework
-#  Copyright © 2010-2020 Lance Edgar
+#  Copyright © 2010-2021 Lance Edgar
 #
 #  This file is part of Rattail.
 #
@@ -24,6 +24,7 @@
 Database schema extensions for CORE-POS integration
 """
 
+from .stores import CoreStore
 from .people import CorePerson, CoreCustomer, CoreMember
 from .products import (CoreDepartment, CoreSubdepartment,
                        CoreVendor, CoreProduct, CoreProductCost)

rattail_corepos/db/model/stores.py

@@ -0,0 +1,65 @@
+# -*- coding: utf-8; -*-
+################################################################################
+#
+#  Rattail -- Retail Software Framework
+#  Copyright © 2010-2021 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/>.
+#
+################################################################################
+"""
+Database schema extensions for CORE-POS integration
+"""
+
+import sqlalchemy as sa
+from sqlalchemy import orm
+
+from rattail.db import model
+
+
+class CoreStore(model.Base):
+    """
+    CORE-specific extensions to :class:`rattail:rattail.db.model.Store`.
+    """
+    __tablename__ = 'corepos_store'
+    __table_args__ = (
+        sa.ForeignKeyConstraint(['uuid'], ['store.uuid'],
+                                name='corepos_store_fk_store'),
+    )
+    __versioned__ = {}
+
+    uuid = model.uuid_column(default=None)
+    store = orm.relationship(
+        model.Store,
+        doc="""
+        Reference to the actual store record, which this one extends.
+        """,
+        backref=orm.backref(
+            '_corepos',
+            uselist=False,
+            cascade='all, delete-orphan',
+            doc="""
+            Reference to the CORE-POS extension record for this store.
+            """))
+
+    corepos_id = sa.Column(sa.Integer(), nullable=False, doc="""
+    ``Stores.storeID`` value for this store, within CORE-POS.
+    """)
+
+    def __str__(self):
+        return str(self.store)
+
+CoreStore.make_proxy(model.Store, '_corepos', 'corepos_id')