wip

app/modules/sightings/models.py

@@ -54,6 +54,13 @@ class SightingStage(str, enum.Enum):
     failed = 'failed'
 
 
+class SightingMatchState(str, enum.Enum):
+    unreviewed = 'unreviewed'
+    in_progress = 'in_progress'
+    reviewed = 'reviewed'
+    unidentifiable = 'unidentifiable'
+
+
 class Sighting(db.Model, HoustonModel, CustomFieldMixin, ExportMixin):
     """
     Sightings database model.
@@ -68,6 +75,11 @@ class Sighting(db.Model, HoustonModel, CustomFieldMixin, ExportMixin):
         db.Enum(SightingStage),
         nullable=False,
     )
+    match_state = db.Column(
+        db.Enum(SightingMatchState),
+        nullable=False,
+        default=SightingMatchState.unreviewed,
+    )
     featured_asset_guid = db.Column(db.GUID, default=None, nullable=True)
 
     # May have multiple jobs outstanding, store as Json obj dictionary, uuid_str is key,

migrations/versions/53b0fec87272_.py

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+"""empty message
+
+Revision ID: 53b0fec87272
+Revises: 2c9d119ac26b
+Create Date: 2023-10-10 23:16:44.527006
+
+"""
+import sqlalchemy as sa
+from alembic import op
+
+from app.modules.sightings.models import SightingMatchState
+
+# revision identifiers, used by Alembic.
+revision = '53b0fec87272'
+down_revision = '2c9d119ac26b'
+
+
+def upgrade():
+    """
+    Upgrade Semantic Description:
+        ENTER DESCRIPTION HERE
+    """
+    match_state_enum = sa.Enum(
+        SightingMatchState,
+        name='sightingmatchstate',
+        values_callable=lambda obj: [e.value for e in obj],
+    )
+    match_state_enum.create(op.get_bind(), checkfirst=True)
+
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('sighting', schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                'match_state',
+                match_state_enum,
+                nullable=False,
+                default=SightingMatchState.unreviewed.value,
+                server_default=SightingMatchState.unreviewed.value,
+            )
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    """
+    Downgrade Semantic Description:
+        ENTER DESCRIPTION HERE
+    """
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('sighting', schema=None) as batch_op:
+        batch_op.drop_column('match_state')
+
+    # Remove the enums created as part of the upgrade above
+    sa.Enum(name='sightingmatchstate').drop(op.get_bind(), checkfirst=False)
+
+    # ### end Alembic commands ###