Add computed is_finalized column

Author: davenquinn

map-integration/macrostrat/map_integration/__init__.py

@@ -202,6 +202,15 @@ def change_slug(map: MapInfo, new_slug: str, dry_run: bool = False):
         print(f"Changed slug from {map.slug} to {new_slug}")
 
 
+@cli.command(name="update-status")
+def update_status():
+    """Update the status of all maps."""
+    from .status import update_status_for_all_maps
+    from .database import db
+
+    update_status_for_all_maps(db)
+
+
 # TODO: integrate this migration command with the main database migrations
 def _run_migrations(database: str = None):
     """Run migrations to convert a Macrostrat v1 sources table to v2 format."""

map-integration/macrostrat/map_integration/process/__init__.py

@@ -15,16 +15,16 @@
 
 """
 
-from ..database import db, sql_file
-from ..match import match_liths, match_strat_names, match_units
-from ..utils import IngestionCLI
-from ..utils.map_info import MapInfo
 from .extract_strat_name_candidates import extract_strat_name_candidates
 from .geometry import create_rgeom, create_webgeom
 from .insert import copy_to_maps
 from .legend_lookup import legend_lookup
 from .lookup import make_lookup
 from .status import processing_status
+from ..database import db, sql_file
+from ..match import match_liths, match_strat_names, match_units
+from ..utils import IngestionCLI
+from ..utils.map_info import MapInfo
 
 cli = IngestionCLI(
     no_args_is_help=True, name="process", help="Process map data once ingested"
@@ -81,3 +81,15 @@ def legend(map: MapInfo):
 
 cli.add_command(make_lookup, name="lookup", rich_help_panel="Lookup")
 cli.add_command(legend_lookup, name="legend-lookup", rich_help_panel="Lookup")
+
+
+def finalize_map(source: MapInfo):
+    """
+    Finalize a map source by setting is_finalized to True in the sources table.
+
+    This is a computed parameter, so we can change its design in the future.
+    """
+    db.run_query(
+        "UPDATE maps.sources SET is_finalized = TRUE WHERE source_id = :source_id",
+        {"source_id": source.id},
+    )

map-integration/macrostrat/map_integration/status.py

@@ -0,0 +1,30 @@
+from rich import print
+
+from .utils.map_info import has_map_schema_data, MapInfo
+
+
+def update_status_for_all_maps(db):
+    """
+    Check the status of a map.
+    """
+    # Get all map IDs
+    maps = db.run_query(
+        "SELECT source_id, slug FROM maps.sources ORDER BY source_id"
+    ).all()
+
+    print("Checking whether all maps have data in the [cyan]maps[/] schema...")
+
+    for _map in maps:
+        map_info = MapInfo(id=_map.source_id, slug=_map.slug)
+        is_finalized = has_map_schema_data(db, map_info)
+
+        db.run_query(
+            "UPDATE maps.sources SET is_finalized = :is_finalized WHERE source_id = :map_id",
+            dict(is_finalized=is_finalized, map_id=map_info.id),
+        )
+
+        _prt_val = "[green]finalized[/]" if is_finalized else "[yellow]not finalized[/]"
+
+        print(f"#{map_info.id} {map_info.slug} is {_prt_val}")
+
+    db.session.commit()

map-integration/macrostrat/map_integration/utils/map_info.py

@@ -1,16 +1,15 @@
 from typing import Optional
 
+from macrostrat.database import Database
 from psycopg2.sql import Identifier
 from pydantic import BaseModel
 from typer import Argument
 from typing_extensions import Annotated
 
 from macrostrat.core import app
 from macrostrat.core.exc import MacrostratError
-from macrostrat.database import Database
-
-from ..database import db
 from ._database import table_exists
+from ..database import db
 
 
 class _MapInfo(BaseModel):
@@ -21,6 +20,10 @@ class _MapInfo(BaseModel):
     url: Optional[str] = None
     name: Optional[str] = None
 
+    @property
+    def source_id(self):
+        return self.id
+
 
 def complete_map_slugs(incomplete: str):
     return (
@@ -117,3 +120,9 @@ def feature_counts(db, info: MapInfo):
         ),
     ).one()
     return res
+
+
+def has_map_schema_data(db: Database, map: MapInfo):
+    counts = feature_counts(db, map)
+    total = counts.n_polygons + counts.n_lines + counts.n_points
+    return total > 0