207 lines
12 KiB
Python
207 lines
12 KiB
Python
|
"""Add new Appr-specific tables
|
||
|
|
|
||
|
|
Revision ID: 610320e9dacf
|
||
|
|
Revises: 5cbbfc95bac7
|
||
|
|
Create Date: 2018-05-24 16:46:13.514562
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
# revision identifiers, used by Alembic.
|
||
|
|
revision = '610320e9dacf'
|
||
|
|
down_revision = '5cbbfc95bac7'
|
||
|
|
|
||
|
|
from alembic import op as original_op
|
||
|
|
from data.migrations.progress import ProgressWrapper
|
||
|
|
import sqlalchemy as sa
|
||
|
|
|
||
|
|
from util.migrate.table_ops import copy_table_contents
|
||
|
|
|
||
|
|
def upgrade(tables, tester, progress_reporter):
|
||
|
|
op = ProgressWrapper(original_op, progress_reporter)
|
||
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
||
|
|
op.create_table('apprblobplacementlocation',
|
||
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('name', sa.String(length=255), nullable=False),
|
||
|
|
sa.PrimaryKeyConstraint('id', name=op.f('pk_apprblobplacementlocation'))
|
||
|
|
)
|
||
|
|
op.create_index('apprblobplacementlocation_name', 'apprblobplacementlocation', ['name'], unique=True)
|
||
|
|
op.create_table('apprtagkind',
|
||
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('name', sa.String(length=255), nullable=False),
|
||
|
|
sa.PrimaryKeyConstraint('id', name=op.f('pk_apprtagkind'))
|
||
|
|
)
|
||
|
|
op.create_index('apprtagkind_name', 'apprtagkind', ['name'], unique=True)
|
||
|
|
op.create_table('apprblob',
|
||
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('digest', sa.String(length=255), nullable=False),
|
||
|
|
sa.Column('media_type_id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('size', sa.BigInteger(), nullable=False),
|
||
|
|
sa.Column('uncompressed_size', sa.BigInteger(), nullable=True),
|
||
|
|
sa.ForeignKeyConstraint(['media_type_id'], ['mediatype.id'], name=op.f('fk_apprblob_media_type_id_mediatype')),
|
||
|
|
sa.PrimaryKeyConstraint('id', name=op.f('pk_apprblob'))
|
||
|
|
)
|
||
|
|
op.create_index('apprblob_digest', 'apprblob', ['digest'], unique=True)
|
||
|
|
op.create_index('apprblob_media_type_id', 'apprblob', ['media_type_id'], unique=False)
|
||
|
|
op.create_table('apprmanifest',
|
||
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('digest', sa.String(length=255), nullable=False),
|
||
|
|
sa.Column('media_type_id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('manifest_json', sa.Text(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(['media_type_id'], ['mediatype.id'], name=op.f('fk_apprmanifest_media_type_id_mediatype')),
|
||
|
|
sa.PrimaryKeyConstraint('id', name=op.f('pk_apprmanifest'))
|
||
|
|
)
|
||
|
|
op.create_index('apprmanifest_digest', 'apprmanifest', ['digest'], unique=True)
|
||
|
|
op.create_index('apprmanifest_media_type_id', 'apprmanifest', ['media_type_id'], unique=False)
|
||
|
|
op.create_table('apprmanifestlist',
|
||
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('digest', sa.String(length=255), nullable=False),
|
||
|
|
sa.Column('manifest_list_json', sa.Text(), nullable=False),
|
||
|
|
sa.Column('schema_version', sa.String(length=255), nullable=False),
|
||
|
|
sa.Column('media_type_id', sa.Integer(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(['media_type_id'], ['mediatype.id'], name=op.f('fk_apprmanifestlist_media_type_id_mediatype')),
|
||
|
|
sa.PrimaryKeyConstraint('id', name=op.f('pk_apprmanifestlist'))
|
||
|
|
)
|
||
|
|
op.create_index('apprmanifestlist_digest', 'apprmanifestlist', ['digest'], unique=True)
|
||
|
|
op.create_index('apprmanifestlist_media_type_id', 'apprmanifestlist', ['media_type_id'], unique=False)
|
||
|
|
op.create_table('apprblobplacement',
|
||
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('blob_id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('location_id', sa.Integer(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(['blob_id'], ['apprblob.id'], name=op.f('fk_apprblobplacement_blob_id_apprblob')),
|
||
|
|
sa.ForeignKeyConstraint(['location_id'], ['apprblobplacementlocation.id'], name=op.f('fk_apprblobplacement_location_id_apprblobplacementlocation')),
|
||
|
|
sa.PrimaryKeyConstraint('id', name=op.f('pk_apprblobplacement'))
|
||
|
|
)
|
||
|
|
op.create_index('apprblobplacement_blob_id', 'apprblobplacement', ['blob_id'], unique=False)
|
||
|
|
op.create_index('apprblobplacement_blob_id_location_id', 'apprblobplacement', ['blob_id', 'location_id'], unique=True)
|
||
|
|
op.create_index('apprblobplacement_location_id', 'apprblobplacement', ['location_id'], unique=False)
|
||
|
|
op.create_table('apprmanifestblob',
|
||
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('manifest_id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('blob_id', sa.Integer(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(['blob_id'], ['apprblob.id'], name=op.f('fk_apprmanifestblob_blob_id_apprblob')),
|
||
|
|
sa.ForeignKeyConstraint(['manifest_id'], ['apprmanifest.id'], name=op.f('fk_apprmanifestblob_manifest_id_apprmanifest')),
|
||
|
|
sa.PrimaryKeyConstraint('id', name=op.f('pk_apprmanifestblob'))
|
||
|
|
)
|
||
|
|
op.create_index('apprmanifestblob_blob_id', 'apprmanifestblob', ['blob_id'], unique=False)
|
||
|
|
op.create_index('apprmanifestblob_manifest_id', 'apprmanifestblob', ['manifest_id'], unique=False)
|
||
|
|
op.create_index('apprmanifestblob_manifest_id_blob_id', 'apprmanifestblob', ['manifest_id', 'blob_id'], unique=True)
|
||
|
|
op.create_table('apprmanifestlistmanifest',
|
||
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('manifest_list_id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('manifest_id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('operating_system', sa.String(length=255), nullable=True),
|
||
|
|
sa.Column('architecture', sa.String(length=255), nullable=True),
|
||
|
|
sa.Column('platform_json', sa.Text(), nullable=True),
|
||
|
|
sa.Column('media_type_id', sa.Integer(), nullable=False),
|
||
|
|
sa.ForeignKeyConstraint(['manifest_id'], ['apprmanifest.id'], name=op.f('fk_apprmanifestlistmanifest_manifest_id_apprmanifest')),
|
||
|
|
sa.ForeignKeyConstraint(['manifest_list_id'], ['apprmanifestlist.id'], name=op.f('fk_apprmanifestlistmanifest_manifest_list_id_apprmanifestlist')),
|
||
|
|
sa.ForeignKeyConstraint(['media_type_id'], ['mediatype.id'], name=op.f('fk_apprmanifestlistmanifest_media_type_id_mediatype')),
|
||
|
|
sa.PrimaryKeyConstraint('id', name=op.f('pk_apprmanifestlistmanifest'))
|
||
|
|
)
|
||
|
|
op.create_index('apprmanifestlistmanifest_manifest_id', 'apprmanifestlistmanifest', ['manifest_id'], unique=False)
|
||
|
|
op.create_index('apprmanifestlistmanifest_manifest_list_id', 'apprmanifestlistmanifest', ['manifest_list_id'], unique=False)
|
||
|
|
op.create_index('apprmanifestlistmanifest_manifest_list_id_media_type_id', 'apprmanifestlistmanifest', ['manifest_list_id', 'media_type_id'], unique=False)
|
||
|
|
op.create_index('apprmanifestlistmanifest_manifest_list_id_operating_system_arch', 'apprmanifestlistmanifest', ['manifest_list_id', 'operating_system', 'architecture', 'media_type_id'], unique=False)
|
||
|
|
op.create_index('apprmanifestlistmanifest_media_type_id', 'apprmanifestlistmanifest', ['media_type_id'], unique=False)
|
||
|
|
op.create_table('apprtag',
|
||
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('name', sa.String(length=255), nullable=False),
|
||
|
|
sa.Column('repository_id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('manifest_list_id', sa.Integer(), nullable=True),
|
||
|
|
sa.Column('lifetime_start', sa.BigInteger(), nullable=False),
|
||
|
|
sa.Column('lifetime_end', sa.BigInteger(), nullable=True),
|
||
|
|
sa.Column('hidden', sa.Boolean(), nullable=False),
|
||
|
|
sa.Column('reverted', sa.Boolean(), nullable=False),
|
||
|
|
sa.Column('protected', sa.Boolean(), nullable=False),
|
||
|
|
sa.Column('tag_kind_id', sa.Integer(), nullable=False),
|
||
|
|
sa.Column('linked_tag_id', sa.Integer(), nullable=True),
|
||
|
|
sa.ForeignKeyConstraint(['linked_tag_id'], ['apprtag.id'], name=op.f('fk_apprtag_linked_tag_id_apprtag')),
|
||
|
|
sa.ForeignKeyConstraint(['manifest_list_id'], ['apprmanifestlist.id'], name=op.f('fk_apprtag_manifest_list_id_apprmanifestlist')),
|
||
|
|
sa.ForeignKeyConstraint(['repository_id'], ['repository.id'], name=op.f('fk_apprtag_repository_id_repository')),
|
||
|
|
sa.ForeignKeyConstraint(['tag_kind_id'], ['apprtagkind.id'], name=op.f('fk_apprtag_tag_kind_id_apprtagkind')),
|
||
|
|
sa.PrimaryKeyConstraint('id', name=op.f('pk_apprtag'))
|
||
|
|
)
|
||
|
|
op.create_index('apprtag_lifetime_end', 'apprtag', ['lifetime_end'], unique=False)
|
||
|
|
op.create_index('apprtag_linked_tag_id', 'apprtag', ['linked_tag_id'], unique=False)
|
||
|
|
op.create_index('apprtag_manifest_list_id', 'apprtag', ['manifest_list_id'], unique=False)
|
||
|
|
op.create_index('apprtag_repository_id', 'apprtag', ['repository_id'], unique=False)
|
||
|
|
op.create_index('apprtag_repository_id_name', 'apprtag', ['repository_id', 'name'], unique=False)
|
||
|
|
op.create_index('apprtag_repository_id_name_hidden', 'apprtag', ['repository_id', 'name', 'hidden'], unique=False)
|
||
|
|
op.create_index('apprtag_repository_id_name_lifetime_end', 'apprtag', ['repository_id', 'name', 'lifetime_end'], unique=True)
|
||
|
|
op.create_index('apprtag_tag_kind_id', 'apprtag', ['tag_kind_id'], unique=False)
|
||
|
|
# ### end Alembic commands ###
|
||
|
|
|
||
|
|
conn = op.get_bind()
|
||
|
|
copy_table_contents('blobplacementlocation', 'apprblobplacementlocation', conn)
|
||
|
|
copy_table_contents('tagkind', 'apprtagkind', conn)
|
||
|
|
|
||
|
|
# ### population of test data ### #
|
||
|
|
|
||
|
|
tester.populate_table('apprmanifest', [
|
||
|
|
('digest', tester.TestDataType.String),
|
||
|
|
('media_type_id', tester.TestDataType.Foreign('mediatype')),
|
||
|
|
('manifest_json', tester.TestDataType.JSON),
|
||
|
|
])
|
||
|
|
|
||
|
|
tester.populate_table('apprmanifestlist', [
|
||
|
|
('digest', tester.TestDataType.String),
|
||
|
|
('manifest_list_json', tester.TestDataType.JSON),
|
||
|
|
('schema_version', tester.TestDataType.String),
|
||
|
|
('media_type_id', tester.TestDataType.Foreign('mediatype')),
|
||
|
|
])
|
||
|
|
|
||
|
|
tester.populate_table('apprmanifestlistmanifest', [
|
||
|
|
('manifest_list_id', tester.TestDataType.Foreign('apprmanifestlist')),
|
||
|
|
('manifest_id', tester.TestDataType.Foreign('apprmanifest')),
|
||
|
|
('operating_system', tester.TestDataType.String),
|
||
|
|
('architecture', tester.TestDataType.String),
|
||
|
|
('platform_json', tester.TestDataType.JSON),
|
||
|
|
('media_type_id', tester.TestDataType.Foreign('mediatype')),
|
||
|
|
])
|
||
|
|
|
||
|
|
tester.populate_table('apprblob', [
|
||
|
|
('digest', tester.TestDataType.String),
|
||
|
|
('media_type_id', tester.TestDataType.Foreign('mediatype')),
|
||
|
|
('size', tester.TestDataType.BigInteger),
|
||
|
|
('uncompressed_size', tester.TestDataType.BigInteger),
|
||
|
|
])
|
||
|
|
|
||
|
|
tester.populate_table('apprmanifestblob', [
|
||
|
|
('manifest_id', tester.TestDataType.Foreign('apprmanifest')),
|
||
|
|
('blob_id', tester.TestDataType.Foreign('apprblob')),
|
||
|
|
])
|
||
|
|
|
||
|
|
tester.populate_table('apprtag', [
|
||
|
|
('name', tester.TestDataType.String),
|
||
|
|
('repository_id', tester.TestDataType.Foreign('repository')),
|
||
|
|
('manifest_list_id', tester.TestDataType.Foreign('apprmanifestlist')),
|
||
|
|
('lifetime_start', tester.TestDataType.Integer),
|
||
|
|
('hidden', tester.TestDataType.Boolean),
|
||
|
|
('reverted', tester.TestDataType.Boolean),
|
||
|
|
('protected', tester.TestDataType.Boolean),
|
||
|
|
('tag_kind_id', tester.TestDataType.Foreign('apprtagkind')),
|
||
|
|
])
|
||
|
|
|
||
|
|
tester.populate_table('apprblobplacement', [
|
||
|
|
('blob_id', tester.TestDataType.Foreign('apprmanifestblob')),
|
||
|
|
('location_id', tester.TestDataType.Foreign('apprblobplacementlocation')),
|
||
|
|
])
|
||
|
|
|
||
|
|
# ### end population of test data ### #
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def downgrade(tables, tester, progress_reporter):
|
||
|
|
op = ProgressWrapper(original_op, progress_reporter)
|
||
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
||
|
|
op.drop_table('apprtag')
|
||
|
|
op.drop_table('apprmanifestlistmanifest')
|
||
|
|
op.drop_table('apprmanifestblob')
|
||
|
|
op.drop_table('apprblobplacement')
|
||
|
|
op.drop_table('apprmanifestlist')
|
||
|
|
op.drop_table('apprmanifest')
|
||
|
|
op.drop_table('apprblob')
|
||
|
|
op.drop_table('apprtagkind')
|
||
|
|
op.drop_table('apprblobplacementlocation')
|
||
|
|
# ### end Alembic commands ###
|