feat: remove COMPLIANCE_REPORT document type

open-contracting · Nov 28, 2023 · 6fdfbe9 · 6fdfbe9
1 parent ee81931
commit 6fdfbe9
Show file tree

Hide file tree

Showing 3 changed files with 37 additions and 2 deletions.
diff --git a/app/models.py b/app/models.py
@@ -106,7 +106,6 @@ class BorrowerDocumentType(Enum):
     BANK_CERTIFICATION_DOCUMENT = "BANK_CERTIFICATION_DOCUMENT"
     FINANCIAL_STATEMENT = "FINANCIAL_STATEMENT"
     SIGNED_CONTRACT = "SIGNED_CONTRACT"
-    COMPLIANCE_REPORT = "COMPLIANCE_REPORT"
     SHAREHOLDER_COMPOSITION = "SHAREHOLDER_COMPOSITION"
     CHAMBER_OF_COMMERCE = "CHAMBER_OF_COMMERCE"
     THREE_LAST_BANK_STATEMENT = "THREE_LAST_BANK_STATEMENT"

diff --git a/migrations/versions/c919c4501192_remove_compliance_doc_type.py b/migrations/versions/c919c4501192_remove_compliance_doc_type.py
@@ -0,0 +1,37 @@
+"""remove compliance doc type
+
+Revision ID: c919c4501192
+Revises: f9f53b0fd892
+Create Date: 2023-11-28 16:31:35.726353
+
+"""
+from alembic import op
+
+
+# revision identifiers, used by Alembic.
+revision = "c919c4501192"
+down_revision = "f9f53b0fd892"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    op.execute(
+        """create type borrower_document_type_tmp as enum ('INCORPORATION_DOCUMENT', 'SUPPLIER_REGISTRATION_DOCUMENT',
+        'BANK_NAME', 'BANK_CERTIFICATION_DOCUMENT', 'FINANCIAL_STATEMENT', 'SIGNED_CONTRACT',
+        'SHAREHOLDER_COMPOSITION', 'CHAMBER_OF_COMMERCE', 'THREE_LAST_BANK_STATEMENT');"""
+    )
+    op.execute("""DELETE from borrower_document where type = 'COMPLIANCE_REPORT'""")
+
+    op.execute(
+        """ALTER TABLE borrower_document ALTER COLUMN type TYPE borrower_document_type_tmp
+        USING (type::text::borrower_document_type_tmp);"""
+    )
+
+    op.execute("""DROP TYPE borrower_document_type""")
+
+    op.execute("""ALTER TYPE borrower_document_type_tmp RENAME TO borrower_document_type;""")
+
+
+def downgrade() -> None:
+    pass
diff --git a/reportlab_mods.py b/reportlab_mods.py
@@ -33,7 +33,6 @@
     BorrowerDocumentType.BANK_CERTIFICATION_DOCUMENT: "Bank Certification Document",
     BorrowerDocumentType.FINANCIAL_STATEMENT: "Financial Statement",
     BorrowerDocumentType.SIGNED_CONTRACT: "Signed Contract",
-    BorrowerDocumentType.COMPLIANCE_REPORT: "Compliance Report",
     BorrowerDocumentType.SHAREHOLDER_COMPOSITION: "Shareholder composition",
     BorrowerDocumentType.CHAMBER_OF_COMMERCE: "Chamber of Commerce",
     BorrowerDocumentType.THREE_LAST_BANK_STATEMENT: "Three last bank statement",