rucio 37.0.0rc1__py3-none-any.whl

rucio/db/sqla/migrate_repo/versions/44278720f774_update_requests_typ_sta_upd_idx_index.py

@@ -0,0 +1,44 @@
+# 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.
+
+''' update REQUESTS_TYP_STA_UPD_IDX index '''
+
+from alembic import context
+from alembic.op import create_index, drop_index
+
+# Alembic revision identifiers
+revision = '44278720f774'
+down_revision = '40ad39ce3160'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_index('REQUESTS_TYP_STA_UPD_IDX', 'requests')
+        create_index('REQUESTS_TYP_STA_UPD_IDX', 'requests', ['request_type', 'state', 'activity'])
+        create_index('REQUESTS_TYP_STA_UPD_IDX_OLD', 'requests', ['request_type', 'state', 'updated_at'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_index('REQUESTS_TYP_STA_UPD_IDX', 'requests')
+        drop_index('REQUESTS_TYP_STA_UPD_IDX_OLD', 'requests')
+        create_index('REQUESTS_TYP_STA_UPD_IDX', 'requests', ['request_type', 'state', 'updated_at'])

rucio/db/sqla/migrate_repo/versions/45378a1e76a8_create_collection_replica_table.py

@@ -0,0 +1,78 @@
+# 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 collection replica table '''
+
+import datetime
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import create_check_constraint, create_foreign_key, create_index, create_primary_key, create_table, drop_table
+
+from rucio.db.sqla.constants import DIDType, ReplicaState
+from rucio.db.sqla.types import GUID
+from rucio.db.sqla.util import try_drop_constraint
+
+# Alembic revision identifiers
+revision = '45378a1e76a8'
+down_revision = 'a93e4e47bda'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_table('collection_replicas',
+                     sa.Column('scope', sa.String(25)),
+                     sa.Column('name', sa.String(255)),
+                     sa.Column('did_type', sa.Enum(DIDType,
+                                                   name='COLLECTION_REPLICAS_TYPE_CHK',
+                                                   create_constraint=True,
+                                                   values_callable=lambda obj: [e.value for e in obj])),
+                     sa.Column('rse_id', GUID()),
+                     sa.Column('bytes', sa.BigInteger),
+                     sa.Column('length', sa.BigInteger),
+                     sa.Column('state', sa.Enum(ReplicaState,
+                                                name='COLLECTION_REPLICAS_STATE_CHK',
+                                                create_constraint=True,
+                                                values_callable=lambda obj: [e.value for e in obj]),
+                               default=ReplicaState.UNAVAILABLE),
+                     sa.Column('accessed_at', sa.DateTime),
+                     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('COLLECTION_REPLICAS_PK', 'collection_replicas', ['scope', 'name', 'rse_id'])
+        create_foreign_key('COLLECTION_REPLICAS_LFN_FK', 'collection_replicas', 'dids', ['scope', 'name'], ['scope', 'name'])
+        create_foreign_key('COLLECTION_REPLICAS_RSE_ID_FK', 'collection_replicas', 'rses', ['rse_id'], ['id'])
+        create_check_constraint('COLLECTION_REPLICAS_SIZE_NN', 'collection_replicas', 'bytes IS NOT NULL')
+        create_check_constraint('COLLECTION_REPLICAS_STATE_NN', 'collection_replicas', 'state IS NOT NULL')
+        create_index('COLLECTION_REPLICAS_RSE_ID_IDX', 'collection_replicas', ['rse_id'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name == 'oracle':
+        try_drop_constraint('COLLECTION_REPLICAS_STATE_CHK', 'collection_replicas')
+        drop_table('collection_replicas')
+
+    elif context.get_context().dialect.name == 'postgresql':
+        drop_table('collection_replicas')
+
+    elif context.get_context().dialect.name == 'mysql':
+        drop_table('collection_replicas')

rucio/db/sqla/migrate_repo/versions/469d262be19_removing_created_at_index.py

@@ -0,0 +1,41 @@
+# 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.
+
+''' removing created_at index '''
+
+from alembic import context
+from alembic.op import create_index, drop_index
+
+revision = '469d262be19'
+down_revision = '16a0aca82e12'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_index('UPDATED_DIDS_SCOPERULENAME_IDX', 'updated_dids', ['scope', 'rule_evaluation_action', 'name'])
+        drop_index('CREATED_AT_IDX', 'updated_dids')
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_index('UPDATED_DIDS_SCOPERULENAME_IDX', 'updated_dids')
+        create_index('CREATED_AT_IDX', 'updated_dids', ['created_at'])

rucio/db/sqla/migrate_repo/versions/4783c1f49cb4_create_distance_table.py

@@ -0,0 +1,59 @@
+# 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 distance table '''
+
+import datetime
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import create_check_constraint, create_foreign_key, create_index, create_primary_key, create_table, drop_table
+
+from rucio.db.sqla.types import GUID
+
+# Alembic revision identifiers
+revision = '4783c1f49cb4'
+down_revision = '277b5fbb41d3'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_table('distances',
+                     sa.Column('src_rse_id', GUID()),
+                     sa.Column('dest_rse_id', GUID()),
+                     sa.Column('ranking', sa.Integer),
+                     sa.Column('agis_distance', sa.Integer),
+                     sa.Column('geoip_distance', sa.Integer),
+                     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('DISTANCES_PK', 'distances', ['src_rse_id', 'dest_rse_id'])
+        create_foreign_key('DISTANCES_SRC_RSES_FK', 'distances', 'rses', ['src_rse_id'], ['id'])
+        create_foreign_key('DISTANCES_DEST_RSES_FK', 'distances', 'rses', ['dest_rse_id'], ['id'])
+        create_check_constraint('DISTANCES_CREATED_NN', 'distances', 'created_at is not null')
+        create_check_constraint('DISTANCES_UPDATED_NN', 'distances', 'updated_at is not null')
+        create_index('DISTANCES_DEST_RSEID_IDX', 'distances', ['dest_rse_id'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_table('distances')

rucio/db/sqla/migrate_repo/versions/49a21b4d4357_create_index_on_table_tokens.py

@@ -0,0 +1,44 @@
+# 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 tokens index '''
+
+from alembic import context
+from alembic.op import create_foreign_key, create_index, drop_constraint, drop_index
+
+# Alembic revision identifiers
+revision = '49a21b4d4357'
+down_revision = '2eef46be23d4'
+
+
+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')
+        create_index('TOKENS_ACCOUNT_EXPIRED_AT_IDX', 'tokens', ['account', 'expired_at'])
+        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_index('TOKENS_ACCOUNT_EXPIRED_AT_IDX', 'tokens')
+        create_foreign_key('TOKENS_ACCOUNT_FK', 'tokens', 'accounts', ['account'], ['account'])

rucio/db/sqla/migrate_repo/versions/4a2cbedda8b9_add_source_replica_expression_column_to_.py

@@ -0,0 +1,43 @@
+# 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_replica_expression_column_to_rules '''
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import add_column, drop_column
+
+# Alembic revision identifiers
+revision = '4a2cbedda8b9'
+down_revision = 'a616581ee47'
+
+
+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('rules', sa.Column('source_replica_expression', sa.String(255)), 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('rules', 'source_replica_expression', schema=schema)

rucio/db/sqla/migrate_repo/versions/4a7182d9578b_added_bytes_length_accessed_at_columns.py

@@ -0,0 +1,49 @@
+# 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 bytes, length, accessed_at columns '''
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import add_column, drop_column
+
+# Alembic revision identifiers
+revision = '4a7182d9578b'
+down_revision = 'c129ccdb2d5'
+
+
+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('dataset_locks', sa.Column('length', sa.BigInteger()), schema=schema)
+        add_column('dataset_locks', sa.Column('bytes', sa.BigInteger()), schema=schema)
+        add_column('dataset_locks', sa.Column('accessed_at', sa.DateTime()), schema=schema)
+        add_column('dids', sa.Column('accessed_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('dataset_locks', 'length', schema=schema)
+        drop_column('dataset_locks', 'bytes', schema=schema)
+        drop_column('dataset_locks', 'accessed_at', schema=schema)
+        drop_column('dids', 'accessed_at', schema=schema)

rucio/db/sqla/migrate_repo/versions/4bab9edd01fc_create_index_on_requests_rule_id.py

@@ -0,0 +1,40 @@
+# 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 index on requests.rule_id '''
+
+from alembic import context
+from alembic.op import create_index, drop_index
+
+# Alembic revision identifiers
+revision = '4bab9edd01fc'
+down_revision = 'ae2a56fcc89'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_index('REQUESTS_RULEID_IDX', 'requests', ['rule_id'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_index('REQUESTS_RULEID_IDX', 'requests')

rucio/db/sqla/migrate_repo/versions/4c3a4acfe006_new_attr_account_table.py

@@ -0,0 +1,63 @@
+# 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.
+
+''' new attr account table '''
+
+import datetime
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import create_check_constraint, create_foreign_key, create_index, create_primary_key, create_table, drop_table
+
+# Alembic revision identifiers
+revision = '4c3a4acfe006'
+down_revision = '25fc855625cf'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_table('account_attr_map',
+                     sa.Column('account', sa.String(25)),
+                     sa.Column('key', sa.String(255)),
+                     sa.Column('value', sa.String(255)),
+                     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('ACCOUNT_ATTR_MAP_PK', 'account_attr_map', ['account', 'key'])
+        create_check_constraint('ACCOUNT_ATTR_MAP_CREATED_NN', 'account_attr_map', 'created_at is not null')
+        create_check_constraint('ACCOUNT_ATTR_MAP_UPDATED_NN', 'account_attr_map', 'updated_at is not null')
+        create_foreign_key('ACCOUNT_ATTR_MAP_ACCOUNT_FK', 'account_attr_map', 'accounts', ['account'], ['account'])
+        create_index('ACCOUNT_ATTR_MAP_KEY_VALUE_IDX', 'account_attr_map', ['key', 'value'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_table('account_attr_map')
+
+    elif context.get_context().dialect.name == 'postgresql':
+        # drop_constraint('ACCOUNT_ATTR_MAP_PK', 'account_attr_map', type_='primary')
+        # drop_constraint('ACCOUNT_ATTR_MAP_CREATED_NN', 'account_attr_map')
+        # drop_constraint('ACCOUNT_ATTR_MAP_UPDATED_NN', 'account_attr_map')
+        # drop_constraint('ACCOUNT_ATTR_MAP_ACCOUNT_FK', 'account_attr_map')
+        # drop_index('ACCOUNT_ATTR_MAP_KEY_VALUE_IDX', 'account_attr_map')
+        # drop_table('account_attr_map')
+        pass

rucio/db/sqla/migrate_repo/versions/4cf0a2e127d4_adding_transient_metadata.py

@@ -0,0 +1,43 @@
+# 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.
+
+''' adding transient metadata '''
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import add_column, drop_column
+
+# Alembic revision identifiers
+revision = '4cf0a2e127d4'
+down_revision = '271a46ea6244'
+
+
+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('transient', sa.Boolean(name='DID_TRANSIENT_CHK', create_constraint=True), server_default='0'), 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('dids', 'transient', schema=schema)

rucio/db/sqla/migrate_repo/versions/4df2c5ddabc0_remove_temporary_dids.py

@@ -0,0 +1,55 @@
+# 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.
+
+''' remove temporary dids '''
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import create_index, create_primary_key, create_table, drop_table
+
+from rucio.common.schema import get_schema_value
+from rucio.db.sqla.types import GUID, InternalScopeString
+
+# Alembic revision identifiers
+revision = '4df2c5ddabc0'
+down_revision = '27e3a68927fb'
+
+
+def upgrade():
+    '''Upgrade the database to this revision'''
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_table('tmp_dids')
+
+
+def downgrade():
+    '''Downgrade the database to the previous revision'''
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_table('tmp_dids',
+                     sa.Column('scope', InternalScopeString(get_schema_value('SCOPE_LENGTH'))),
+                     sa.Column('name', sa.String(get_schema_value('NAME_LENGTH'))),
+                     sa.Column('rse_id', GUID()),
+                     sa.Column('path', sa.String(1024)),
+                     sa.Column('bytes', sa.BigInteger),
+                     sa.Column('md5', sa.String(32)),
+                     sa.Column('adler32', sa.String(8)),
+                     sa.Column('expired_at', sa.DateTime),
+                     sa.Column('guid', GUID()),
+                     sa.Column('events', sa.BigInteger),
+                     sa.Column('task_id', sa.Integer),
+                     sa.Column('panda_id', sa.Integer),
+                     sa.Column('parent_scope', InternalScopeString(get_schema_value('SCOPE_LENGTH'))),
+                     sa.Column('parent_name', sa.String(get_schema_value('NAME_LENGTH'))),
+                     sa.Column('offset', sa.BigInteger))
+        create_primary_key('TMP_DIDS_PK', 'tmp_dids', ['scope', 'name'])
+        create_index('TMP_DIDS_EXPIRED_AT_IDX', 'tmp_dids', ['expired_at'])

rucio/db/sqla/migrate_repo/versions/50280c53117c_add_qos_class_to_rse.py

@@ -0,0 +1,45 @@
+# 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 qos class to rse '''
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import add_column, drop_column
+
+# Alembic revision identifiers
+revision = '50280c53117c'
+down_revision = 'c0937668555f'
+
+
+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 in ['oracle', 'postgresql', 'mysql']:
+        add_column('rses', sa.Column('qos_class', sa.String(64)), schema=schema[:-1])
+
+
+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 in ['oracle', 'postgresql', 'mysql']:
+        drop_column('rses', 'qos_class', schema=schema[:-1])

rucio/db/sqla/migrate_repo/versions/52153819589c_add_rse_id_to_replicas_table.py

@@ -0,0 +1,43 @@
+# 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_rse_id_to_replicas_table '''
+
+
+from alembic import context
+from alembic.op import create_foreign_key, create_index, drop_constraint, drop_index
+
+# Alembic revision identifiers
+revision = '52153819589c'
+down_revision = '30fa38b6434e'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    create_index('REPLICAS_RSE_ID_IDX', 'replicas', ['rse_id'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['mysql']:
+        drop_constraint('REPLICAS_RSE_ID_FK', 'replicas', type_='foreignkey')
+    drop_index('REPLICAS_RSE_ID_IDX', 'replicas')
+    if context.get_context().dialect.name in ['mysql']:
+        create_foreign_key('REPLICAS_RSE_ID_FK', 'replicas', 'rses', ['rse_id'], ['id'])