From 88c212f3bade9ae895d6c25a075983210e7c5800 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ra=C3=BAl=20Mar=C3=ADn?= Date: Mon, 16 Nov 2020 17:31:38 +0100 Subject: [PATCH] __CDB_RegenerateTable_Get_Commands: Replace with SET LOCAL --- scripts-available/CDB_RegenerateTable.sql | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/scripts-available/CDB_RegenerateTable.sql b/scripts-available/CDB_RegenerateTable.sql index 041f186..c62aef5 100644 --- a/scripts-available/CDB_RegenerateTable.sql +++ b/scripts-available/CDB_RegenerateTable.sql @@ -6,6 +6,7 @@ CREATE OR REPLACE FUNCTION @extschema@.__CDB_RegenerateTable_Get_Commands(tableo RETURNS text[] AS $$ import subprocess + import re query = "SELECT current_database()::text as dname" rv = plpy.execute(query, 1) @@ -37,12 +38,10 @@ AS $$ sublines = [line.rstrip() for line in sublines] sublines = [line for line in sublines if line] sublines = [line for line in sublines if not line.startswith('--')] + sublines = [re.sub(r'^SET ', 'SET LOCAL ', line) for line in sublines] if len(sublines): clean_lines.append("".join(sublines)) - # Add an extra query to reset the environment - clean_lines.append("RESET ALL"); - return clean_lines $$ LANGUAGE @@plpythonu@@ VOLATILE PARALLEL UNSAFE;