api - change trigger target function type

Target of trigger cannot be procedure, must be function that returns type trigger.
godon-dev · Nov 15, 2023 · 9e0aaba · 9e0aaba
1 parent e01bb6a
commit 9e0aaba
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/api/archive_db.py b/api/archive_db.py
@@ -54,9 +54,7 @@ def create_trigger(trigger_name=None, table_name=None, procedure_name=None):
     @staticmethod
     def create_procedure(procedure_name=None, probability=1.0, source_table_name=None, target_table_name=None):
         query = f"""
-        CREATE OR REPLACE PROCEDURE {procedure_name}()
-        LANGUAGE plpgsql
-        AS $body$
+        CREATE OR REPLACE FUNCTION {procedure_name}() RETURN TRIGGER AS $$
         DECLARE
           random_value real;
         BEGIN
@@ -71,7 +69,7 @@ def create_procedure(procedure_name=None, probability=1.0, source_table_name=Non
           END IF;
 
         END;
-        $body$;
+        $$ LANGUAGE plpgsql;
         """
 
         return query