Execution fails with missing foreign key error

[brki]

Experienced on the master branch, on commit 6ece4d9 .

I was able to work around this by:

• merging in the changes from Add --table argument + misc bug fixes and improvements #26
• dropping the foreign key constraints in the target db
• changing the code so that the foreign key constraints from the source db are used while determining which rows to include. Change was made in the create_subset_in method:

             if target.completeness_score() > 0.97:
                 break
             (source_row, prioritized) = target.source.next_row()
+            target.fks = self.tables[(target.schema, target.name)].fks
+            target.child_fks = self.tables[(target.schema, target.name)].child_fks
             self.create_row_in(source_row, target_db, target,
                                prioritized=prioritized)

• re-adding the foreign key constraints on the target db after subset extraction

The missing foreign key error I saw was:

Traceback (most recent call last):
  File "/path/to/code/rdbms-subsetter/subsetter.py", line 451, in <module>
    generate()
  File "/path/to/code/rdbms-subsetter/subsetter.py", line 447, in generate
    source.create_subset_in(target)
  File "/path/to/code/rdbms-subsetter/subsetter.py", line 358, in create_subset_in
    target_db.flush()
  File "/path/to/code/rdbms-subsetter/subsetter.py", line 311, in flush
    self.conn.execute(table.insert(), list(table.pending.values()))
  File "/path/to/.virtualenvs/rdbms-subsetter/lib/python2.7/site-packages/sqlalchemy/engine/base.py", line 914, in execute
    return meth(self, multiparams, params)
  File "/path/to/.virtualenvs/rdbms-subsetter/lib/python2.7/site-packages/sqlalchemy/sql/elements.py", line 323, in _execute_on_connection
    return connection._execute_clauseelement(self, multiparams, params)
  File "/path/to/.virtualenvs/rdbms-subsetter/lib/python2.7/site-packages/sqlalchemy/engine/base.py", line 1010, in _execute_clauseelement
    compiled_sql, distilled_params
  File "/path/to/.virtualenvs/rdbms-subsetter/lib/python2.7/site-packages/sqlalchemy/engine/base.py", line 1166, in _execute_context
    self._root._commit_impl(autocommit=True)
  File "/path/to/.virtualenvs/rdbms-subsetter/lib/python2.7/site-packages/sqlalchemy/engine/base.py", line 690, in _commit_impl
    self._handle_dbapi_exception(e, None, None, None, None)
  File "/path/to/.virtualenvs/rdbms-subsetter/lib/python2.7/site-packages/sqlalchemy/engine/base.py", line 1341, in _handle_dbapi_exception
    exc_info
  File "/path/to/.virtualenvs/rdbms-subsetter/lib/python2.7/site-packages/sqlalchemy/util/compat.py", line 199, in raise_from_cause
    reraise(type(exception), exception, tb=exc_tb)
  File "/path/to/.virtualenvs/rdbms-subsetter/lib/python2.7/site-packages/sqlalchemy/engine/base.py", line 688, in _commit_impl
    self.engine.dialect.do_commit(self.connection)
  File "/path/to/.virtualenvs/rdbms-subsetter/lib/python2.7/site-packages/sqlalchemy/engine/default.py", line 423, in do_commit
    dbapi_connection.commit()
sqlalchemy.exc.IntegrityError: (psycopg2.IntegrityError) insert or update on table "foo" violates foreign key constraint "foo_constaint_name"
DETAIL:  Key (bar_id)=(544e83f4-2a8a-4103-9d27-7663baadb60f) is not present in table "bar".

[brki]

I've noticed that I do sometimes still have an error, even with my described workaround. e.g when trying to re-enable the fk constraints, I see:

ERROR:  insert or update on table "foo" violates foreign key constraint "foo_constraint_name"
DETAIL:  Key (bar_id)=(77a2d8a0-23ab-4661-bd85-dc194e56d4e5) is not present in table "bar".