From 7da440f30cbca7fa669d6801876ac0c29b6fe5cc Mon Sep 17 00:00:00 2001 From: mike cullerton Date: Thu, 14 Apr 2022 15:12:41 -0400 Subject: [PATCH] Update the data_store table with document_id for each file data_store --- ...2c0_migrate_file_data_to_document_table.py | 21 +++++++++++-------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/migrations/versions/3489d5a6a2c0_migrate_file_data_to_document_table.py b/migrations/versions/3489d5a6a2c0_migrate_file_data_to_document_table.py index 2b488435..6d4fe81a 100644 --- a/migrations/versions/3489d5a6a2c0_migrate_file_data_to_document_table.py +++ b/migrations/versions/3489d5a6a2c0_migrate_file_data_to_document_table.py @@ -62,17 +62,20 @@ def upgrade(): archived=archived ) session.add(document_model) + session.commit() count += 1 - try: - session.commit() - except IntegrityError as ie: - app.logger.info( - f'Error migrating file data. File ID: {file_model.id}, File Data ID: {file_data_model.id}, Original error: {ie}') - session.rollback() - except Exception as e: - app.logger.info( - f'Error migrating file data. File ID: {file_model.id}, File Data ID: {file_data_model.id}, Original error: {e}') + update_data_store(file_model.id, document_model.id, session) + # try: + # session.commit() + # except IntegrityError as ie: + # app.logger.info( + # f'Error migrating file data. File ID: {file_model.id}, File Data ID: {file_data_model.id}, Original error: {ie}') + # session.rollback() + # except Exception as e: + # app.logger.info( + # f'Error migrating file data. File ID: {file_model.id}, File Data ID: {file_data_model.id}, Original error: {e}') def downgrade(): op.execute('DELETE FROM document;') + op.execute('UPDATE data_store SET document_id = null')