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