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

Fixing a bad migration

This commit is contained in:
danfunk 2023-05-10 11:48:55 -04:00
parent 59490fc350
commit 8bf3415c34
1 changed files with 30 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
migrations/versions/3489d5a6a2c0_migrate_file_data_to_document_table.py
View File

@ -34,37 +34,37 @@ def upgrade():
# session.flush()
# migrate data from old file table and file data table to new file table
old_file_models = session.query(OldFileModel).all()
largest_file_id = 0
for old_file_model in old_file_models:
if old_file_model.irb_doc_code is not None:
largest_file_id = max(largest_file_id, old_file_model.id)
file_data_models = session.query(FileDataModel).\
filter(FileDataModel.file_model_id == old_file_model.id).\
order_by(sa.desc(FileDataModel.date_created)).\
all()
if len(file_data_models) > 0:
file_data_model = file_data_models[0]
file_model = FileModel(
id=old_file_model.id,
name=old_file_model.name,
type=old_file_model.type.value,
content_type=old_file_model.content_type,
workflow_id=old_file_model.workflow_id,
task_spec=old_file_model.task_spec,
irb_doc_code=old_file_model.irb_doc_code,
md5_hash=file_data_model.md5_hash,
data=file_data_model.data,
size=file_data_model.size,
date_modified=file_data_model.date_created,
date_created=file_data_model.date_created,
user_uid=file_data_model.user_uid,
archived=False
)
session.add(file_model)
session.commit()
# old_file_models = session.query(OldFileModel).all()
# largest_file_id = 0
# for old_file_model in old_file_models:
# if old_file_model.irb_doc_code is not None:
# largest_file_id = max(largest_file_id, old_file_model.id)
# file_data_models = session.query(FileDataModel).\
# filter(FileDataModel.file_model_id == old_file_model.id).\
# order_by(sa.desc(FileDataModel.date_created)).\
# all()
# if len(file_data_models) > 0:
# file_data_model = file_data_models[0]
# file_model = FileModel(
# id=old_file_model.id,
# name=old_file_model.name,
# type=old_file_model.type.value,
# content_type=old_file_model.content_type,
# workflow_id=old_file_model.workflow_id,
# task_spec=old_file_model.task_spec,
# irb_doc_code=old_file_model.irb_doc_code,
# md5_hash=file_data_model.md5_hash,
# data=file_data_model.data,
# size=file_data_model.size,
# date_modified=file_data_model.date_created,
# date_created=file_data_model.date_created,
# user_uid=file_data_model.user_uid,
# archived=False
# )
# session.add(file_model)
# session.commit()
sequence = FileModel.__tablename__ + '_id_seq1'
new_start_id = largest_file_id + 1
new_start_id = 1
alter_sequence = f'ALTER SEQUENCE {sequence} RESTART WITH {new_start_id}'
op.execute(alter_sequence)