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

View File

@ -62,17 +62,20 @@ def upgrade():
archived=archived
)
session.add(document_model)
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}')
count += 1
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')