rucio 32.8.6__py3-none-any.whl → 35.8.0__py3-none-any.whl

rucio/db/sqla/migrate_repo/versions/52fd9f4916fa_added_activity_to_rules.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,11 +15,9 @@
 ''' added share to rules '''
 
 import sqlalchemy as sa
-
 from alembic import context
 from alembic.op import add_column, drop_column
 
-
 # Alembic revision identifiers
 revision = '52fd9f4916fa'
 down_revision = '4a2cbedda8b9'

rucio/db/sqla/migrate_repo/versions/53b479c3cb0f_fix_did_meta_table_missing_updated_at_.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,11 +15,9 @@
 ''' fix did_meta table missing updated_at, created_at columns '''
 
 import sqlalchemy as sa
-
 from alembic import context
 from alembic.op import add_column, drop_column
 
-
 # Alembic revision identifiers
 revision = '53b479c3cb0f'
 down_revision = '2cbee484dcf9'

rucio/db/sqla/migrate_repo/versions/5673b4b6e843_add_wfms_metadata_to_rule_tables.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,11 +15,9 @@
 ''' add metadata to rule tables '''
 
 import sqlalchemy as sa
-
 from alembic import context
 from alembic.op import add_column, drop_column
 
-
 # Alembic revision identifiers
 revision = '5673b4b6e843'
 down_revision = 'e59300c8b179'

rucio/db/sqla/migrate_repo/versions/575767d9f89_added_source_history_table.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -17,7 +16,7 @@
 
 import sqlalchemy as sa
 from alembic import context
-from alembic.op import create_table, add_column, drop_column, drop_table
+from alembic.op import add_column, create_table, drop_column, drop_table
 
 from rucio.db.sqla.types import GUID
 

rucio/db/sqla/migrate_repo/versions/58bff7008037_add_started_at_to_requests.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,11 +15,9 @@
 ''' add started_at to requests '''
 
 import sqlalchemy as sa
-
 from alembic import context
 from alembic.op import add_column, drop_column
 
-
 # Alembic revision identifiers
 revision = '58bff7008037'
 down_revision = '3c9df354071b'

rucio/db/sqla/migrate_repo/versions/58c8b78301ab_rename_callback_to_message.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,8 +15,7 @@
 ''' rename callback to message '''
 
 from alembic import context, op
-from alembic.op import (create_primary_key, create_check_constraint,
-                        drop_constraint, rename_table)
+from alembic.op import create_check_constraint, create_primary_key, drop_constraint, rename_table
 
 from rucio.db.sqla.util import try_drop_constraint
 

rucio/db/sqla/migrate_repo/versions/5f139f77382a_added_child_rule_id_column.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -17,8 +16,7 @@
 
 import sqlalchemy as sa
 from alembic import context
-from alembic.op import (create_foreign_key, add_column, create_index,
-                        drop_constraint, drop_column, drop_index)
+from alembic.op import add_column, create_foreign_key, create_index, drop_column, drop_constraint, drop_index
 
 from rucio.db.sqla.types import GUID
 

rucio/db/sqla/migrate_repo/versions/688ef1840840_adding_did_meta_table.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -17,7 +16,7 @@
 
 import sqlalchemy as sa
 from alembic import context
-from alembic.op import create_primary_key, create_table, create_foreign_key, drop_table
+from alembic.op import create_foreign_key, create_primary_key, create_table, drop_table
 
 from rucio.db.sqla.types import JSON
 

rucio/db/sqla/migrate_repo/versions/6e572a9bfbf3_add_new_split_container_column_to_rules.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,11 +15,9 @@
 ''' add new split_container column to rules '''
 
 import sqlalchemy as sa
-
 from alembic import context
 from alembic.op import add_column, drop_column
 
-
 # Alembic revision identifiers
 revision = '6e572a9bfbf3'
 down_revision = '914b8f02df38'

rucio/db/sqla/migrate_repo/versions/70587619328_add_comment_column_for_subscriptions.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,11 +15,9 @@
 ''' add comment column for subscriptions '''
 
 import sqlalchemy as sa
-
 from alembic import context
 from alembic.op import add_column, drop_column
 
-
 # Alembic revision identifiers
 revision = '70587619328'
 down_revision = '4207be2fd914'

rucio/db/sqla/migrate_repo/versions/739064d31565_remove_history_table_pks.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,7 +15,7 @@
 ''' remove history table pks '''
 
 from alembic import context
-from alembic.op import drop_constraint, create_primary_key
+from alembic.op import create_primary_key, drop_constraint
 
 # Alembic revision identifiers
 revision = '739064d31565'

rucio/db/sqla/migrate_repo/versions/7541902bf173_add_didsfollowed_and_followevents_table.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -13,14 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-''' add DidsFollowed and FollowEvents table '''
+''' add DidFollowed and FollowEvent table '''
 
 import datetime
 
 import sqlalchemy as sa
 from alembic import context
-from alembic.op import (create_table, create_primary_key, create_check_constraint,
-                        create_foreign_key, create_index, drop_table)
+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
 

rucio/db/sqla/migrate_repo/versions/7ec22226cdbf_new_replica_state_for_temporary_.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

rucio/db/sqla/migrate_repo/versions/810a41685bc1_added_columns_rse_transfer_limits.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,10 +15,8 @@
 ''' added_columns_rse_transfer_limits '''
 
 import sqlalchemy as sa
-
 from alembic import context
-from alembic.op import (add_column, drop_column)
-
+from alembic.op import add_column, drop_column
 
 # Alembic revision identifiers
 revision = '810a41685bc1'

rucio/db/sqla/migrate_repo/versions/83f991c63a93_correct_rse_expression_length.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,7 +15,6 @@
 ''' correct rse_expression length '''
 
 import sqlalchemy as sa
-
 from alembic import context
 from alembic.op import alter_column
 

rucio/db/sqla/migrate_repo/versions/8523998e2e76_increase_size_of_extended_attributes_.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,11 +15,9 @@
 ''' increase size of extended_attributes column '''
 
 import sqlalchemy as sa
-
 from alembic import context
 from alembic.op import alter_column
 
-
 # Alembic revision identifiers
 revision = '8523998e2e76'
 down_revision = '7ec22226cdbf'

rucio/db/sqla/migrate_repo/versions/8ea9122275b1_adding_missing_function_based_indices.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,7 +15,7 @@
 ''' Adding missing function based indices '''
 
 from alembic import context
-from alembic.op import create_index, drop_index, drop_constraint, create_foreign_key
+from alembic.op import create_foreign_key, create_index, drop_constraint, drop_index
 
 # Alembic revision identifiers
 revision = '8ea9122275b1'

rucio/db/sqla/migrate_repo/versions/90f47792bb76_add_clob_payload_to_messages.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,11 +15,9 @@
 ''' add clob payload to messages '''
 
 import sqlalchemy as sa
-
 from alembic import context
 from alembic.op import add_column, drop_column
 
-
 # Alembic revision identifiers
 revision = '90f47792bb76'
 down_revision = 'bf3baa1c1474'

rucio/db/sqla/migrate_repo/versions/914b8f02df38_new_table_for_lifetime_model_exceptions.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -19,8 +18,7 @@ import datetime
 
 import sqlalchemy as sa
 from alembic import context
-from alembic.op import (create_table, create_primary_key,
-                        create_check_constraint, drop_table)
+from alembic.op import create_check_constraint, create_primary_key, create_table, drop_table
 
 from rucio.db.sqla.constants import DIDType, LifetimeExceptionsState
 from rucio.db.sqla.types import GUID

rucio/db/sqla/migrate_repo/versions/94a5961ddbf2_add_estimator_columns.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,11 +15,9 @@
 ''' add estimator columns to request table '''
 
 import sqlalchemy as sa
-
 from alembic import context
 from alembic.op import add_column, drop_column
 
-
 # Alembic revision identifiers
 revision = '94a5961ddbf2'
 down_revision = '1c45d9730ca6'

rucio/db/sqla/migrate_repo/versions/9a1b149a2044_add_saml_identity_type.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

rucio/db/sqla/migrate_repo/versions/9a45bc4ea66d_add_vp_table.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -17,7 +16,7 @@
 
 import sqlalchemy as sa
 from alembic import context
-from alembic.op import create_primary_key, create_table, create_foreign_key, drop_table
+from alembic.op import create_foreign_key, create_primary_key, create_table, drop_table
 
 from rucio.common.schema import get_schema_value
 from rucio.db.sqla.types import JSON

rucio/db/sqla/migrate_repo/versions/9eb936a81eb1_true_is_true.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -17,7 +16,6 @@
 
 from alembic import context, op
 
-
 # Alembic revision identifiers
 revision = '9eb936a81eb1'
 down_revision = 'b96a1c7e1cc4'

rucio/db/sqla/migrate_repo/versions/a08fa8de1545_transfer_stats_table.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.
+
+''' transfer_stats 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 = 'a08fa8de1545'
+down_revision = '4df2c5ddabc0'
+
+
+def upgrade():
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_table('transfer_stats',
+                     sa.Column('id', GUID()),
+                     sa.Column('resolution', sa.Integer),
+                     sa.Column('timestamp', sa.DateTime),
+                     sa.Column('dest_rse_id', GUID()),
+                     sa.Column('src_rse_id', GUID()),
+                     sa.Column('activity', sa.String(50)),
+                     sa.Column('files_done', sa.BigInteger),
+                     sa.Column('bytes_done', sa.BigInteger),
+                     sa.Column('files_failed', sa.BigInteger),
+                     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('TRANSFER_STATS_PK', 'transfer_stats', ['id'])
+        create_foreign_key('TRANSFER_STATS_DEST_RSE_FK', 'transfer_stats', 'rses', ['dest_rse_id'], ['id'])
+        create_foreign_key('TRANSFER_STATS_SRC_RSE_FK', 'transfer_stats', 'rses', ['src_rse_id'], ['id'])
+        create_index('TRANSFER_STATS_KEY_IDX', 'transfer_stats', ['resolution', 'timestamp', 'dest_rse_id', 'src_rse_id', 'activity'])
+        create_check_constraint('TRANSFER_STATS_CREATED_NN', 'transfer_stats', 'created_at is not null')
+        create_check_constraint('TRANSFER_STATS_UPDATED_NN', 'transfer_stats', 'updated_at is not null')
+
+
+def downgrade():
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_table('transfer_stats')

rucio/db/sqla/migrate_repo/versions/a118956323f8_added_vo_table_and_vo_col_to_rse.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -19,8 +18,7 @@ import datetime
 
 import sqlalchemy as sa
 from alembic import context
-from alembic.op import (add_column, create_primary_key, create_table, create_unique_constraint,
-                        drop_column, drop_constraint, drop_table, bulk_insert)
+from alembic.op import add_column, bulk_insert, create_primary_key, create_table, create_unique_constraint, drop_column, drop_constraint, drop_table
 from sqlalchemy import String
 
 # Alembic revision identifiers

rucio/db/sqla/migrate_repo/versions/a193a275255c_add_status_column_in_messages.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,7 +15,6 @@
 ''' Add status column in messages '''
 
 import sqlalchemy as sa
-
 from alembic import context
 from alembic.op import add_column, drop_column
 

rucio/db/sqla/migrate_repo/versions/a5f6f6e928a7_1_7_0.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,10 +15,8 @@
 ''' add columns for 1.7.0 release '''
 
 import sqlalchemy as sa
-
 from alembic import context
-from alembic.op import create_check_constraint, create_foreign_key, add_column, drop_column, drop_constraint
-
+from alembic.op import add_column, create_check_constraint, create_foreign_key, drop_column, drop_constraint
 
 # Alembic revision identifiers
 revision = 'a5f6f6e928a7'

rucio/db/sqla/migrate_repo/versions/a616581ee47_added_columns_to_table_requests.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

rucio/db/sqla/migrate_repo/versions/a6eb23955c28_state_idx_non_functional.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

rucio/db/sqla/migrate_repo/versions/a74275a1ad30_added_global_quota_table.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -19,8 +18,7 @@ import datetime
 
 import sqlalchemy as sa
 from alembic import context
-from alembic.op import (create_primary_key, create_check_constraint,
-                        create_table, drop_table, create_foreign_key)
+from alembic.op import create_check_constraint, create_foreign_key, create_primary_key, create_table, drop_table
 
 # Alembic revision identifiers
 revision = 'a74275a1ad30'

rucio/db/sqla/migrate_repo/versions/a93e4e47bda_heartbeats.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -19,9 +18,7 @@ import datetime
 
 import sqlalchemy as sa
 from alembic import context
-from alembic.op import (create_table, create_primary_key,
-                        create_check_constraint, create_index,
-                        drop_constraint, drop_table)
+from alembic.op import create_check_constraint, create_index, create_primary_key, create_table, drop_constraint, drop_table
 
 # Alembic revision identifiers
 revision = 'a93e4e47bda'

rucio/db/sqla/migrate_repo/versions/ae2a56fcc89_added_comment_column_to_rules.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

rucio/db/sqla/migrate_repo/versions/b0070f3695c8_add_deletedidmeta_table.py

@@ -0,0 +1,57 @@
+# 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 deleted_did_meta table '''
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import create_index, create_primary_key, create_table, drop_table
+
+from rucio.db.sqla.constants import DIDType
+from rucio.db.sqla.types import JSON
+
+# Alembic revision identifiers
+revision = 'b0070f3695c8'
+down_revision = 'b5493606bbf5'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_table('deleted_did_meta',
+                     sa.Column('scope', sa.String(25)),
+                     sa.Column('name', sa.String(255)),
+                     sa.Column('did_type', sa.Enum(DIDType,
+                                                   name='DEL_DID_META_DID_TYPE_CHK',
+                                                   create_constraint=True,
+                                                   values_callable=lambda obj: [e.value for e in obj])),
+                     sa.Column('meta', JSON()),
+                     sa.Column('created_at', sa.DateTime),
+                     sa.Column('updated_at', sa.DateTime),
+                     sa.Column('deleted_at', sa.DateTime))
+
+        create_primary_key('DEL_DID_META_PK', 'deleted_did_meta', ['scope', 'name'])
+        create_index('DEL_DID_META_DID_TYPE_IDX', 'deleted_did_meta', ['did_type'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_table('deleted_did_meta')

rucio/db/sqla/migrate_repo/versions/b4293a99f344_added_column_identity_to_table_tokens.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,11 +15,9 @@
 ''' added column identity to table tokens '''
 
 import sqlalchemy as sa
-
 from alembic import context
 from alembic.op import add_column, drop_column
 
-
 # Alembic revision identifiers
 revision = 'b4293a99f344'
 down_revision = '3ac1660a1a72'

rucio/db/sqla/migrate_repo/versions/b5493606bbf5_fix_primary_key_for_subscription_history.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.
+
+''' Fix primary key for subscription_history '''
+
+
+from alembic import context
+from alembic.op import create_primary_key, drop_constraint
+
+# Alembic revision identifiers
+revision = 'b5493606bbf5'
+down_revision = 'a08fa8de1545'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_constraint(constraint_name='SUBSCRIPTIONS_PK', table_name='subscriptions_history', type_='primary')
+        create_primary_key('SUBSCRIPTIONS_HISTORY_PK', 'subscriptions_history', ['id', 'updated_at'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_constraint(constraint_name='SUBSCRIPTIONS_HISTORY_PK', table_name='subscriptions_history', type_='primary')
+        create_primary_key('SUBSCRIPTIONS_PK', 'subscriptions_history', ['id', 'updated_at'])

rucio/db/sqla/migrate_repo/versions/b7d287de34fd_removal_of_replicastate_source.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,7 +15,7 @@
 ''' Removal of ReplicaState.SOURCE '''
 
 from alembic import context, op
-from alembic.op import (create_check_constraint)
+from alembic.op import create_check_constraint
 
 from rucio.db.sqla.util import try_drop_constraint
 

rucio/db/sqla/migrate_repo/versions/b818052fa670_add_index_to_quarantined_replicas.py

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright European Organization for Nuclear Research (CERN) since 2012
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,8 +15,7 @@
 ''' add index to quarantined replicas '''
 
 from alembic import context
-from alembic.op import (create_index, drop_index)
-
+from alembic.op import create_index, drop_index
 
 # revision identifiers, used by Alembic.
 revision = 'b818052fa670'