From 18f200d998120107ad181b720b52d9fee1b141ee Mon Sep 17 00:00:00 2001
From: Nathan Hertz <nhertz@nrao.edu>
Date: Tue, 13 Jul 2021 22:51:45 +0000
Subject: [PATCH] WS-516: Add ingest-complete flag to capability request table

---
 ...add_ingest_complete_flag_to_capability_.py | 23 +++++++++++++++++++
 .../workspaces/capability/schema.py           |  1 +
 2 files changed, 24 insertions(+)
 create mode 100644 schema/versions/0bca0d8b3564_add_ingest_complete_flag_to_capability_.py

diff --git a/schema/versions/0bca0d8b3564_add_ingest_complete_flag_to_capability_.py b/schema/versions/0bca0d8b3564_add_ingest_complete_flag_to_capability_.py
new file mode 100644
index 000000000..1debf1ecb
--- /dev/null
+++ b/schema/versions/0bca0d8b3564_add_ingest_complete_flag_to_capability_.py
@@ -0,0 +1,23 @@
+"""add ingest-complete flag to capability requests
+
+Revision ID: 0bca0d8b3564
+Revises: a7c2b4682aae
+Create Date: 2021-07-13 16:01:10.908563
+
+"""
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "0bca0d8b3564"
+down_revision = "a7c2b4682aae"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    op.execute("ALTER TABLE capability_requests ADD COLUMN ingested bool NOT NULL DEFAULT false")
+
+
+def downgrade():
+    op.execute("ALTER TABLE capability_requests DROP COLUMN IF EXISTS ingested")
diff --git a/shared/workspaces/workspaces/capability/schema.py b/shared/workspaces/workspaces/capability/schema.py
index f652d2f7c..e0438c05f 100644
--- a/shared/workspaces/workspaces/capability/schema.py
+++ b/shared/workspaces/workspaces/capability/schema.py
@@ -121,6 +121,7 @@ class CapabilityRequest(Base, CapabilityRequestIF):
     capability_name = sa.Column(
         "capability_name", sa.String, sa.ForeignKey("capabilities.capability_name")
     )
+    ingested = sa.Column("ingested", sa.Boolean, server_default="false", nullable=False)
     created_at = sa.Column(
         "created_at",
         sa.DateTime(timezone=True),
-- 
GitLab