rucio 37.0.0rc1__py3-none-any.whl

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

@@ -0,0 +1,103 @@
+# 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 account_limits table '''
+
+import datetime
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import create_check_constraint, create_foreign_key, create_primary_key, create_table, drop_table
+
+from rucio.db.sqla.types import GUID
+
+# Alembic revision identifiers
+revision = 'd91002c5841'
+down_revision = '469d262be19'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'postgresql']:
+        drop_table('account_limits')
+
+        create_table('account_limits',
+                     sa.Column('account', sa.String(25)),
+                     sa.Column('rse_id', GUID()),
+                     sa.Column('bytes', 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('ACCOUNT_LIMITS_PK', 'account_limits', ['account', 'rse_id'])
+        create_check_constraint('ACCOUNT_LIMITS_CREATED_NN', 'account_limits', 'created_at is not null')
+        create_check_constraint('ACCOUNT_LIMITS_UPDATED_NN', 'account_limits', 'updated_at is not null')
+        create_foreign_key('ACCOUNT_LIMITS_ACCOUNT_FK', 'account_limits', 'accounts', ['account'], ['account'])
+        create_foreign_key('ACCOUNT_LIMITS_RSE_ID_FK', 'account_limits', 'rses', ['rse_id'], ['id'])
+
+    elif context.get_context().dialect.name == 'mysql':
+        drop_table('account_limits')
+
+        create_table('account_limits',
+                     sa.Column('account', sa.String(25)),
+                     sa.Column('rse_id', GUID()),
+                     sa.Column('bytes', 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('ACCOUNT_LIMITS_PK', 'account_limits', ['account', 'rse_id'])
+        create_check_constraint('ACCOUNT_LIMITS_CREATED_NN', 'account_limits', 'created_at is not null')
+        create_check_constraint('ACCOUNT_LIMITS_UPDATED_NN', 'account_limits', 'updated_at is not null')
+        create_foreign_key('ACCOUNT_LIMITS_ACCOUNT_FK', 'account_limits', 'accounts', ['account'], ['account'])
+        create_foreign_key('ACCOUNT_LIMITS_RSE_ID_FK', 'account_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('account_limits')
+
+        create_table('account_limits',
+                     sa.Column('account', sa.String(25)),
+                     sa.Column('rse_expression', sa.String(255)),
+                     sa.Column('name', sa.String(255)),
+                     sa.Column('value', 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('ACCOUNT_LIMITS_PK', 'account_limits', ['account', 'rse_expression', 'name'])
+        create_check_constraint('ACCOUNT_LIMITS_CREATED_NN', 'account_limits', 'created_at is not null')
+        create_check_constraint('ACCOUNT_LIMITS_UPDATED_NN', 'account_limits', 'updated_at is not null')
+        create_foreign_key('ACCOUNT_LIMITS_ACCOUNT_FK', 'account_limits', 'accounts', ['account'], ['account'])
+
+    elif context.get_context().dialect.name == 'mysql':
+        drop_table('account_limits')
+
+        create_table('account_limits',
+                     sa.Column('account', sa.String(25)),
+                     sa.Column('rse_expression', sa.String(255)),
+                     sa.Column('name', sa.String(255)),
+                     sa.Column('value', 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('ACCOUNT_LIMITS_PK', 'account_limits', ['account', 'rse_expression', 'name'])
+        create_check_constraint('ACCOUNT_LIMITS_CREATED_NN', 'account_limits', 'created_at is not null')
+        create_check_constraint('ACCOUNT_LIMITS_UPDATED_NN', 'account_limits', 'updated_at is not null')
+        create_foreign_key('ACCOUNT_LIMITS_ACCOUNT_FK', 'account_limits', 'accounts', ['account'], ['account'])

rucio/db/sqla/migrate_repo/versions/e138c364ebd0_extending_columns_for_filter_and_.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.
+
+''' Extending columns for filter and replication_rules in subscriptions '''
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import alter_column
+
+# Alembic revision identifiers
+revision = 'e138c364ebd0'
+down_revision = 'f85a2962b021'
+
+
+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', 'postgresql']:
+        alter_column('subscriptions', 'filter', existing_type=sa.String(2048), type_=sa.String(4000), schema=schema)
+        alter_column('subscriptions', 'replication_rules', existing_type=sa.String(1024), type_=sa.String(4000), schema=schema)
+        alter_column('subscriptions_history', 'filter', existing_type=sa.String(2048), type_=sa.String(4000), schema=schema)
+        alter_column('subscriptions_history', 'replication_rules', existing_type=sa.String(1024), type_=sa.String(4000), schema=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 in ['oracle', 'mysql', 'postgresql']:
+        alter_column('subscriptions', 'filter', existing_type=sa.String(4000), type_=sa.String(2048), schema=schema)
+        alter_column('subscriptions', 'replication_rules', existing_type=sa.String(4000), type_=sa.String(1024), schema=schema)
+        alter_column('subscriptions_history', 'filter', existing_type=sa.String(4000), type_=sa.String(2048), schema=schema)
+        alter_column('subscriptions_history', 'replication_rules', existing_type=sa.String(4000), type_=sa.String(1024), schema=schema)

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

@@ -0,0 +1,104 @@
+# 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.
+
+''' support for archive '''
+
+import datetime
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import add_column, create_foreign_key, create_index, create_primary_key, create_table, drop_column, drop_table
+
+from rucio.db.sqla.models import String
+from rucio.db.sqla.types import GUID
+
+# Alembic revision identifiers
+revision = 'e59300c8b179'
+down_revision = '6e572a9bfbf3'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        create_table('archive_contents',
+                     sa.Column('child_scope', String(25)),
+                     sa.Column('child_name', String(255)),
+                     sa.Column('scope', String(25)),
+                     sa.Column('name', String(255)),
+                     sa.Column('bytes', sa.BigInteger),
+                     sa.Column('adler32', String(8)),
+                     sa.Column('md5', String(32)),
+                     sa.Column('guid', GUID()),
+                     sa.Column('length', 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_table('archive_contents_history',
+                     sa.Column('child_scope', String(25)),
+                     sa.Column('child_name', String(255)),
+                     sa.Column('scope', String(25)),
+                     sa.Column('name', String(255)),
+                     sa.Column('bytes', sa.BigInteger),
+                     sa.Column('adler32', String(8)),
+                     sa.Column('md5', String(32)),
+                     sa.Column('guid', GUID()),
+                     sa.Column('length', 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('ARCH_CONTENTS_PK',
+                           'archive_contents',
+                           ['scope', 'name', 'child_scope', 'child_name'])
+        create_primary_key('ARCH_CONT_HIST_PK',
+                           'archive_contents_history',
+                           ['scope', 'name', 'child_scope', 'child_name'])
+        create_foreign_key('ARCH_CONTENTS_PARENT_FK', 'archive_contents', 'dids',
+                           ['scope', 'name'], ['scope', 'name'])
+        create_foreign_key('ARCH_CONTENTS_CHILD_FK', 'archive_contents', 'dids',
+                           ['child_scope', 'child_name'], ['scope', 'name'])
+
+        create_index('ARCH_CONTENTS_CHILD_IDX', 'archive_contents',
+                     ['child_scope', 'child_name', 'scope', 'name'])
+
+        create_index('ARCH_CONT_HIST_IDX', 'archive_contents_history',
+                     ['scope', 'name'])
+
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        add_column('dids', sa.Column('is_archive',
+                                     sa.Boolean(name='DIDS_ARCHIVE_CHK', create_constraint=True)),
+                   schema=schema)
+        add_column('dids', sa.Column('constituent',
+                                     sa.Boolean(name='DIDS_CONSTITUENT_CHK', create_constraint=True)),
+                   schema=schema)
+        add_column('deleted_dids', sa.Column('is_archive', sa.Boolean()), schema=schema)
+        add_column('deleted_dids', sa.Column('constituent', sa.Boolean()), schema=schema)
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'mysql', 'postgresql']:
+        drop_table('archive_contents')
+        drop_table('archive_contents_history')
+
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        drop_column('dids', 'is_archive', schema=schema)
+        drop_column('dids', 'constituent', schema=schema)
+        drop_column('deleted_dids', 'is_archive', schema=schema)
+        drop_column('deleted_dids', 'constituent', schema=schema)

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

@@ -0,0 +1,29 @@
+# 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.
+
+''' postgres_use_check_constraints '''
+
+# Alembic revision identifiers
+revision = 'f1b14a8c2ac1'
+down_revision = 'b8caac94d7f0'
+
+
+def upgrade():
+    # not needed anymore after SQLAlchemy 1.3.8
+    pass
+
+
+def downgrade():
+    # not needed anymore after SQLAlchemy 1.3.8
+    pass

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

@@ -0,0 +1,74 @@
+# 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.
+
+''' oracle_global_temporary_tables '''
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import create_table, drop_table
+
+from rucio.common.schema import get_schema_value
+from rucio.db.sqla.types import GUID, InternalScopeString, String
+
+# Alembic revision identifiers
+revision = 'f41ffe206f37'
+down_revision = 'd6e2c3b2cf26'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name == 'oracle':
+        additional_kwargs = {
+            'oracle_on_commit': 'DELETE ROWS',
+            'prefixes': ['GLOBAL TEMPORARY'],
+        }
+        for idx in range(5):
+            create_table(
+                f'TEMPORARY_SCOPE_NAME_{idx}',
+                sa.Column("scope", InternalScopeString(get_schema_value('SCOPE_LENGTH'))),
+                sa.Column("name", String(get_schema_value('NAME_LENGTH'))),
+                sa.PrimaryKeyConstraint('scope', 'name', name=f'TEMPORARY_SCOPE_NAME_{idx}_PK'),
+                **additional_kwargs,
+            )
+            create_table(
+                f'TEMPORARY_ASSOCIATION_{idx}',
+                sa.Column("scope", InternalScopeString(get_schema_value('SCOPE_LENGTH'))),
+                sa.Column("name", String(get_schema_value('NAME_LENGTH'))),
+                sa.Column("child_scope", InternalScopeString(get_schema_value('SCOPE_LENGTH'))),
+                sa.Column("child_name", String(get_schema_value('NAME_LENGTH'))),
+                sa.PrimaryKeyConstraint('scope', 'name', 'child_scope', 'child_name', name=f'TEMPORARY_ASSOCIATION_{idx}_PK'),
+                **additional_kwargs,
+            )
+            create_table(
+                f'TEMPORARY_ID_{idx}',
+                sa.Column("id", GUID()),
+                sa.PrimaryKeyConstraint('id', name=f'TEMPORARY_ID_{idx}_PK'),
+                **additional_kwargs,
+            )
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name == 'oracle':
+        global_temp_tables = sa.inspect(context.get_bind()).get_temp_table_names()
+        for idx in range(5):
+            for table_name in [f'TEMPORARY_ID_{idx}', f'TEMPORARY_ASSOCIATION_{idx}', f'TEMPORARY_SCOPE_NAME_{idx}']:
+                if table_name in global_temp_tables:
+                    drop_table(table_name)

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

@@ -0,0 +1,47 @@
+# 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 transfertool column and index to requests table'''
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import add_column, create_index, drop_column, drop_index
+
+# Alembic revision identifiers
+revision = 'f85a2962b021'
+down_revision = 'd23453595260'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'postgresql', 'mysql']:
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        add_column('requests', sa.Column('transfertool', sa.String(64)), schema=schema)
+        add_column('requests_history', sa.Column('transfertool', sa.String(64)), schema=schema)
+        create_index('REQUESTS_TYP_STA_TRA_ACT_IDX', 'requests', ['request_type', 'state', 'transfertool', 'activity'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    if context.get_context().dialect.name in ['oracle', 'postgresql', 'mysql']:
+        schema = context.get_context().version_table_schema if context.get_context().version_table_schema else ''
+        drop_index('REQUESTS_TYP_STA_TRA_ACT_IDX', 'requests')
+        drop_column('requests', 'transfertool', schema=schema)
+        drop_column('requests_history', 'transfertool', schema=schema)

rucio/db/sqla/migrate_repo/versions/fa7a7d78b602_increase_refresh_token_size.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.
+
+''' increase refresh token size '''
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import alter_column
+
+# Alembic revision identifiers
+revision = 'fa7a7d78b602'
+down_revision = '1677d4d803c8'
+
+
+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('tokens', 'refresh_token', existing_type=sa.String(315), type_=sa.String(3072), 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 ''
+        alter_column('tokens', 'refresh_token', existing_type=sa.String(3072), type_=sa.String(315), schema=schema)

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

@@ -0,0 +1,37 @@
+# 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_replicas_rse_id_tombstone_idx '''
+
+from alembic.op import create_index, drop_index
+
+# Alembic revision identifiers
+revision = 'fb28a95fe288'
+down_revision = '140fef722e91'
+
+
+def upgrade():
+    '''
+    Upgrade the database to this revision
+    '''
+
+    create_index('REPLICAS_RSE_ID_TOMBSTONE_IDX', 'replicas', ['rse_id', 'tombstone'])
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+
+    drop_index('REPLICAS_RSE_ID_TOMBSTONE_IDX', 'replicas')

rucio/db/sqla/migrate_repo/versions/fe1a65b176c9_set_third_party_copy_read_and_write_.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.
+
+''' set third_party_copy_read and write fields '''
+
+from alembic import context
+from alembic.op import alter_column, execute  # pylint: disable=no-member
+
+# Alembic revision identifiers
+revision = 'fe1a65b176c9'
+down_revision = '0f1adb7a599a'
+
+
+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 ''
+        schema_prefix = schema + '.' if schema else ''
+        execute('UPDATE ' + schema_prefix + 'rse_protocols SET third_party_copy_read=third_party_copy WHERE third_party_copy_read is NULL')
+        execute('UPDATE ' + schema_prefix + 'rse_protocols SET third_party_copy_write=third_party_copy WHERE third_party_copy_write is NULL')
+        # Add server default to 0. The initial alembic migration creates the column without the default, even if it is set in 'models'
+        alter_column('rse_protocols', 'third_party_copy_read', server_default='0', schema=schema)
+        alter_column('rse_protocols', 'third_party_copy_write', server_default='0', schema=schema)
+
+
+def downgrade():
+    '''
+    Downgrade the database to the previous revision
+    '''
+    pass

rucio/db/sqla/migrate_repo/versions/fe8ea2fa9788_added_third_party_copy_column_to_rse_.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 third_party_copy column to rse_protocols '''
+
+import sqlalchemy as sa
+from alembic import context
+from alembic.op import add_column, drop_column
+
+# Alembic revision identifiers
+revision = 'fe8ea2fa9788'
+down_revision = '0437a40dbfd1'
+
+
+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_protocols', sa.Column('third_party_copy', sa.Integer, 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('rse_protocols', 'third_party_copy', schema=schema)