Files
arrr-erp/alembic/versions/20260822_einvoice_ewaybill_enrichment_phase8.py
2026-08-22 15:30:46 +05:30

113 lines
7.5 KiB
Python

"""Phase 8 E-Invoice and E-Way Bill purchase enrichment.
Revision ID: 20260822_purchase_enrich_p8
Revises: 20260822_gstr2b_purchase_p7
"""
from alembic import op
import sqlalchemy as sa
revision = "20260822_purchase_enrich_p8"
down_revision = "20260822_gstr2b_purchase_p7"
branch_labels = None
depends_on = None
def upgrade():
op.create_table(
"accounting_purchase_enrichment_batches",
sa.Column("id", sa.Integer(), primary_key=True),
sa.Column("tenant_id", sa.Integer(), sa.ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False),
sa.Column("client_id", sa.Integer(), sa.ForeignKey("clients.id", ondelete="CASCADE"), nullable=False),
sa.Column("tally_guid", sa.String(120), nullable=False, server_default=""),
sa.Column("source_type", sa.String(30), nullable=False),
sa.Column("original_filename", sa.String(260), nullable=False, server_default=""),
sa.Column("file_sha256", sa.String(64), nullable=False),
sa.Column("source_period", sa.String(20), nullable=False, server_default=""),
sa.Column("status", sa.String(30), nullable=False, server_default="imported"),
sa.Column("records_read", sa.Integer(), nullable=False, server_default="0"),
sa.Column("records_imported", sa.Integer(), nullable=False, server_default="0"),
sa.Column("records_duplicate", sa.Integer(), nullable=False, server_default="0"),
sa.Column("records_linked", sa.Integer(), nullable=False, server_default="0"),
sa.Column("records_unmatched", sa.Integer(), nullable=False, server_default="0"),
sa.Column("records_ambiguous", sa.Integer(), nullable=False, server_default="0"),
sa.Column("imported_by_user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
sa.Column("created_at_utc", sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
sa.Column("completed_at_utc", sa.DateTime(timezone=True), nullable=True),
)
for col in ("tenant_id", "client_id", "tally_guid", "source_type", "file_sha256", "source_period", "status", "created_at_utc"):
op.create_index(f"ix_accounting_purchase_enrichment_batches_{col}", "accounting_purchase_enrichment_batches", [col])
op.create_table(
"accounting_purchase_enrichment_records",
sa.Column("id", sa.Integer(), primary_key=True),
sa.Column("tenant_id", sa.Integer(), sa.ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False),
sa.Column("client_id", sa.Integer(), sa.ForeignKey("clients.id", ondelete="CASCADE"), nullable=False),
sa.Column("batch_id", sa.Integer(), sa.ForeignKey("accounting_purchase_enrichment_batches.id", ondelete="CASCADE"), nullable=False),
sa.Column("tally_guid", sa.String(120), nullable=False, server_default=""),
sa.Column("source_type", sa.String(30), nullable=False),
sa.Column("source_document_key", sa.String(500), nullable=False),
sa.Column("external_reference", sa.String(180), nullable=False, server_default=""),
sa.Column("supplier_gstin", sa.String(20), nullable=False, server_default=""),
sa.Column("supplier_name", sa.String(260), nullable=False, server_default=""),
sa.Column("recipient_gstin", sa.String(20), nullable=False, server_default=""),
sa.Column("document_number", sa.String(160), nullable=False, server_default=""),
sa.Column("document_date", sa.String(20), nullable=False, server_default=""),
sa.Column("document_type", sa.String(40), nullable=False, server_default="invoice"),
sa.Column("taxable_value", sa.Float(), nullable=False, server_default="0"),
sa.Column("igst", sa.Float(), nullable=False, server_default="0"),
sa.Column("cgst", sa.Float(), nullable=False, server_default="0"),
sa.Column("sgst", sa.Float(), nullable=False, server_default="0"),
sa.Column("cess", sa.Float(), nullable=False, server_default="0"),
sa.Column("invoice_value", sa.Float(), nullable=False, server_default="0"),
sa.Column("place_of_supply", sa.String(120), nullable=False, server_default=""),
sa.Column("transport_mode", sa.String(80), nullable=False, server_default=""),
sa.Column("vehicle_number", sa.String(40), nullable=False, server_default=""),
sa.Column("transporter_id", sa.String(40), nullable=False, server_default=""),
sa.Column("raw_summary_json", sa.Text(), nullable=True),
sa.Column("source_row_hash", sa.String(64), nullable=False, server_default=""),
sa.Column("gstr2b_purchase_id", sa.Integer(), sa.ForeignKey("accounting_gstr2b_purchases.id", ondelete="SET NULL"), nullable=True),
sa.Column("match_status", sa.String(30), nullable=False, server_default="unmatched"),
sa.Column("match_method", sa.String(80), nullable=False, server_default=""),
sa.Column("match_confidence", sa.Integer(), nullable=False, server_default="0"),
sa.Column("manually_linked", sa.Boolean(), nullable=False, server_default=sa.false()),
sa.Column("created_at_utc", sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
sa.Column("updated_at_utc", sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
sa.UniqueConstraint(
"tenant_id", "client_id", "source_type", "source_document_key",
name="uq_accounting_purchase_enrichment_source_document",
),
)
for col in (
"tenant_id", "client_id", "batch_id", "tally_guid", "source_type", "source_document_key",
"external_reference", "supplier_gstin", "supplier_name", "recipient_gstin", "document_number",
"document_date", "document_type", "source_row_hash", "gstr2b_purchase_id", "match_status", "created_at_utc",
):
op.create_index(f"ix_accounting_purchase_enrichment_records_{col}", "accounting_purchase_enrichment_records", [col])
op.create_table(
"accounting_purchase_enrichment_items",
sa.Column("id", sa.Integer(), primary_key=True),
sa.Column("record_id", sa.Integer(), sa.ForeignKey("accounting_purchase_enrichment_records.id", ondelete="CASCADE"), nullable=False),
sa.Column("line_number", sa.Integer(), nullable=False, server_default="0"),
sa.Column("product_name", sa.String(500), nullable=False, server_default=""),
sa.Column("description_text", sa.Text(), nullable=True),
sa.Column("hsn_code", sa.String(20), nullable=False, server_default=""),
sa.Column("quantity", sa.Float(), nullable=False, server_default="0"),
sa.Column("unit", sa.String(30), nullable=False, server_default=""),
sa.Column("unit_price", sa.Float(), nullable=False, server_default="0"),
sa.Column("taxable_value", sa.Float(), nullable=False, server_default="0"),
sa.Column("gst_rate", sa.Float(), nullable=False, server_default="0"),
sa.Column("igst", sa.Float(), nullable=False, server_default="0"),
sa.Column("cgst", sa.Float(), nullable=False, server_default="0"),
sa.Column("sgst", sa.Float(), nullable=False, server_default="0"),
sa.Column("cess", sa.Float(), nullable=False, server_default="0"),
)
op.create_index("ix_accounting_purchase_enrichment_items_record_id", "accounting_purchase_enrichment_items", ["record_id"])
op.create_index("ix_accounting_purchase_enrichment_items_hsn_code", "accounting_purchase_enrichment_items", ["hsn_code"])
def downgrade():
op.drop_table("accounting_purchase_enrichment_items")
op.drop_table("accounting_purchase_enrichment_records")
op.drop_table("accounting_purchase_enrichment_batches")