diff --git a/rattail_corepos/db/alembic/versions/c61f78243ff3_add_person_extension.py b/rattail_corepos/db/alembic/versions/c61f78243ff3_add_person_extension.py new file mode 100644 index 0000000..6c428d9 --- /dev/null +++ b/rattail_corepos/db/alembic/versions/c61f78243ff3_add_person_extension.py @@ -0,0 +1,51 @@ +# -*- coding: utf-8 -*- +"""add Person extension + +Revision ID: c61f78243ff3 +Revises: b43e93d32275 +Create Date: 2020-03-16 16:50:55.266325 + +""" + +# revision identifiers, used by Alembic. +revision = 'c61f78243ff3' +down_revision = 'b43e93d32275' +branch_labels = None +depends_on = None + +from alembic import op +import sqlalchemy as sa +import rattail.db.types + + + +def upgrade(): + + # corepos_person + op.create_table('corepos_person', + sa.Column('uuid', sa.String(length=32), nullable=False), + sa.Column('corepos_customer_id', sa.Integer(), nullable=False), + sa.ForeignKeyConstraint(['uuid'], ['person.uuid'], name='corepos_person_fk_person'), + sa.PrimaryKeyConstraint('uuid') + ) + op.create_table('corepos_person_version', + sa.Column('uuid', sa.String(length=32), autoincrement=False, nullable=False), + sa.Column('corepos_customer_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_person_version_end_transaction_id'), 'corepos_person_version', ['end_transaction_id'], unique=False) + op.create_index(op.f('ix_corepos_person_version_operation_type'), 'corepos_person_version', ['operation_type'], unique=False) + op.create_index(op.f('ix_corepos_person_version_transaction_id'), 'corepos_person_version', ['transaction_id'], unique=False) + + +def downgrade(): + + # corepos_person + op.drop_index(op.f('ix_corepos_person_version_transaction_id'), table_name='corepos_person_version') + op.drop_index(op.f('ix_corepos_person_version_operation_type'), table_name='corepos_person_version') + op.drop_index(op.f('ix_corepos_person_version_end_transaction_id'), table_name='corepos_person_version') + op.drop_table('corepos_person_version') + op.drop_table('corepos_person') diff --git a/rattail_corepos/db/model.py b/rattail_corepos/db/model.py index 6f4f87e..9bf3dbf 100644 --- a/rattail_corepos/db/model.py +++ b/rattail_corepos/db/model.py @@ -30,7 +30,42 @@ from sqlalchemy import orm from rattail.db import model -__all__ = ['CoreVendor'] +__all__ = ['CorePerson', 'CoreVendor'] + + +class CorePerson(model.Base): + """ + CORE-specific extensions to :class:`rattail:rattail.db.model.Person`. + """ + __tablename__ = 'corepos_person' + __table_args__ = ( + sa.ForeignKeyConstraint(['uuid'], ['person.uuid'], + name='corepos_person_fk_person'), + ) + __versioned__ = {} + + uuid = model.uuid_column(default=None) + person = orm.relationship( + model.Person, + doc=""" + Reference to the actual person 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 person. + """)) + + corepos_customer_id = sa.Column(sa.Integer(), nullable=False, doc=""" + ``Customers.customerID`` value for this person, within CORE-POS. + """) + + def __str__(self): + return str(self.person) + +CorePerson.make_proxy(model.Person, '_corepos', 'corepos_customer_id') class CoreVendor(model.Base):