86 lines
3.8 KiB
Python
86 lines
3.8 KiB
Python
from typing import Sequence, Union
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '20251107_150001_add_warehouse_docs'
|
|
down_revision: Union[str, None] = 'ac9e4b3dcffc'
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
bind = op.get_bind()
|
|
inspector = sa.inspect(bind)
|
|
|
|
tables = inspector.get_table_names()
|
|
|
|
if 'warehouse_documents' not in tables:
|
|
op.create_table(
|
|
'warehouse_documents',
|
|
sa.Column('id', sa.Integer(), primary_key=True, autoincrement=True),
|
|
sa.Column('business_id', sa.Integer(), nullable=False),
|
|
sa.Column('fiscal_year_id', sa.Integer(), nullable=True),
|
|
sa.Column('code', sa.String(length=64), nullable=False, unique=True),
|
|
sa.Column('document_date', sa.Date(), nullable=False),
|
|
sa.Column('status', sa.String(length=16), nullable=False, server_default='draft'),
|
|
sa.Column('doc_type', sa.String(length=32), nullable=False),
|
|
sa.Column('warehouse_id_from', sa.Integer(), nullable=True),
|
|
sa.Column('warehouse_id_to', sa.Integer(), nullable=True),
|
|
sa.Column('source_type', sa.String(length=32), nullable=True),
|
|
sa.Column('source_document_id', sa.Integer(), nullable=True),
|
|
sa.Column('extra_info', sa.JSON(), nullable=True),
|
|
sa.Column('created_by_user_id', sa.Integer(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP')),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=False, server_default=sa.text('CURRENT_TIMESTAMP')),
|
|
)
|
|
try:
|
|
op.create_index('ix_wh_docs_business_date', 'warehouse_documents', ['business_id', 'document_date'])
|
|
op.create_index('ix_wh_docs_code', 'warehouse_documents', ['code'])
|
|
except Exception:
|
|
pass
|
|
|
|
if 'warehouse_document_lines' not in tables:
|
|
op.create_table(
|
|
'warehouse_document_lines',
|
|
sa.Column('id', sa.Integer(), primary_key=True, autoincrement=True),
|
|
sa.Column('warehouse_document_id', sa.Integer(), sa.ForeignKey('warehouse_documents.id', ondelete='CASCADE'), nullable=False),
|
|
sa.Column('product_id', sa.Integer(), nullable=False),
|
|
sa.Column('warehouse_id', sa.Integer(), nullable=True),
|
|
sa.Column('movement', sa.String(length=8), nullable=False),
|
|
sa.Column('quantity', sa.Numeric(18, 6), nullable=False),
|
|
sa.Column('cost_price', sa.Numeric(18, 6), nullable=True),
|
|
sa.Column('cogs_amount', sa.Numeric(18, 6), nullable=True),
|
|
sa.Column('extra_info', sa.JSON(), nullable=True),
|
|
)
|
|
try:
|
|
op.create_index('ix_wh_lines_doc', 'warehouse_document_lines', ['warehouse_document_id'])
|
|
op.create_index('ix_wh_lines_product', 'warehouse_document_lines', ['product_id'])
|
|
op.create_index('ix_wh_lines_warehouse', 'warehouse_document_lines', ['warehouse_id'])
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
def downgrade() -> None:
|
|
try:
|
|
op.drop_index('ix_wh_lines_warehouse', table_name='warehouse_document_lines')
|
|
op.drop_index('ix_wh_lines_product', table_name='warehouse_document_lines')
|
|
op.drop_index('ix_wh_lines_doc', table_name='warehouse_document_lines')
|
|
except Exception:
|
|
pass
|
|
try:
|
|
op.drop_table('warehouse_document_lines')
|
|
except Exception:
|
|
pass
|
|
try:
|
|
op.drop_index('ix_wh_docs_code', table_name='warehouse_documents')
|
|
op.drop_index('ix_wh_docs_business_date', table_name='warehouse_documents')
|
|
except Exception:
|
|
pass
|
|
try:
|
|
op.drop_table('warehouse_documents')
|
|
except Exception:
|
|
pass
|
|
|
|
|