Merge pull request maxtepkeev#44 from soyking/create_table_if_not_exists

Use CREATE TABLE IF NOT EXISTS in Postgres insert trigger
dlr-eoc · Sep 14, 2017 · d4eb81f · d4eb81f
2 parents 512ea89 + f75a471
commit d4eb81f
Showing 1 changed file with 8 additions and 12 deletions.
diff --git a/architect/databases/postgresql/partition.py b/architect/databases/postgresql/partition.py
@@ -45,18 +45,14 @@ def prepare(self):
                         {variables}
                     END IF;
 
-                    IF NOT EXISTS(
-                        SELECT 1 FROM information_schema.tables WHERE table_name=tablename)
-                    THEN
-                        BEGIN
-                            EXECUTE 'CREATE TABLE ' || tablename || ' (
-                                CHECK (' || checks || '),
-                                LIKE "{{parent_table}}" INCLUDING DEFAULTS INCLUDING CONSTRAINTS INCLUDING INDEXES
-                            ) INHERITS ("{{parent_table}}");';
-                        EXCEPTION WHEN duplicate_table THEN
-                            -- pass
-                        END;
-                    END IF;
+                    BEGIN
+                        EXECUTE 'CREATE TABLE IF NOT EXISTS ' || tablename || ' (
+                            CHECK (' || checks || '),
+                            LIKE "{{parent_table}}" INCLUDING DEFAULTS INCLUDING CONSTRAINTS INCLUDING INDEXES
+                        ) INHERITS ("{{parent_table}}");';
+                    EXCEPTION WHEN duplicate_table THEN
+                        -- pass
+                    END;
 
                     EXECUTE 'INSERT INTO ' || tablename || ' VALUES (($1).*);' USING NEW;
                     RETURN NEW;