Merge pull request #412 from cymed/measurement-view

vw_tww_measurement_series

datamodel/app/create_app.py

@@ -13,6 +13,7 @@
 from triggers.set_defaults_and_triggers import set_defaults_and_triggers
 from view.vw_tww_additional_ws import vw_tww_additional_ws
 from view.vw_tww_infiltration_installation import vw_tww_infiltration_installation
+from view.vw_tww_measurement_series import vw_tww_measurement_series
 from view.vw_tww_reach import vw_tww_reach
 from view.vw_tww_wastewater_structure import vw_tww_wastewater_structure
 from view.vw_wastewater_structure import vw_wastewater_structure
@@ -166,6 +167,7 @@ def create_app(
     vw_tww_infiltration_installation(srid, pg_service=pg_service, extra_definition=tww_ii_extra)
     vw_tww_reach(pg_service=pg_service, extra_definition=tww_reach_extra)
     vw_tww_additional_ws(srid, pg_service=pg_service)
+    vw_tww_measurement_series(pg_service=pg_service)
 
     run_sql_file("view/vw_file.sql", pg_service, variables)
 

datamodel/app/view/vw_tww_measurement_series.py

@@ -0,0 +1,150 @@
+#!/usr/bin/env python3
+#
+# -- View: vw_tww_measurement_series
+
+import argparse
+import os
+
+try:
+    import psycopg
+except ImportError:
+    import psycopg2 as psycopg
+
+from pirogue.utils import insert_command, select_columns, update_command
+
+
+def vw_tww_measurement_series(pg_service: str = None):
+    """
+    Creates tww_measurement_series view
+    :param pg_service: the PostgreSQL service name
+    """
+    if not pg_service:
+        pg_service = os.getenv("PGSERVICE")
+    assert pg_service
+
+    conn = psycopg.connect(f"service={pg_service}")
+    cursor = conn.cursor()
+
+    view_sql = """
+    DROP VIEW IF EXISTS tww_app.vw_tww_measurement_series;
+
+    CREATE OR REPLACE VIEW tww_app.vw_tww_measurement_series AS
+     SELECT
+        {ms_cols}
+        , array_agg(mr.value) AS mr_values
+
+        FROM tww_od.measurement_series ms
+        LEFT JOIN tww_od.measurement_result mr ON ms.obj_id = mr.fk_measurement_series
+        GROUP BY {ms_cols};
+
+    """.format(
+        ms_cols=select_columns(
+            pg_cur=cursor,
+            table_schema="tww_od",
+            table_name="measurement_series",
+            table_alias="ms",
+            remove_pkey=False,
+            indent=4,
+            skip_columns=[],
+        ),
+    )
+
+    cursor.execute(view_sql)
+
+    trigger_insert_sql = """
+    CREATE OR REPLACE FUNCTION tww_app.ft_vw_tww_measurement_series_INSERT()
+      RETURNS trigger AS
+    $BODY$
+    BEGIN
+
+      NEW.identifier = COALESCE(NEW.identifier, NEW.obj_id);
+
+    {insert_ms}
+
+      RETURN NEW;
+    END; $BODY$ LANGUAGE plpgsql VOLATILE;
+
+    DROP TRIGGER IF EXISTS vw_tww_measurement_series_INSERT ON tww_app.vw_tww_measurement_series;
+
+    CREATE TRIGGER vw_tww_measurement_series_INSERT INSTEAD OF INSERT ON tww_app.vw_tww_measurement_series
+      FOR EACH ROW EXECUTE PROCEDURE tww_app.ft_vw_tww_measurement_series_INSERT();
+    """.format(
+        insert_ms=insert_command(
+            pg_cur=cursor,
+            table_schema="tww_od",
+            table_name="measurement_series",
+            table_alias="ms",
+            remove_pkey=False,
+            indent=2,
+            skip_columns=[],
+        ),
+    )
+
+    cursor.execute(trigger_insert_sql)
+
+    update_trigger_sql = """
+    CREATE OR REPLACE FUNCTION tww_app.ft_vw_tww_measurement_series_UPDATE()
+      RETURNS trigger AS
+    $BODY$
+    BEGIN
+      {update_ms}
+       RETURN NEW;
+    END;
+    $BODY$
+    LANGUAGE plpgsql;
+
+
+
+    DROP TRIGGER IF EXISTS vw_tww_measurement_series_UPDATE ON tww_app.vw_tww_measurement_series;
+
+    CREATE TRIGGER vw_tww_measurement_series_UPDATE INSTEAD OF UPDATE ON tww_app.vw_tww_measurement_series
+      FOR EACH ROW EXECUTE PROCEDURE tww_app.ft_vw_tww_measurement_series_UPDATE();
+    """.format(
+        update_ms=update_command(
+            pg_cur=cursor,
+            table_schema="tww_od",
+            table_name="measurement_series",
+            table_alias="ms",
+            indent=6,
+            skip_columns=[],
+            update_values={},
+        ),
+    )
+
+    cursor.execute(update_trigger_sql)
+
+    trigger_delete_sql = """
+    CREATE OR REPLACE FUNCTION tww_app.ft_vw_tww_measurement_series_DELETE()
+      RETURNS trigger AS
+    $BODY$
+    DECLARE
+    BEGIN
+      DELETE FROM tww_od.measurement_series WHERE obj_id = OLD.obj_id;
+    RETURN OLD;
+    END; $BODY$ LANGUAGE plpgsql VOLATILE;
+
+    DROP TRIGGER IF EXISTS vw_tww_measurement_series_DELETE ON tww_app.vw_tww_measurement_series;
+
+    CREATE TRIGGER vw_tww_measurement_series_DELETE INSTEAD OF DELETE ON tww_app.vw_tww_measurement_series
+      FOR EACH ROW EXECUTE PROCEDURE tww_app.ft_vw_tww_measurement_series_DELETE();
+    """
+
+    cursor.execute(trigger_delete_sql)
+
+    extras = """
+        ALTER VIEW tww_app.vw_tww_measurement_series ALTER obj_id SET DEFAULT tww_app.generate_oid('tww_od','measurement_series');
+    """
+    cursor.execute(extras)
+
+    conn.commit()
+    conn.close()
+
+
+if __name__ == "__main__":
+    # create the top-level parser
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument("-p", "--pg_service", help="the PostgreSQL service name")
+    args = parser.parse_args()
+    pg_service = args.pg_service or os.getenv("PGSERVICE")
+    vw_tww_measurement_series(pg_service=pg_service)