rucio 32.8.6__py3-none-any.whl

rucio/db/sqla/migrate_repo/versions/35ef10d1e11b_change_index_on_table_requests.py

@@ -0,0 +1,44 @@
+# -*- 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 index on table requests '''
+
+from alembic import context
+from alembic.op import create_index, drop_index
+
+
+# Alembic revision identifiers
+revision = '35ef10d1e11b'
+down_revision = '3152492b110b'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_index('REQUESTS_TYP_STA_UPD_IDX', 'requests', ["request_type", "state", "updated_at"])
+        drop_index('REQUESTS_TYP_STA_CRE_IDX', 'requests')
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_index('REQUESTS_TYP_STA_CRE_IDX', 'requests', ["request_type", "state", "created_at"])
+        drop_index('REQUESTS_TYP_STA_UPD_IDX', 'requests')

rucio/db/sqla/migrate_repo/versions/379a19b5332d_create_rse_limits_table.py

@@ -0,0 +1,67 @@
+# -*- 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 rse transfer limits 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, drop_table)
+
+from rucio.db.sqla.types import GUID
+
+# Alembic revision identifiers
+revision = '379a19b5332d'
+down_revision = '58bff7008037'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_table('rse_transfer_limits',
+                     sa.Column('rse_id', GUID()),
+                     sa.Column('activity', sa.String(50)),
+                     sa.Column('rse_expression', sa.String(3000)),
+                     sa.Column('max_transfers', sa.BigInteger),
+                     sa.Column('transfers', sa.BigInteger),
+                     sa.Column('waitings', 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('RSE_TRANSFER_LIMITS_PK', 'rse_transfer_limits', ['rse_id', 'activity'])
+        create_check_constraint('RSE_TRANSFER_LIMITS_CREATED_NN', 'rse_transfer_limits', 'created_at is not null')
+        create_check_constraint('RSE_TRANSFER_LIMITS_UPDATED_NN', 'rse_transfer_limits', 'updated_at is not null')
+        create_foreign_key('RSE_TRANSFER_LIMITS_RSE_ID_FK', 'rse_transfer_limits', 'rses', ['rse_id'], ['id'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_table('rse_transfer_limits')
+
+    elif context.get_context().dialect.name == 'postgresql':
+        # drop_constraint('RSE_TRANSFER_LIMITS_PK', 'rse_transfer_limits', type_='primary')
+        # drop_constraint('RSE_TRANSFER_LIMITS_CREATED_NN', 'rse_transfer_limits')
+        # drop_constraint('RSE_TRANSFER_LIMITS_UPDATED_NN', 'rse_transfer_limits')
+        # drop_constraint('RSE_TRANSFER_LIMITS_RSE_ID_FK', 'rse_transfer_limits')
+        pass

rucio/db/sqla/migrate_repo/versions/384b96aa0f60_created_rule_history_tables.py

@@ -0,0 +1,134 @@
+# -*- 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.
+
+''' created rule history tables '''
+
+import datetime
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import create_table, create_index, drop_table, drop_index, create_primary_key
+
+from rucio.db.sqla.constants import (DIDType, RuleGrouping, RuleState, RuleNotification)
+from rucio.db.sqla.types import GUID
+
+# Alembic revision identifiers
+revision = '384b96aa0f60'
+down_revision = '4cf0a2e127d4'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_table('rules_hist_recent',
+                     sa.Column('history_id', GUID()),
+                     sa.Column('id', GUID()),
+                     sa.Column('subscription_id', GUID()),
+                     sa.Column('account', sa.String(25)),
+                     sa.Column('scope', sa.String(25)),
+                     sa.Column('name', sa.String(255)),
+                     sa.Column('did_type', sa.Enum(DIDType,
+                                                   name='RULES_HIST_RECENT_DIDTYPE_CHK',
+                                                   create_constraint=True,
+                                                   values_callable=lambda obj: [e.value for e in obj])),
+                     sa.Column('state', sa.Enum(RuleState,
+                                                name='RULES_HIST_RECENT_STATE_CHK',
+                                                create_constraint=True,
+                                                values_callable=lambda obj: [e.value for e in obj])),
+                     sa.Column('error', sa.String(255)),
+                     sa.Column('rse_expression', sa.String(255)),
+                     sa.Column('copies', sa.SmallInteger),
+                     sa.Column('expires_at', sa.DateTime),
+                     sa.Column('weight', sa.String(255)),
+                     sa.Column('locked', sa.Boolean()),
+                     sa.Column('locks_ok_cnt', sa.BigInteger),
+                     sa.Column('locks_replicating_cnt', sa.BigInteger),
+                     sa.Column('locks_stuck_cnt', sa.BigInteger),
+                     sa.Column('source_replica_expression', sa.String(255)),
+                     sa.Column('activity', sa.String(50)),
+                     sa.Column('grouping', sa.Enum(RuleGrouping,
+                                                   name='RULES_HIST_RECENT_GROUPING_CHK',
+                                                   create_constraint=True,
+                                                   values_callable=lambda obj: [e.value for e in obj])),
+                     sa.Column('notification', sa.Enum(RuleNotification,
+                                                       name='RULES_HIST_RECENT_NOTIFY_CHK',
+                                                       create_constraint=True,
+                                                       values_callable=lambda obj: [e.value for e in obj])),
+                     sa.Column('stuck_at', sa.DateTime),
+                     sa.Column('purge_replicas', sa.Boolean()),
+                     sa.Column('ignore_availability', sa.Boolean()),
+                     sa.Column('ignore_account_limit', sa.Boolean()),
+                     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_table('rules_history',
+                     sa.Column('history_id', GUID()),
+                     sa.Column('id', GUID()),
+                     sa.Column('subscription_id', GUID()),
+                     sa.Column('account', sa.String(25)),
+                     sa.Column('scope', sa.String(25)),
+                     sa.Column('name', sa.String(255)),
+                     sa.Column('did_type', sa.Enum(DIDType,
+                                                   name='RULES_HISTORY_DIDTYPE_CHK',
+                                                   create_constraint=True,
+                                                   values_callable=lambda obj: [e.value for e in obj])),
+                     sa.Column('state', sa.Enum(RuleState,
+                                                name='RULES_HISTORY_STATE_CHK',
+                                                create_constraint=True,
+                                                values_callable=lambda obj: [e.value for e in obj])),
+                     sa.Column('error', sa.String(255)),
+                     sa.Column('rse_expression', sa.String(255)),
+                     sa.Column('copies', sa.SmallInteger),
+                     sa.Column('expires_at', sa.DateTime),
+                     sa.Column('weight', sa.String(255)),
+                     sa.Column('locked', sa.Boolean()),
+                     sa.Column('locks_ok_cnt', sa.BigInteger),
+                     sa.Column('locks_replicating_cnt', sa.BigInteger),
+                     sa.Column('locks_stuck_cnt', sa.BigInteger),
+                     sa.Column('source_replica_expression', sa.String(255)),
+                     sa.Column('activity', sa.String(50)),
+                     sa.Column('grouping', sa.Enum(RuleGrouping,
+                                                   name='RULES_HISTORY_GROUPING_CHK',
+                                                   create_constraint=True,
+                                                   values_callable=lambda obj: [e.value for e in obj])),
+                     sa.Column('notification', sa.Enum(RuleNotification,
+                                                       name='RULES_HISTORY_NOTIFY_CHK',
+                                                       create_constraint=True,
+                                                       values_callable=lambda obj: [e.value for e in obj])),
+                     sa.Column('stuck_at', sa.DateTime),
+                     sa.Column('purge_replicas', sa.Boolean()),
+                     sa.Column('ignore_availability', sa.Boolean()),
+                     sa.Column('ignore_account_limit', sa.Boolean()),
+                     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('RULES_HIST_RECENT_PK', 'rules_hist_recent', ['history_id'])
+        create_index('RULES_HIST_RECENT_ID_IDX', 'rules_hist_recent', ["id"])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql']:
+        drop_index('RULES_HIST_RECENT_ID_IDX', 'rules_hist_recent')
+        drop_table('rules_hist_recent')
+        drop_table('rules_history')
+
+    elif context.get_context().dialect.name == 'postgresql':
+        drop_table('rules_hist_recent')
+        drop_table('rules_history')

rucio/db/sqla/migrate_repo/versions/3ac1660a1a72_extend_distance_table.py

@@ -0,0 +1,58 @@
+# -*- 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.
+
+''' extend distance table '''
+
+import sqlalchemy as sa
+
+from alembic import context
+from alembic.op import add_column, drop_column
+
+
+# Alembic revision identifiers
+revision = '3ac1660a1a72'
+down_revision = '5673b4b6e843'
+
+
+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('distances', sa.Column('packet_loss', sa.Integer), schema=schema)
+        add_column('distances', sa.Column('latency', sa.Integer), schema=schema)
+        add_column('distances', sa.Column('mbps_file', sa.Integer), schema=schema)
+        add_column('distances', sa.Column('mbps_link', sa.Integer), schema=schema)
+        add_column('distances', sa.Column('queued_total', sa.Integer), schema=schema)
+        add_column('distances', sa.Column('done_1h', sa.Integer), schema=schema)
+        add_column('distances', sa.Column('done_6h', sa.Integer), 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('distances', 'packet_loss', schema=schema)
+        drop_column('distances', 'latency', schema=schema)
+        drop_column('distances', 'mbps_file', schema=schema)
+        drop_column('distances', 'mbps_link', schema=schema)
+        drop_column('distances', 'queued_total', schema=schema)
+        drop_column('distances', 'done_1h', schema=schema)
+        drop_column('distances', 'done_6h', schema=schema)

rucio/db/sqla/migrate_repo/versions/3ad36e2268b0_create_collection_replicas_updates_table.py

@@ -0,0 +1,79 @@
+# -*- 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 collection_replicas_updates table '''
+
+import datetime
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import (create_table, create_primary_key, add_column,
+                        create_check_constraint, create_index,
+                        drop_constraint, drop_column, drop_table, drop_index)
+
+from rucio.db.sqla.constants import DIDType
+from rucio.db.sqla.types import GUID
+
+# Alembic revision identifiers
+revision = '3ad36e2268b0'
+down_revision = '42db2617c364'
+
+
+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('collection_replicas', sa.Column('available_replicas_cnt', sa.BigInteger()), schema=schema)
+        add_column('collection_replicas', sa.Column('available_bytes', sa.BigInteger()), schema=schema)
+
+        create_table('updated_col_rep',
+                     sa.Column('id', GUID()),
+                     sa.Column('scope', sa.String(25)),
+                     sa.Column('name', sa.String(255)),
+                     sa.Column('did_type', sa.Enum(DIDType,
+                                                   name='UPDATED_COL_REP_TYPE_CHK',
+                                                   create_constraint=True,
+                                                   values_callable=lambda obj: [e.value for e in obj])),
+                     sa.Column('rse_id', GUID()),
+                     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('UPDATED_COL_REP_PK', 'updated_col_rep', ['id'])
+        create_check_constraint('UPDATED_COL_REP_SCOPE_NN', 'updated_col_rep', 'scope IS NOT NULL')
+        create_check_constraint('UPDATED_COL_REP_NAME_NN', 'updated_col_rep', 'name IS NOT NULL')
+        create_index('UPDATED_COL_REP_SNR_IDX', 'updated_col_rep', ['scope', 'name', 'rse_id'])
+
+
+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']:
+        drop_column('collection_replicas', 'available_replicas_cnt', schema=schema)
+        drop_column('collection_replicas', 'available_bytes', schema=schema)
+        drop_table('updated_col_rep')
+
+    elif context.get_context().dialect.name == 'mysql':
+        drop_column('collection_replicas', 'available_replicas_cnt', schema=schema)
+        drop_column('collection_replicas', 'available_bytes', schema=schema)
+        drop_constraint('UPDATED_COL_REP_PK', 'updated_col_rep', type_='primary')
+        drop_index('UPDATED_COL_REP_SNR_IDX', 'updated_col_rep')
+        drop_table('updated_col_rep')

rucio/db/sqla/migrate_repo/versions/3c9df354071b_extend_waiting_request_state.py

@@ -0,0 +1,61 @@
+# -*- 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.
+
+''' extend waiting request state '''
+
+from alembic import context, op
+from alembic.op import create_check_constraint
+
+from rucio.db.sqla.util import try_drop_constraint
+
+# Alembic revision identifiers
+revision = '3c9df354071b'
+down_revision = '2edee4a83846'
+
+
+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']:
+        try_drop_constraint('REQUESTS_STATE_CHK', 'requests')
+        create_check_constraint(constraint_name='REQUESTS_STATE_CHK', table_name='requests',
+                                condition="state in ('Q', 'G', 'S', 'D', 'F', 'L', 'N', 'O', 'A', 'U','W')")
+
+    elif context.get_context().dialect.name == 'mysql':
+        op.execute('ALTER TABLE ' + schema + 'requests DROP CHECK REQUESTS_STATE_CHK')  # pylint: disable=no-member
+        create_check_constraint(constraint_name='REQUESTS_STATE_CHK', table_name='requests',
+                                condition="state in ('Q', 'G', 'S', 'D', 'F', 'L', 'N', 'O', 'A', 'U','W')")
+
+
+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']:
+        try_drop_constraint('REQUESTS_STATE_CHK', 'requests')
+        create_check_constraint(constraint_name='REQUESTS_STATE_CHK', table_name='requests',
+                                condition="state in ('Q', 'G', 'S', 'D', 'F', 'L', 'N', 'O', 'A', 'U')")
+
+    elif context.get_context().dialect.name == 'mysql':
+        op.execute('ALTER TABLE ' + schema + 'requests DROP CHECK REQUESTS_STATE_CHK')  # pylint: disable=no-member
+        create_check_constraint(constraint_name='REQUESTS_STATE_CHK', table_name='requests',
+                                condition="state in ('Q', 'G', 'S', 'D', 'F', 'L', 'N', 'O', 'A', 'U')")

rucio/db/sqla/migrate_repo/versions/3d9813fab443_add_a_new_state_lost_in_badfilesstatus.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.
+
+''' add a new state LOST in BadFilesStatus '''
+
+from alembic import context
+from alembic.op import create_check_constraint
+
+from rucio.db.sqla.util import try_drop_constraint
+
+# Alembic revision identifiers
+revision = '3d9813fab443'
+down_revision = '1fc15ab60d43'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_check_constraint(constraint_name='BAD_REPLICAS_STATE_CHK', table_name='bad_replicas',
+                                condition="state in ('B', 'D', 'L', 'R', 'S')")
+
+
+def downgrade():
+
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'postgresql']:
+        try_drop_constraint('BAD_REPLICAS_STATE_CHK', 'bad_replicas')

rucio/db/sqla/migrate_repo/versions/40ad39ce3160_add_transferred_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_transferred_at_to_requests_table '''
+
+import sqlalchemy as sa
+
+from alembic import context
+from alembic.op import add_column, drop_column
+
+
+# Alembic revision identifiers
+revision = '40ad39ce3160'
+down_revision = '2ba5229cb54c'
+
+
+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('transferred_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', 'transferred_at', schema=schema)

rucio/db/sqla/migrate_repo/versions/4207be2fd914_add_notification_column_to_rules.py

@@ -0,0 +1,65 @@
+# -*- 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 notification column to rules '''
+
+import sqlalchemy as sa
+from alembic import context, op
+from alembic.op import add_column, drop_column
+
+from rucio.db.sqla.constants import RuleNotification
+from rucio.db.sqla.util import try_drop_constraint
+
+# Alembic revision identifiers
+revision = '4207be2fd914'
+down_revision = '14ec5aeb64cf'
+
+
+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', 'mysql']:
+        add_column('rules', sa.Column('notification', sa.Enum(RuleNotification,
+                                                              name='RULES_NOTIFICATION_CHK',
+                                                              create_constraint=True,
+                                                              values_callable=lambda obj: [e.value for e in obj]),
+                                      default=RuleNotification.NO), schema=schema[:-1])
+    elif context.get_context().dialect.name == 'postgresql':
+        op.execute("CREATE TYPE \"RULES_NOTIFICATION_CHK\" AS ENUM('Y', 'N', 'C', 'P')")
+        op.execute("ALTER TABLE %srules ADD COLUMN notification \"RULES_NOTIFICATION_CHK\"" % schema)
+
+
+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('RULES_NOTIFICATION_CHK', 'rules')
+        drop_column('rules', 'notification', schema=schema[:-1])
+
+    elif context.get_context().dialect.name == 'postgresql':
+        op.execute('ALTER TABLE %srules DROP CONSTRAINT IF EXISTS "RULES_NOTIFICATION_CHK", ALTER COLUMN notification TYPE CHAR' % schema)
+        op.execute('ALTER TABLE %srules DROP COLUMN notification' % schema)
+        op.execute('DROP TYPE \"RULES_NOTIFICATION_CHK\"')
+
+    elif context.get_context().dialect.name == 'mysql':
+        drop_column('rules', 'notification', schema=schema[:-1])

rucio/db/sqla/migrate_repo/versions/42db2617c364_create_index_on_requests_external_id.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.
+
+''' create index on requests.external_id '''
+
+from alembic import context
+from alembic.op import create_index, drop_index
+
+
+# Alembic revision identifiers
+revision = '42db2617c364'
+down_revision = '4bab9edd01fc'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_index('REQUESTS_EXTERNALID_UQ', 'requests', ['external_id'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_index('REQUESTS_EXTERNALID_UQ', 'requests')