Files
arrr-erp/alembic/versions/20260822_gstr2b_purchase_intelligence_phase7.py
2026-08-22 14:31:24 +05:30

100 lines
6.7 KiB
Python

"""Phase 7 GSTR-2B purchase intelligence.
Revision ID: 20260822_gstr2b_purchase_p7
Revises: 20260822_ledger_learning_p6
"""
from alembic import op
import sqlalchemy as sa
revision = "20260822_gstr2b_purchase_p7"
down_revision = "20260822_ledger_learning_p6"
branch_labels = None
depends_on = None
def upgrade():
op.create_table(
"accounting_gstr2b_import_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("original_filename", sa.String(260), nullable=False, server_default=""),
sa.Column("file_sha256", sa.String(64), nullable=False),
sa.Column("return_period", sa.String(20), nullable=False, server_default=""),
sa.Column("source_kind", sa.String(30), nullable=False, server_default="gstr2b_upload"),
sa.Column("status", sa.String(30), nullable=False, server_default="imported"),
sa.Column("rows_read", sa.Integer(), nullable=False, server_default="0"),
sa.Column("rows_imported", sa.Integer(), nullable=False, server_default="0"),
sa.Column("rows_skipped_duplicate", sa.Integer(), nullable=False, server_default="0"),
sa.Column("rows_skipped_invalid", sa.Integer(), nullable=False, server_default="0"),
sa.Column("analyzed_rows", sa.Integer(), nullable=False, server_default="0"),
sa.Column("reviewed_rows", sa.Integer(), nullable=False, server_default="0"),
sa.Column("error_message", sa.Text(), nullable=True),
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("analyzed_at_utc", sa.DateTime(timezone=True), nullable=True),
sa.Column("completed_at_utc", sa.DateTime(timezone=True), nullable=True),
)
for col in ("tenant_id", "client_id", "tally_guid", "file_sha256", "return_period", "source_kind", "status", "created_at_utc"):
op.create_index(f"ix_accounting_gstr2b_import_batches_{col}", "accounting_gstr2b_import_batches", [col])
op.create_table(
"accounting_gstr2b_purchases",
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_gstr2b_import_batches.id", ondelete="CASCADE"), nullable=False),
sa.Column("tally_guid", sa.String(120), nullable=False, server_default=""),
sa.Column("return_period", sa.String(20), 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("invoice_number", sa.String(160), nullable=False, server_default=""),
sa.Column("invoice_date", sa.String(20), nullable=False, server_default=""),
sa.Column("document_type", sa.String(40), nullable=False, server_default="invoice"),
sa.Column("invoice_type", sa.String(80), nullable=False, server_default=""),
sa.Column("place_of_supply", sa.String(120), nullable=False, server_default=""),
sa.Column("reverse_charge", sa.String(20), nullable=False, server_default=""),
sa.Column("itc_availability", sa.String(80), nullable=False, server_default=""),
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("hsn_code", sa.String(20), nullable=False, server_default=""),
sa.Column("description_text", sa.Text(), nullable=True),
sa.Column("source_sheet", sa.String(160), nullable=False, server_default=""),
sa.Column("source_row_number", sa.Integer(), nullable=False, server_default="0"),
sa.Column("source_row_hash", sa.String(64), nullable=False, server_default=""),
sa.Column("review_status", sa.String(30), nullable=False, server_default="pending_analysis"),
sa.Column("suggested_nature_id", sa.Integer(), sa.ForeignKey("accounting_natures.id", ondelete="SET NULL"), nullable=True),
sa.Column("suggested_ledger_name", sa.String(240), nullable=False, server_default=""),
sa.Column("suggested_confidence", sa.Integer(), nullable=False, server_default="0"),
sa.Column("suggestion_explanation_json", sa.Text(), nullable=True),
sa.Column("final_nature_id", sa.Integer(), sa.ForeignKey("accounting_natures.id", ondelete="SET NULL"), nullable=True),
sa.Column("final_ledger_name", sa.String(240), nullable=False, server_default=""),
sa.Column("reviewed_by_user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
sa.Column("reviewed_at_utc", sa.DateTime(timezone=True), nullable=True),
sa.Column("posting_status", sa.String(30), nullable=False, server_default="not_enabled"),
sa.Column("is_duplicate_source", 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", "supplier_gstin", "invoice_number", "invoice_date", "document_type",
name="uq_accounting_gstr2b_purchase_document",
),
)
for col in (
"tenant_id", "client_id", "batch_id", "tally_guid", "return_period", "supplier_gstin",
"supplier_name", "invoice_number", "invoice_date", "document_type", "hsn_code",
"source_row_hash", "review_status", "suggested_nature_id", "final_nature_id",
"posting_status", "created_at_utc",
):
op.create_index(f"ix_accounting_gstr2b_purchases_{col}", "accounting_gstr2b_purchases", [col])
def downgrade():
op.drop_table("accounting_gstr2b_purchases")
op.drop_table("accounting_gstr2b_import_batches")