sartography
/
cr-connect-workflow
mirror of https://github.com/sartography/cr-connect-workflow.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update the data_store table with document_id for each file data_store

This commit is contained in:
mike cullerton 2022-04-14 15:12:41 -04:00
parent b304b4725b
commit 7da440f30c
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
migrations/versions/3489d5a6a2c0_migrate_file_data_to_document_table.py
View File

@ -62,17 +62,20 @@ def upgrade():
archived=archived archived=archived
) )
session.add(document_model) session.add(document_model)
session.commit()
count += 1 count += 1
try: update_data_store(file_model.id, document_model.id, session)
session.commit() # try:
except IntegrityError as ie: # session.commit()
app.logger.info( # except IntegrityError as ie:
f'Error migrating file data. File ID: {file_model.id}, File Data ID: {file_data_model.id}, Original error: {ie}') # app.logger.info(
session.rollback() # f'Error migrating file data. File ID: {file_model.id}, File Data ID: {file_data_model.id}, Original error: {ie}')
except Exception as e: # session.rollback()
app.logger.info( # except Exception as e:
f'Error migrating file data. File ID: {file_model.id}, File Data ID: {file_data_model.id}, Original error: {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(): def downgrade():
op.execute('DELETE FROM document;') op.execute('DELETE FROM document;')
op.execute('UPDATE data_store SET document_id = null')