rucio 32.8.6__py3-none-any.whl

rucio/db/sqla/migrate_repo/versions/2b8e7bcb4783_add_config_table.py

@@ -0,0 +1,72 @@
+# -*- coding: utf-8 -*-
+# Copyright European Organization for Nuclear Research (CERN) since 2012
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+''' add config table '''
+
+import datetime
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import (create_table, create_primary_key,
+                        create_check_constraint,
+                        drop_constraint, drop_table)
+
+# Alembic revision identifiers
+revision = '2b8e7bcb4783'
+down_revision = 'd91002c5841'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_table('configs',
+                     sa.Column('section', sa.String(128)),
+                     sa.Column('opt', sa.String(128)),
+                     sa.Column('value', sa.String(4000)),
+                     sa.Column('created_at', sa.DateTime, default=datetime.datetime.utcnow),
+                     sa.Column('updated_at', sa.DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow))
+
+        create_primary_key('configs_pk', 'configs', ['section', 'opt'])
+        create_check_constraint('configs_created_nn', 'configs', 'created_at is not null')
+        create_check_constraint('configs_updated_nn', 'configs', 'updated_at is not null')
+        create_table('configs_history',
+                     sa.Column('section', sa.String(128)),
+                     sa.Column('opt', sa.String(128)),
+                     sa.Column('value', sa.String(4000)),
+                     sa.Column('created_at', sa.DateTime),
+                     sa.Column('updated_at', sa.DateTime))
+
+        create_primary_key('configs_history_pk', 'configs_history', ['section', 'opt', 'updated_at'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_table('configs')
+        drop_table('configs_history')
+
+    elif context.get_context().dialect.name == 'postgresql':
+        drop_constraint('configs_pk', 'configs', type_='primary')
+        drop_constraint('configs_created_nn', 'configs', type_='check')
+        drop_constraint('configs_updated_nn', 'configs', type_='check')
+        drop_table('configs')
+        drop_constraint('configs_history_pk', 'configs_history', type_='check')
+        drop_table('configs_history')

rucio/db/sqla/migrate_repo/versions/2ba5229cb54c_add_submitted_at_to_requests_table.py

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+# Copyright European Organization for Nuclear Research (CERN) since 2012
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+''' add_submitted_at_to_requests_table '''
+
+import sqlalchemy as sa
+
+from alembic import context
+from alembic.op import add_column, drop_column
+
+
+# Alembic revision identifiers
+revision = '2ba5229cb54c'
+down_revision = '22d887e4ec0a'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        add_column('requests', sa.Column('submitted_at', sa.DateTime()), schema=schema)
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        drop_column('requests', 'submitted_at', schema=schema)

rucio/db/sqla/migrate_repo/versions/2cbee484dcf9_added_column_volume_to_rse_transfer_.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# Copyright European Organization for Nuclear Research (CERN) since 2012
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+''' added column volume to rse_transfer_limits '''
+
+import sqlalchemy as sa
+
+from alembic import context
+from alembic.op import (drop_column, add_column)
+
+
+# Alembic revision identifiers
+revision = '2cbee484dcf9'
+down_revision = 'cebad904c4dd'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        add_column('rse_transfer_limits', sa.Column('volume', sa.BigInteger), schema=schema)
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        drop_column('rse_transfer_limits', 'volume', schema=schema)

rucio/db/sqla/migrate_repo/versions/2edee4a83846_add_source_to_requests_and_requests_.py

@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+# Copyright European Organization for Nuclear Research (CERN) since 2012
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+''' add source to requests and requests_history '''
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import add_column, drop_column
+
+from rucio.db.sqla.types import GUID
+
+# Alembic revision identifiers
+revision = '2edee4a83846'
+down_revision = '2f648fc909f3'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        add_column('requests', sa.Column('source_rse_id', GUID()), schema=schema)
+        add_column('requests_history', sa.Column('source_rse_id', GUID()), schema=schema)
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        drop_column('requests', 'source_rse_id', schema=schema)
+        drop_column('requests_history', 'source_rse_id', schema=schema)

rucio/db/sqla/migrate_repo/versions/2eef46be23d4_change_tokens_pk.py

@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+# Copyright European Organization for Nuclear Research (CERN) since 2012
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+''' change tokens pk '''
+
+from alembic import context
+from alembic.op import create_primary_key, create_foreign_key, drop_constraint
+
+
+# Alembic revision identifiers
+revision = '2eef46be23d4'
+down_revision = '58c8b78301ab'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_constraint('TOKENS_ACCOUNT_FK', 'tokens', type_='foreignkey')
+        drop_constraint('TOKENS_PK', 'tokens', type_='primary')
+        create_primary_key('TOKENS_PK', 'tokens', ['token'])
+        create_foreign_key('TOKENS_ACCOUNT_FK', 'tokens', 'accounts', ['account'], ['account'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_constraint('TOKENS_ACCOUNT_FK', 'tokens', type_='foreignkey')
+        drop_constraint('TOKENS_PK', 'tokens', type_='primary')
+        create_primary_key('TOKENS_PK', 'tokens', ['account', 'token'])
+        create_foreign_key('TOKENS_ACCOUNT_FK', 'tokens', 'accounts', ['account'], ['account'])

rucio/db/sqla/migrate_repo/versions/2f648fc909f3_index_in_rule_history_on_scope_name.py

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+# Copyright European Organization for Nuclear Research (CERN) since 2012
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+''' index in rule_history on scope, name '''
+
+from alembic import context
+from alembic.op import create_index, drop_index
+
+
+# Alembic revision identifiers
+revision = '2f648fc909f3'
+down_revision = 'bb695f45c04'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_index('RULES_HISTORY_SCOPENAME_IDX', 'rules_history', ['scope', 'name'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_index('RULES_HISTORY_SCOPENAME_IDX', 'rules_history')

rucio/db/sqla/migrate_repo/versions/3082b8cef557_add_naming_convention_table_and_closed_.py

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+# Copyright European Organization for Nuclear Research (CERN) since 2012
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+''' add convention table and closed_at to dids '''
+
+import datetime
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import (create_table, create_primary_key, create_foreign_key, add_column,
+                        create_check_constraint, drop_column, drop_table)
+
+from rucio.common.schema import get_schema_value
+from rucio.db.sqla.constants import KeyType
+
+# Alembic revision identifiers
+revision = '3082b8cef557'
+down_revision = '269fee20dee9'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        add_column('dids', sa.Column('closed_at', sa.DateTime), schema=schema)
+        add_column('contents_history', sa.Column('deleted_at', sa.DateTime), schema=schema)
+        create_table('naming_conventions',
+                     sa.Column('scope', sa.String(get_schema_value('SCOPE_LENGTH'))),
+                     sa.Column('regexp', sa.String(255)),
+                     sa.Column('convention_type', sa.Enum(KeyType,
+                                                          name='CVT_TYPE_CHK',
+                                                          create_constraint=True,
+                                                          values_callable=lambda obj: [e.value for e in obj])),
+                     sa.Column('updated_at', sa.DateTime, default=datetime.datetime.utcnow),
+                     sa.Column('created_at', sa.DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow))
+        create_primary_key('NAMING_CONVENTIONS_PK', 'naming_conventions', ['scope'])
+        create_foreign_key('NAMING_CONVENTIONS_SCOPE_FK', 'naming_conventions',
+                           'scopes', ['scope'], ['scope'])
+        create_check_constraint('NAMING_CONVENTIONS_CREATED_NN', 'naming_conventions',
+                                'created_at is not null')
+        create_check_constraint('NAMING_CONVENTIONS_UPDATED_NN', 'naming_conventions',
+                                'updated_at is not null')
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        drop_column('dids', 'closed_at', schema=schema)
+        drop_column('contents_history', 'deleted_at', schema=schema)
+        drop_table('naming_conventions', schema=schema)

rucio/db/sqla/migrate_repo/versions/30fa38b6434e_add_index_on_service_column_in_the_message_table.py

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+# Copyright European Organization for Nuclear Research (CERN) since 2012
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+''' Add index on service column in the message table '''
+import sqlalchemy as sa
+
+from alembic import context
+from alembic.op import create_index, drop_index, alter_column
+
+# Alembic revision identifiers
+revision = '30fa38b6434e'
+down_revision = 'e138c364ebd0'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        alter_column('messages', 'services', existing_type=sa.String(2048), type_=sa.String(256), schema=schema)
+        alter_column('messages', 'event_type', existing_type=sa.String(1024), type_=sa.String(256), schema=schema)
+    create_index('MESSAGES_SERVICES_IDX', 'messages', ['services', 'event_type'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+    drop_index('MESSAGES_SERVICES_IDX', 'messages')
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        alter_column('messages', 'services', existing_type=sa.String(256), type_=sa.String(2048), schema=schema)
+        alter_column('messages', 'event_type', existing_type=sa.String(256), type_=sa.String(1024), schema=schema)

rucio/db/sqla/migrate_repo/versions/3152492b110b_added_staging_area_column.py

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+# Copyright European Organization for Nuclear Research (CERN) since 2012
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+''' added staging_area column '''
+
+import sqlalchemy as sa
+from alembic import context, op
+from alembic.op import add_column, create_check_constraint, drop_constraint, drop_column
+
+from rucio.db.sqla.util import try_drop_constraint
+
+# Alembic revision identifiers
+revision = '3152492b110b'
+down_revision = '22cf51430c78'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    schema = context.get_context().version_table_schema + '.' if context.get_context().version_table_schema else ''
+
+    if context.get_context().dialect.name == 'oracle':
+        add_column('rses', sa.Column('staging_area', sa.Boolean(name='RSE_STAGING_AREA_CHK', create_constraint=True), default=False))
+        try_drop_constraint('REQUESTS_TYPE_CHK', 'requests')
+        create_check_constraint(constraint_name='REQUESTS_TYPE_CHK', table_name='requests',
+                                condition="request_type in ('U', 'D', 'T', 'I', '0')")
+
+    elif context.get_context().dialect.name == 'postgresql':
+        add_column('rses', sa.Column('staging_area', sa.Boolean(name='RSE_STAGING_AREA_CHK', create_constraint=True), default=False), schema=schema[:-1])
+        drop_constraint('REQUESTS_TYPE_CHK', 'requests', type_='check')
+        create_check_constraint(constraint_name='REQUESTS_TYPE_CHK', table_name='requests',
+                                condition="request_type in ('U', 'D', 'T', 'I', '0')")
+
+    elif context.get_context().dialect.name == 'mysql':
+        add_column('rses', sa.Column('staging_area', sa.Boolean(name='RSE_STAGING_AREA_CHK', create_constraint=True), default=False), schema=schema[:-1])
+        op.execute('ALTER TABLE ' + schema + 'requests DROP CHECK REQUESTS_TYPE_CHK')  # pylint: disable=no-member
+        create_check_constraint(constraint_name='REQUESTS_TYPE_CHK', table_name='requests',
+                                condition="request_type in ('U', 'D', 'T', 'I', '0')")
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    schema = context.get_context().version_table_schema + '.' if context.get_context().version_table_schema else ''
+
+    if context.get_context().dialect.name == 'oracle':
+        try_drop_constraint('RSE_STAGING_AREA_CHK', 'rses')
+        try_drop_constraint('REQUESTS_TYPE_CHK', 'requests')
+        create_check_constraint(constraint_name='REQUESTS_TYPE_CHK', table_name='requests',
+                                condition="request_type in ('U', 'D', 'T')")
+        drop_column('rses', 'staging_area')
+
+    elif context.get_context().dialect.name == 'postgresql':
+        op.execute('ALTER TABLE ' + schema + 'requests DROP CONSTRAINT IF EXISTS "REQUESTS_TYPE_CHK", ALTER COLUMN request_type TYPE CHAR')  # pylint: disable=no-member
+        create_check_constraint(constraint_name='REQUESTS_TYPE_CHK', table_name='requests',
+                                condition="request_type in ('U', 'D', 'T')")
+        drop_column('rses', 'staging_area', schema=schema[:-1])
+
+    elif context.get_context().dialect.name == 'mysql':
+        create_check_constraint(constraint_name='REQUESTS_TYPE_CHK', table_name='requests',
+                                condition="request_type in ('U', 'D', 'T')")
+        drop_column('rses', 'staging_area', schema=schema[:-1])

rucio/db/sqla/migrate_repo/versions/32c7d2783f7e_create_bad_replicas_table.py

@@ -0,0 +1,62 @@
+# -*- coding: utf-8 -*-
+# Copyright European Organization for Nuclear Research (CERN) since 2012
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+''' create bad replicas table '''
+
+import datetime
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import (create_table, create_primary_key, create_foreign_key,
+                        create_check_constraint, create_index, drop_table)
+
+from rucio.db.sqla.types import GUID
+
+# Alembic revision identifiers
+revision = '32c7d2783f7e'
+down_revision = '384b96aa0f60'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_table('bad_replicas',
+                     sa.Column('scope', sa.String(25)),
+                     sa.Column('name', sa.String(255)),
+                     sa.Column('rse_id', GUID()),
+                     sa.Column('reason', sa.String(255)),
+                     sa.Column('state', sa.String(1)),
+                     sa.Column('account', sa.String(25)),
+                     sa.Column('created_at', sa.DateTime, default=datetime.datetime.utcnow),
+                     sa.Column('updated_at', sa.DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow))
+
+        create_primary_key('BAD_REPLICAS_STATE_PK', 'bad_replicas', ['scope', 'name', 'rse_id', 'created_at'])
+        create_check_constraint('BAD_REPLICAS_SCOPE_NN', 'bad_replicas', 'scope is not null')
+        create_check_constraint('BAD_REPLICAS_NAME_NN', 'bad_replicas', 'name is not null')
+        create_check_constraint('BAD_REPLICAS_RSE_ID_NN', 'bad_replicas', 'rse_id is not null')
+        create_foreign_key('BAD_REPLICAS_ACCOUNT_FK', 'bad_replicas', 'accounts', ['account'], ['account'])
+        create_index('BAD_REPLICAS_STATE_IDX', 'bad_replicas', ['rse_id', 'state'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_table('bad_replicas')

rucio/db/sqla/migrate_repo/versions/3345511706b8_replicas_table_pk_definition_is_in_.py

@@ -0,0 +1,74 @@
+# -*- coding: utf-8 -*-
+# Copyright European Organization for Nuclear Research (CERN) since 2012
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+''' replicas table PK definition is in wrong order '''
+
+from alembic import context
+from alembic.op import create_primary_key, drop_constraint, create_foreign_key, drop_index
+
+
+# revision identifiers used by alembic
+revision = '3345511706b8'
+down_revision = '01eaf73ab656'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'postgresql']:
+        drop_constraint('SOURCES_REPLICA_FK', 'sources', type_='foreignkey')
+        drop_constraint('REPLICAS_PK', 'replicas', type_='primary')
+        create_primary_key('REPLICAS_PK', 'replicas', ['scope', 'name', 'rse_id'])
+        create_foreign_key('SOURCES_REPLICA_FK', 'sources', 'replicas', ['scope', 'name', 'rse_id'], ['scope', 'name', 'rse_id'])
+
+    elif context.get_context().dialect.name == 'mysql':
+        drop_constraint('SOURCES_REPLICA_FK', 'sources', type_='foreignkey')
+        # The constraint has an internal index which is not automatically dropped,
+        # we have to do that manually
+        drop_index('SOURCES_REPLICA_FK', 'sources')
+        drop_constraint(constraint_name='REPLICAS_LFN_FK', table_name='replicas', type_='foreignkey')
+        drop_constraint(constraint_name='REPLICAS_RSE_ID_FK', table_name='replicas', type_='foreignkey')
+        drop_constraint('REPLICAS_PK', 'replicas', type_='primary')
+        create_foreign_key('REPLICAS_LFN_FK', 'replicas', 'dids', ['scope', 'name'], ['scope', 'name'])
+        create_foreign_key('REPLICAS_RSE_ID_FK', 'replicas', 'rses', ['rse_id'], ['id'])
+        create_primary_key('REPLICAS_PK', 'replicas', ['scope', 'name', 'rse_id'])
+        create_foreign_key('SOURCES_REPLICA_FK', 'sources', 'replicas', ['scope', 'name', 'rse_id'], ['scope', 'name', 'rse_id'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'postgresql']:
+        drop_constraint(constraint_name='SOURCES_REPLICA_FK', table_name='sources', type_='foreignkey')
+        drop_constraint(constraint_name='REPLICAS_PK', table_name='replicas', type_='primary')
+        create_primary_key('REPLICAS_PK', 'replicas', ['rse_id', 'scope', 'name'])
+        create_foreign_key('SOURCES_REPLICA_FK', 'sources', 'replicas', ['rse_id', 'scope', 'name'], ['rse_id', 'scope', 'name'])
+
+    elif context.get_context().dialect.name == 'mysql':
+        drop_constraint(constraint_name='SOURCES_REPLICA_FK', table_name='sources', type_='foreignkey')
+        # The constraint has an internal index which is not automatically dropped,
+        # we have to do that manually
+        drop_index('SOURCES_REPLICA_FK', 'sources')
+        drop_constraint(constraint_name='REPLICAS_LFN_FK', table_name='replicas', type_='foreignkey')
+        drop_constraint(constraint_name='REPLICAS_RSE_ID_FK', table_name='replicas', type_='foreignkey')
+        drop_constraint(constraint_name='REPLICAS_PK', table_name='replicas', type_='primary')
+        create_foreign_key('REPLICAS_LFN_FK', 'replicas', 'dids', ['scope', 'name'], ['scope', 'name'])
+        create_foreign_key('REPLICAS_RSE_ID_FK', 'replicas', 'rses', ['rse_id'], ['id'])
+        create_primary_key('REPLICAS_PK', 'replicas', ['rse_id', 'scope', 'name'])
+        create_foreign_key('SOURCES_REPLICA_FK', 'sources', 'replicas', ['rse_id', 'scope', 'name'], ['rse_id', 'scope', 'name'])