141 lines
8.7 KiB
Python
141 lines
8.7 KiB
Python
"""Rename all foreign key constraints to have predictable names
|
|
|
|
Revision ID: 983247d75af3
|
|
Revises: 8981dabd329f
|
|
Create Date: 2016-08-08 16:33:00.198527
|
|
|
|
"""
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = '983247d75af3'
|
|
down_revision = '8981dabd329f'
|
|
|
|
from alembic import op
|
|
from functools import wraps
|
|
import sqlalchemy as sa
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
CORRECT_FKC_NAMES = {
|
|
# Names that existed according to the constraint in previous migrations
|
|
"fk_accesstoken_kind_id_accesstokenkind",
|
|
"fk_blobupload_location_id_imagestoragelocation",
|
|
"fk_blobupload_repository_id_repository",
|
|
"fk_derivedstorageforimage_derivative_id_imagestorage",
|
|
"fk_derivedstorageforimage_source_image_id_image",
|
|
"fk_imagestoragesignature_kind_id_imagestoragesignaturekind",
|
|
"fk_imagestoragesignature_storage_id_imagestorage",
|
|
"fk_quayrelease_region_id_quayregion",
|
|
"fk_quayrelease_service_id_quayservice",
|
|
"fk_repositoryactioncount_repository_id_repository",
|
|
"fk_servicekey_approval_id_servicekeyapproval",
|
|
"fk_star_repository_id_repository",
|
|
"fk_star_user_id_user",
|
|
"fk_tagmanifest_tag_id_repositorytag",
|
|
"fk_torrentinfo_storage_id_imagestorage",
|
|
"fk_userregion_location_id_imagestoragelocation",
|
|
"fk_userregion_user_id_user",
|
|
|
|
# Names that had to be set manually as truncated versions
|
|
'fk_derivedstorageforimage_transformation_constraint',
|
|
}
|
|
|
|
|
|
def _generate_all_foreign_key_constraint_names():
|
|
conn = op.get_bind()
|
|
inspector = sa.inspect(conn.engine)
|
|
for table_name, fkc_list in inspector.get_sorted_table_and_fkc_names():
|
|
for fkc_name in fkc_list:
|
|
yield table_name, fkc_name
|
|
|
|
|
|
def _wrap_create_op(create_foreign_key_func, existing_fk_tuple_set):
|
|
fk_name_set = set()
|
|
@wraps(create_foreign_key_func)
|
|
def wrapped(fk_name, table_name, *args, **kwargs):
|
|
fk_name_set.add(fk_name)
|
|
|
|
if (table_name, fk_name) in existing_fk_tuple_set:
|
|
logger.debug('Skipping already correct fkc: %s', fk_name)
|
|
else:
|
|
logger.debug('Creating foreign key constraint: %s', fk_name)
|
|
return create_foreign_key_func(op.f(fk_name), table_name, *args, **kwargs)
|
|
return wrapped, fk_name_set
|
|
|
|
|
|
def _disable_constraints():
|
|
conn = op.get_bind()
|
|
if conn.dialect.name == 'mysql':
|
|
logger.debug('Setting mysql foreign_key_checks=0')
|
|
op.execute('SET FOREIGN_KEY_CHECKS=0')
|
|
else:
|
|
logger.warning('Unable to disable foreign key checks for dialect: %s', conn.dialect.name)
|
|
|
|
|
|
def upgrade(tables):
|
|
existing_fk_tuples = set(_generate_all_foreign_key_constraint_names())
|
|
create_fk, new_fk_name_set = _wrap_create_op(op.create_foreign_key, existing_fk_tuples)
|
|
|
|
_disable_constraints()
|
|
|
|
create_fk('fk_accesstoken_role_id_role', 'accesstoken', 'role', ['role_id'], ['id'])
|
|
create_fk('fk_accesstoken_repository_id_repository', 'accesstoken', 'repository', ['repository_id'], ['id'])
|
|
create_fk('fk_emailconfirmation_user_id_user', 'emailconfirmation', 'user', ['user_id'], ['id'])
|
|
create_fk('fk_federatedlogin_user_id_user', 'federatedlogin', 'user', ['user_id'], ['id'])
|
|
create_fk('fk_federatedlogin_service_id_loginservice', 'federatedlogin', 'loginservice', ['service_id'], ['id'])
|
|
create_fk('fk_image_repository_id_repository', 'image', 'repository', ['repository_id'], ['id'])
|
|
create_fk('fk_image_storage_id_imagestorage', 'image', 'imagestorage', ['storage_id'], ['id'])
|
|
create_fk('fk_imagestorageplacement_location_id_imagestoragelocation', 'imagestorageplacement', 'imagestoragelocation', ['location_id'], ['id'])
|
|
create_fk('fk_imagestorageplacement_storage_id_imagestorage', 'imagestorageplacement', 'imagestorage', ['storage_id'], ['id'])
|
|
create_fk('fk_logentry_kind_id_logentrykind', 'logentry', 'logentrykind', ['kind_id'], ['id'])
|
|
create_fk('fk_notification_target_id_user', 'notification', 'user', ['target_id'], ['id'])
|
|
create_fk('fk_notification_kind_id_notificationkind', 'notification', 'notificationkind', ['kind_id'], ['id'])
|
|
create_fk('fk_oauthaccesstoken_authorized_user_id_user', 'oauthaccesstoken', 'user', ['authorized_user_id'], ['id'])
|
|
create_fk('fk_oauthaccesstoken_application_id_oauthapplication', 'oauthaccesstoken', 'oauthapplication', ['application_id'], ['id'])
|
|
create_fk('fk_oauthapplication_organization_id_user', 'oauthapplication', 'user', ['organization_id'], ['id'])
|
|
create_fk('fk_oauthauthorizationcode_application_id_oauthapplication', 'oauthauthorizationcode', 'oauthapplication', ['application_id'], ['id'])
|
|
create_fk('fk_permissionprototype_delegate_team_id_team', 'permissionprototype', 'team', ['delegate_team_id'], ['id'])
|
|
create_fk('fk_permissionprototype_role_id_role', 'permissionprototype', 'role', ['role_id'], ['id'])
|
|
create_fk('fk_permissionprototype_delegate_user_id_user', 'permissionprototype', 'user', ['delegate_user_id'], ['id'])
|
|
create_fk('fk_permissionprototype_activating_user_id_user', 'permissionprototype', 'user', ['activating_user_id'], ['id'])
|
|
create_fk('fk_permissionprototype_org_id_user', 'permissionprototype', 'user', ['org_id'], ['id'])
|
|
create_fk('fk_repository_namespace_user_id_user', 'repository', 'user', ['namespace_user_id'], ['id'])
|
|
create_fk('fk_repository_visibility_id_visibility', 'repository', 'visibility', ['visibility_id'], ['id'])
|
|
create_fk('fk_repositoryauthorizedemail_repository_id_repository', 'repositoryauthorizedemail', 'repository', ['repository_id'], ['id'])
|
|
create_fk('fk_repositorybuild_access_token_id_accesstoken', 'repositorybuild', 'accesstoken', ['access_token_id'], ['id'])
|
|
create_fk('fk_repositorybuild_pull_robot_id_user', 'repositorybuild', 'user', ['pull_robot_id'], ['id'])
|
|
create_fk('fk_repositorybuild_repository_id_repository', 'repositorybuild', 'repository', ['repository_id'], ['id'])
|
|
create_fk('fk_repositorybuild_trigger_id_repositorybuildtrigger', 'repositorybuild', 'repositorybuildtrigger', ['trigger_id'], ['id'])
|
|
create_fk('fk_repositorybuildtrigger_repository_id_repository', 'repositorybuildtrigger', 'repository', ['repository_id'], ['id'])
|
|
create_fk('fk_repositorybuildtrigger_connected_user_id_user', 'repositorybuildtrigger', 'user', ['connected_user_id'], ['id'])
|
|
create_fk('fk_repositorybuildtrigger_service_id_buildtriggerservice', 'repositorybuildtrigger', 'buildtriggerservice', ['service_id'], ['id'])
|
|
create_fk('fk_repositorybuildtrigger_pull_robot_id_user', 'repositorybuildtrigger', 'user', ['pull_robot_id'], ['id'])
|
|
create_fk('fk_repositorybuildtrigger_write_token_id_accesstoken', 'repositorybuildtrigger', 'accesstoken', ['write_token_id'], ['id'])
|
|
create_fk('fk_repositorynotification_method_id_externalnotificationmethod', 'repositorynotification', 'externalnotificationmethod', ['method_id'], ['id'])
|
|
create_fk('fk_repositorynotification_repository_id_repository', 'repositorynotification', 'repository', ['repository_id'], ['id'])
|
|
create_fk('fk_repositorynotification_event_id_externalnotificationevent', 'repositorynotification', 'externalnotificationevent', ['event_id'], ['id'])
|
|
create_fk('fk_repositorypermission_role_id_role', 'repositorypermission', 'role', ['role_id'], ['id'])
|
|
create_fk('fk_repositorypermission_user_id_user', 'repositorypermission', 'user', ['user_id'], ['id'])
|
|
create_fk('fk_repositorypermission_repository_id_repository', 'repositorypermission', 'repository', ['repository_id'], ['id'])
|
|
create_fk('fk_repositorypermission_team_id_team', 'repositorypermission', 'team', ['team_id'], ['id'])
|
|
create_fk('fk_repositorytag_repository_id_repository', 'repositorytag', 'repository', ['repository_id'], ['id'])
|
|
create_fk('fk_repositorytag_image_id_image', 'repositorytag', 'image', ['image_id'], ['id'])
|
|
create_fk('fk_team_organization_id_user', 'team', 'user', ['organization_id'], ['id'])
|
|
create_fk('fk_team_role_id_teamrole', 'team', 'teamrole', ['role_id'], ['id'])
|
|
create_fk('fk_teammember_user_id_user', 'teammember', 'user', ['user_id'], ['id'])
|
|
create_fk('fk_teammember_team_id_team', 'teammember', 'team', ['team_id'], ['id'])
|
|
create_fk('fk_teammemberinvite_inviter_id_user', 'teammemberinvite', 'user', ['inviter_id'], ['id'])
|
|
create_fk('fk_teammemberinvite_team_id_team', 'teammemberinvite', 'team', ['team_id'], ['id'])
|
|
create_fk('fk_teammemberinvite_user_id_user', 'teammemberinvite', 'user', ['user_id'], ['id'])
|
|
|
|
# Drop all of the fk names that aren't correct
|
|
final_correct_fk_names = new_fk_name_set | CORRECT_FKC_NAMES
|
|
for table_name, fk_name in existing_fk_tuples:
|
|
if fk_name not in final_correct_fk_names:
|
|
logger.debug("dropping: %s", fk_name)
|
|
op.drop_constraint(fk_name, table_name, type_='foreignkey')
|
|
|
|
def downgrade(tables):
|
|
# nah
|
|
pass
|