mirror of
https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
synced 2024-12-26 19:34:19 +00:00
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
"""Rename pipeline model tables
|
|
|
|
Revision ID: 63b2cc26a01f
|
|
Revises: 260b57d5f4e7
|
|
Create Date: 2022-10-11 14:32:13.227364
|
|
|
|
"""
|
|
from alembic import op
|
|
from flask import current_app
|
|
import os
|
|
from app.models import User
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = '63b2cc26a01f'
|
|
down_revision = '260b57d5f4e7'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
users = User.query.all()
|
|
for user in users:
|
|
old_tesseract_ocr_pipeline_model_path = os.path.join(
|
|
user.path,
|
|
'tesseract_ocr_models'
|
|
)
|
|
new_tesseract_ocr_pipeline_model_path = os.path.join(
|
|
user.path,
|
|
'tesseract_ocr_pipeline_models'
|
|
)
|
|
os.rename(
|
|
old_tesseract_ocr_pipeline_model_path,
|
|
new_tesseract_ocr_pipeline_model_path
|
|
)
|
|
op.rename_table('tesseract_ocr_models', 'tesseract_ocr_pipeline_models')
|
|
op.rename_table('transkribus_htr_models', 'transkribus_htr_pipeline_models')
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade():
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
users = User.query.all()
|
|
for user in users:
|
|
old_tesseract_ocr_pipeline_model_path = os.path.join(
|
|
user.path,
|
|
'tesseract_ocr_models'
|
|
)
|
|
new_tesseract_ocr_pipeline_model_path = os.path.join(
|
|
user.path,
|
|
'tesseract_ocr_pipeline_models'
|
|
)
|
|
os.rename(
|
|
new_tesseract_ocr_pipeline_model_path,
|
|
old_tesseract_ocr_pipeline_model_path
|
|
)
|
|
op.rename_table('tesseract_ocr_pipeline_models', 'tesseract_ocr_models')
|
|
op.rename_table('transkribus_htr_pipeline_models', 'transkribus_htr_models')
|
|
# ### end Alembic commands ###
|