always return union for original statement as well

toluaina · toluaina · commit d68e695aafbe · 2021-04-26T16:21:03.000+01:00
diff --git a/pgsync/base.py b/pgsync/base.py
@@ -539,7 +539,7 @@ def _foreign_key_view_statement(
             table_constraints.c.table_name
         )
 
-        unions = []
+        unions = [statement]
         if FOREIGN_KEY_VIEWNAME in views:
             values = self.fetchall(sa.select([
                 sa.column('table_name'),
@@ -580,10 +580,7 @@ def _foreign_key_view_statement(
                 )
             )
 
-        if unions:
-            statement = sa.union(*unions)
-
-        return statement
+        return sa.union(*unions)
 
     def create_views(self, schema, tables, user_defined_fkey_tables):
 

Original file line number	Diff line number	Diff line change
`@@ -539,7 +539,7 @@ def _foreign_key_view_statement(`
`539`	`539`	`table_constraints.c.table_name`
`540`	`540`	`)`
`541`	`541`
`542`		`- unions = []`
	`542`	`+ unions = [statement]`
`543`	`543`	`if FOREIGN_KEY_VIEWNAME in views:`
`544`	`544`	`values = self.fetchall(sa.select([`
`545`	`545`	`sa.column('table_name'),`
`@@ -580,10 +580,7 @@ def _foreign_key_view_statement(`
`580`	`580`	`)`
`581`	`581`	`)`
`582`	`582`
`583`		`- if unions:`
`584`		`- statement = sa.union(*unions)`
`585`		`-`
`586`		`- return statement`
	`583`	`+ return sa.union(*unions)`
`587`	`584`
`588`	`585`	`def create_views(self, schema, tables, user_defined_fkey_tables):`
`589`	`586`