def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.add_column(
"repositorybuildtrigger",
sa.Column("successive_failure_count", sa.Integer(), server_default="0", nullable=False),
)
op.add_column(
"repositorybuildtrigger",
sa.Column(
"successive_internal_error_count", sa.Integer(), server_default="0", nullable=False
),
)
# ### end Alembic commands ###
op.bulk_insert(
tables.disablereason,
[
{"id": 2, "name": "successive_build_failures"},
{"id": 3, "name": "successive_build_internal_errors"},
],
)
# ### population of test data ### #
tester.populate_column(
"repositorybuildtrigger", "successive_failure_count", tester.TestDataType.Integer
)
tester.populate_column(
"repositorybuildtrigger", "successive_internal_error_count", tester.TestDataType.Integer
)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.add_column('user', sa.Column('creation_date', sa.DateTime(), nullable=True))
# ### end Alembic commands ###
# ### population of test data ### #
tester.populate_column('user', 'creation_date', tester.TestDataType.DateTime)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.add_column("user", sa.Column("maximum_queued_builds_count", sa.Integer(), nullable=True))
# ### end Alembic commands ###
# ### population of test data ### #
tester.populate_column("user", "maximum_queued_builds_count", tester.TestDataType.Integer)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
op.add_column(
"user", sa.Column("location", UTF8CharField(length=255),
nullable=True))
# ### population of test data ### #
tester.populate_column("user", "location", tester.TestDataType.UTF8Char)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.create_table(
"disablereason",
sa.Column("id", sa.Integer(), nullable=False),
sa.Column("name", sa.String(length=255), nullable=False),
sa.PrimaryKeyConstraint("id", name=op.f("pk_disablereason")),
)
op.create_index("disablereason_name",
"disablereason", ["name"],
unique=True)
op.bulk_insert(
tables.disablereason,
[
{
"id": 1,
"name": "user_toggled"
},
],
)
op.bulk_insert(tables.logentrykind, [
{
"name": "toggle_repo_trigger"
},
])
op.add_column(u"repositorybuildtrigger",
sa.Column("disabled_reason_id", sa.Integer(), nullable=True))
op.add_column(
u"repositorybuildtrigger",
sa.Column("enabled",
sa.Boolean(),
nullable=False,
server_default=sa.sql.expression.true()),
)
op.create_index(
"repositorybuildtrigger_disabled_reason_id",
"repositorybuildtrigger",
["disabled_reason_id"],
unique=False,
)
op.create_foreign_key(
op.f("fk_repositorybuildtrigger_disabled_reason_id_disablereason"),
"repositorybuildtrigger",
"disablereason",
["disabled_reason_id"],
["id"],
)
# ### end Alembic commands ###
# ### population of test data ### #
tester.populate_column("repositorybuildtrigger", "disabled_reason_id",
tester.TestDataType.Foreign("disablereason"))
tester.populate_column("repositorybuildtrigger", "enabled",
tester.TestDataType.Boolean)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.add_column('queueitem', sa.Column('state_id', sa.String(length=255), nullable=False, server_default=''))
op.create_index('queueitem_state_id', 'queueitem', ['state_id'], unique=False)
# ### end Alembic commands ###
# ### population of test data ### #
tester.populate_column('queueitem', 'state_id', tester.TestDataType.String)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.add_column("user", sa.Column("last_accessed", sa.DateTime(), nullable=True))
op.create_index("user_last_accessed", "user", ["last_accessed"], unique=False)
# ### end Alembic commands ###
# ### population of test data ### #
tester.populate_column("user", "last_accessed", tester.TestDataType.DateTime)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
### commands auto generated by Alembic - please adjust! ###
op.add_column('repository', sa.Column('trust_enabled', sa.Boolean(), nullable=False, server_default=sa.sql.expression.false()))
### end Alembic commands ###
op.bulk_insert(tables.logentrykind, [
{'name': 'change_repo_trust'},
])
# ### population of test data ### #
tester.populate_column('repository', 'trust_enabled', tester.TestDataType.Boolean)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
op.add_column('repositorynotification', sa.Column('number_of_failures',
sa.Integer(),
nullable=False,
server_default='0'))
op.bulk_insert(tables.logentrykind, [
{'name': 'reset_repo_notification'},
])
# ### population of test data ### #
tester.populate_column('repositorynotification', 'number_of_failures', tester.TestDataType.Integer)
def downgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.add_column(
'manifestblob',
sa.Column('blob_index',
mysql.INTEGER(display_width=11),
autoincrement=False,
nullable=True))
op.create_index('manifestblob_manifest_id_blob_index',
'manifestblob', ['manifest_id', 'blob_index'],
unique=True)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.add_column('repositorybuildtrigger',
sa.Column('disabled_datetime', sa.DateTime(), nullable=True))
op.create_index('repositorybuildtrigger_disabled_datetime',
'repositorybuildtrigger', ['disabled_datetime'],
unique=False)
# ### end Alembic commands ###
# ### population of test data ### #
tester.populate_column('repositorybuildtrigger', 'disabled_datetime',
tester.TestDataType.DateTime)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.add_column(
"messages",
sa.Column("media_type_id",
sa.Integer(),
nullable=False,
server_default="1"))
op.add_column(
"messages",
sa.Column("severity",
sa.String(length=255),
nullable=False,
server_default="info"),
)
op.alter_column(
"messages",
"uuid",
existing_type=mysql.VARCHAR(length=36),
server_default="",
nullable=False,
)
op.create_index("messages_media_type_id",
"messages", ["media_type_id"],
unique=False)
op.create_index("messages_severity",
"messages", ["severity"],
unique=False)
op.create_index("messages_uuid", "messages", ["uuid"], unique=False)
op.create_foreign_key(
op.f("fk_messages_media_type_id_mediatype"),
"messages",
"mediatype",
["media_type_id"],
["id"],
)
# ### end Alembic commands ###
op.bulk_insert(tables.mediatype, [
{
"name": "text/markdown"
},
])
# ### population of test data ### #
tester.populate_column("messages", "media_type_id",
tester.TestDataType.Foreign("mediatype"))
tester.populate_column("messages", "severity", lambda: "info")
tester.populate_column("messages", "uuid", tester.TestDataType.UUID)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.create_table(
'disablereason', sa.Column('id', sa.Integer(), nullable=False),
sa.Column('name', sa.String(length=255), nullable=False),
sa.PrimaryKeyConstraint('id', name=op.f('pk_disablereason')))
op.create_index('disablereason_name',
'disablereason', ['name'],
unique=True)
op.bulk_insert(
tables.disablereason,
[
{
'id': 1,
'name': 'user_toggled'
},
],
)
op.bulk_insert(tables.logentrykind, [
{
'name': 'toggle_repo_trigger'
},
])
op.add_column(u'repositorybuildtrigger',
sa.Column('disabled_reason_id', sa.Integer(), nullable=True))
op.add_column(
u'repositorybuildtrigger',
sa.Column('enabled',
sa.Boolean(),
nullable=False,
server_default=sa.sql.expression.true()))
op.create_index('repositorybuildtrigger_disabled_reason_id',
'repositorybuildtrigger', ['disabled_reason_id'],
unique=False)
op.create_foreign_key(
op.f('fk_repositorybuildtrigger_disabled_reason_id_disablereason'),
'repositorybuildtrigger', 'disablereason', ['disabled_reason_id'],
['id'])
# ### end Alembic commands ###
# ### population of test data ### #
tester.populate_column('repositorybuildtrigger', 'disabled_reason_id',
tester.TestDataType.Foreign('disablereason'))
tester.populate_column('repositorybuildtrigger', 'enabled',
tester.TestDataType.Boolean)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
op.create_table(
"repositorykind",
sa.Column("id", sa.Integer(), nullable=False),
sa.Column("name", sa.String(length=255), nullable=False),
sa.PrimaryKeyConstraint("id", name=op.f("pk_repositorykind")),
)
op.create_index("repositorykind_name",
"repositorykind", ["name"],
unique=True)
op.bulk_insert(
tables.repositorykind,
[
{
"id": 1,
"name": "image"
},
{
"id": 2,
"name": "application"
},
],
)
op.add_column(
u"repository",
sa.Column("kind_id", sa.Integer(), nullable=False, server_default="1"))
op.create_index("repository_kind_id",
"repository", ["kind_id"],
unique=False)
op.create_foreign_key(
op.f("fk_repository_kind_id_repositorykind"),
"repository",
"repositorykind",
["kind_id"],
["id"],
)
# ### population of test data ### #
tester.populate_column("repository", "kind_id",
tester.TestDataType.Foreign("repositorykind"))
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.add_column(
'messages',
sa.Column('media_type_id',
sa.Integer(),
nullable=False,
server_default='1'))
op.add_column(
'messages',
sa.Column('severity',
sa.String(length=255),
nullable=False,
server_default='info'))
op.alter_column('messages',
'uuid',
existing_type=mysql.VARCHAR(length=36),
server_default='',
nullable=False)
op.create_index('messages_media_type_id',
'messages', ['media_type_id'],
unique=False)
op.create_index('messages_severity',
'messages', ['severity'],
unique=False)
op.create_index('messages_uuid', 'messages', ['uuid'], unique=False)
op.create_foreign_key(op.f('fk_messages_media_type_id_mediatype'),
'messages', 'mediatype', ['media_type_id'], ['id'])
# ### end Alembic commands ###
op.bulk_insert(tables.mediatype, [
{
'name': 'text/markdown'
},
])
# ### population of test data ### #
tester.populate_column('messages', 'media_type_id',
tester.TestDataType.Foreign('mediatype'))
tester.populate_column('messages', 'severity', lambda: 'info')
tester.populate_column('messages', 'uuid', tester.TestDataType.UUID)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
### commands auto generated by Alembic - please adjust! ###
op.add_column(
"user", sa.Column("company", UTF8CharField(length=255), nullable=True))
op.add_column(
"user",
sa.Column("family_name", UTF8CharField(length=255), nullable=True))
op.add_column(
"user",
sa.Column("given_name", UTF8CharField(length=255), nullable=True))
### end Alembic commands ###
op.bulk_insert(tables.userpromptkind, [
{
"name": "enter_name"
},
{
"name": "enter_company"
},
])
# ### population of test data ### #
tester.populate_column("user", "company", tester.TestDataType.UTF8Char)
tester.populate_column("user", "family_name", tester.TestDataType.UTF8Char)
tester.populate_column("user", "given_name", tester.TestDataType.UTF8Char)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.add_column(
'repositorybuildtrigger',
sa.Column('successive_failure_count',
sa.Integer(),
server_default='0',
nullable=False))
op.add_column(
'repositorybuildtrigger',
sa.Column('successive_internal_error_count',
sa.Integer(),
server_default='0',
nullable=False))
# ### end Alembic commands ###
op.bulk_insert(
tables.disablereason,
[
{
'id': 2,
'name': 'successive_build_failures'
},
{
'id': 3,
'name': 'successive_build_internal_errors'
},
],
)
# ### population of test data ### #
tester.populate_column('repositorybuildtrigger',
'successive_failure_count',
tester.TestDataType.Integer)
tester.populate_column('repositorybuildtrigger',
'successive_internal_error_count',
tester.TestDataType.Integer)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
op.create_table(
'repositorykind', sa.Column('id', sa.Integer(), nullable=False),
sa.Column('name', sa.String(length=255), nullable=False),
sa.PrimaryKeyConstraint('id', name=op.f('pk_repositorykind')))
op.create_index('repositorykind_name',
'repositorykind', ['name'],
unique=True)
op.bulk_insert(
tables.repositorykind,
[
{
'id': 1,
'name': 'image'
},
{
'id': 2,
'name': 'application'
},
],
)
op.add_column(
u'repository',
sa.Column('kind_id', sa.Integer(), nullable=False, server_default='1'))
op.create_index('repository_kind_id',
'repository', ['kind_id'],
unique=False)
op.create_foreign_key(op.f('fk_repository_kind_id_repositorykind'),
'repository', 'repositorykind', ['kind_id'], ['id'])
# ### population of test data ### #
tester.populate_column('repository', 'kind_id',
tester.TestDataType.Foreign('repositorykind'))
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.create_table(
'robotaccounttoken', sa.Column('id', sa.Integer(), nullable=False),
sa.Column('robot_account_id', sa.Integer(), nullable=False),
sa.Column('token', sa.String(length=255), nullable=False),
sa.Column('fully_migrated',
sa.Boolean(),
nullable=False,
server_default='0'),
sa.ForeignKeyConstraint(
['robot_account_id'], ['user.id'],
name=op.f('fk_robotaccounttoken_robot_account_id_user')),
sa.PrimaryKeyConstraint('id', name=op.f('pk_robotaccounttoken')))
op.create_index('robotaccounttoken_robot_account_id',
'robotaccounttoken', ['robot_account_id'],
unique=True)
op.add_column(
u'accesstoken',
sa.Column('token_code', sa.String(length=255), nullable=True))
op.add_column(
u'accesstoken',
sa.Column('token_name', sa.String(length=255), nullable=True))
op.create_index('accesstoken_token_name',
'accesstoken', ['token_name'],
unique=True)
op.add_column(
u'appspecificauthtoken',
sa.Column('token_name', sa.String(length=255), nullable=True))
op.add_column(
u'appspecificauthtoken',
sa.Column('token_secret', sa.String(length=255), nullable=True))
op.create_index('appspecificauthtoken_token_name',
'appspecificauthtoken', ['token_name'],
unique=True)
op.add_column(
u'emailconfirmation',
sa.Column('verification_code', sa.String(length=255), nullable=True))
op.add_column(
u'oauthaccesstoken',
sa.Column('token_code', sa.String(length=255), nullable=True))
op.add_column(
u'oauthaccesstoken',
sa.Column('token_name', sa.String(length=255), nullable=True))
op.create_index('oauthaccesstoken_token_name',
'oauthaccesstoken', ['token_name'],
unique=True)
op.add_column(
u'oauthapplication',
sa.Column('secure_client_secret', sa.String(length=255),
nullable=True))
op.add_column(
u'oauthapplication',
sa.Column('fully_migrated',
sa.Boolean(),
server_default='0',
nullable=False))
op.add_column(
u'oauthauthorizationcode',
sa.Column('code_credential', sa.String(length=255), nullable=True))
op.add_column(u'oauthauthorizationcode',
sa.Column('code_name', sa.String(length=255), nullable=True))
op.create_index('oauthauthorizationcode_code_name',
'oauthauthorizationcode', ['code_name'],
unique=True)
op.drop_index('oauthauthorizationcode_code',
table_name='oauthauthorizationcode')
op.create_index('oauthauthorizationcode_code',
'oauthauthorizationcode', ['code'],
unique=True)
op.add_column(
u'repositorybuildtrigger',
sa.Column('secure_auth_token', sa.String(length=255), nullable=True))
op.add_column(u'repositorybuildtrigger',
sa.Column('secure_private_key', sa.Text(), nullable=True))
op.add_column(
u'repositorybuildtrigger',
sa.Column('fully_migrated',
sa.Boolean(),
server_default='0',
nullable=False))
# ### end Alembic commands ###
# ### population of test data ### #
tester.populate_table('robotaccounttoken', [
('robot_account_id', tester.TestDataType.Foreign('user')),
('token', tester.TestDataType.Token),
('fully_migrated', tester.TestDataType.Boolean),
])
tester.populate_column('accesstoken', 'code', tester.TestDataType.Token)
tester.populate_column('appspecificauthtoken', 'token_code',
tester.TestDataType.Token)
tester.populate_column('emailconfirmation', 'verification_code',
tester.TestDataType.Token)
tester.populate_column('oauthaccesstoken', 'token_code',
tester.TestDataType.Token)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.create_table(
"robotaccounttoken",
sa.Column("id", sa.Integer(), nullable=False),
sa.Column("robot_account_id", sa.Integer(), nullable=False),
sa.Column("token", sa.String(length=255), nullable=False),
sa.Column("fully_migrated",
sa.Boolean(),
nullable=False,
server_default="0"),
sa.ForeignKeyConstraint(
["robot_account_id"],
["user.id"],
name=op.f("fk_robotaccounttoken_robot_account_id_user"),
),
sa.PrimaryKeyConstraint("id", name=op.f("pk_robotaccounttoken")),
)
op.create_index("robotaccounttoken_robot_account_id",
"robotaccounttoken", ["robot_account_id"],
unique=True)
op.add_column(
u"accesstoken",
sa.Column("token_code", sa.String(length=255), nullable=True))
op.add_column(
u"accesstoken",
sa.Column("token_name", sa.String(length=255), nullable=True))
op.create_index("accesstoken_token_name",
"accesstoken", ["token_name"],
unique=True)
op.add_column(
u"appspecificauthtoken",
sa.Column("token_name", sa.String(length=255), nullable=True))
op.add_column(
u"appspecificauthtoken",
sa.Column("token_secret", sa.String(length=255), nullable=True))
op.create_index("appspecificauthtoken_token_name",
"appspecificauthtoken", ["token_name"],
unique=True)
op.add_column(
u"emailconfirmation",
sa.Column("verification_code", sa.String(length=255), nullable=True))
op.add_column(
u"oauthaccesstoken",
sa.Column("token_code", sa.String(length=255), nullable=True))
op.add_column(
u"oauthaccesstoken",
sa.Column("token_name", sa.String(length=255), nullable=True))
op.create_index("oauthaccesstoken_token_name",
"oauthaccesstoken", ["token_name"],
unique=True)
op.add_column(
u"oauthapplication",
sa.Column("secure_client_secret", sa.String(length=255),
nullable=True))
op.add_column(
u"oauthapplication",
sa.Column("fully_migrated",
sa.Boolean(),
server_default="0",
nullable=False),
)
op.add_column(
u"oauthauthorizationcode",
sa.Column("code_credential", sa.String(length=255), nullable=True),
)
op.add_column(u"oauthauthorizationcode",
sa.Column("code_name", sa.String(length=255), nullable=True))
op.create_index("oauthauthorizationcode_code_name",
"oauthauthorizationcode", ["code_name"],
unique=True)
op.drop_index("oauthauthorizationcode_code",
table_name="oauthauthorizationcode")
op.create_index("oauthauthorizationcode_code",
"oauthauthorizationcode", ["code"],
unique=True)
op.add_column(
u"repositorybuildtrigger",
sa.Column("secure_auth_token", sa.String(length=255), nullable=True),
)
op.add_column(u"repositorybuildtrigger",
sa.Column("secure_private_key", sa.Text(), nullable=True))
op.add_column(
u"repositorybuildtrigger",
sa.Column("fully_migrated",
sa.Boolean(),
server_default="0",
nullable=False),
)
# ### end Alembic commands ###
# ### population of test data ### #
tester.populate_table(
"robotaccounttoken",
[
("robot_account_id", tester.TestDataType.Foreign("user")),
("token", tester.TestDataType.Token),
("fully_migrated", tester.TestDataType.Boolean),
],
)
tester.populate_column("accesstoken", "code", tester.TestDataType.Token)
tester.populate_column("appspecificauthtoken", "token_code",
tester.TestDataType.Token)
tester.populate_column("emailconfirmation", "verification_code",
tester.TestDataType.Token)
tester.populate_column("oauthaccesstoken", "token_code",
tester.TestDataType.Token)
def downgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
op.add_column("imagestorage",
sa.Column("checksum", sa.String(length=255), nullable=True))
def downgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.add_column(
u"oauthaccesstoken", sa.Column("refresh_token", sa.String(length=255), nullable=True)
)
op.create_index(
"oauthaccesstoken_refresh_token", "oauthaccesstoken", ["refresh_token"], unique=False
)
op.add_column(
"repositorybuildtrigger", sa.Column("auth_token", sa.String(length=255), nullable=True)
)
op.add_column("repositorybuildtrigger", sa.Column("private_key", sa.Text(), nullable=True))
op.add_column("oauthauthorizationcode", sa.Column("code", sa.String(length=255), nullable=True))
op.create_index("oauthauthorizationcode_code", "oauthauthorizationcode", ["code"], unique=True)
op.add_column(
"oauthapplication", sa.Column("client_secret", sa.String(length=255), nullable=True)
)
op.add_column(
"oauthaccesstoken", sa.Column("access_token", sa.String(length=255), nullable=True)
)
op.create_index(
"oauthaccesstoken_access_token", "oauthaccesstoken", ["access_token"], unique=False
)
op.add_column(
"appspecificauthtoken", sa.Column("token_code", sa.String(length=255), nullable=True)
)
op.create_index(
"appspecificauthtoken_token_code", "appspecificauthtoken", ["token_code"], unique=True
)
op.add_column("accesstoken", sa.Column("code", sa.String(length=255), nullable=True))
op.create_index("accesstoken_code", "accesstoken", ["code"], unique=True)
def downgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
# ### commands auto generated by Alembic - please adjust! ###
op.add_column(
u'oauthaccesstoken',
sa.Column('refresh_token', sa.String(length=255), nullable=True))
op.create_index('oauthaccesstoken_refresh_token',
'oauthaccesstoken', ['refresh_token'],
unique=False)
op.add_column(
'repositorybuildtrigger',
sa.Column('auth_token', sa.String(length=255), nullable=True))
op.add_column('repositorybuildtrigger',
sa.Column('private_key', sa.Text(), nullable=True))
op.add_column('oauthauthorizationcode',
sa.Column('code', sa.String(length=255), nullable=True))
op.create_index('oauthauthorizationcode_code',
'oauthauthorizationcode', ['code'],
unique=True)
op.add_column(
'oauthapplication',
sa.Column('client_secret', sa.String(length=255), nullable=True))
op.add_column(
'oauthaccesstoken',
sa.Column('access_token', sa.String(length=255), nullable=True))
op.create_index('oauthaccesstoken_access_token',
'oauthaccesstoken', ['access_token'],
unique=False)
op.add_column(
'appspecificauthtoken',
sa.Column('token_code', sa.String(length=255), nullable=True))
op.create_index('appspecificauthtoken_token_code',
'appspecificauthtoken', ['token_code'],
unique=True)
op.add_column('accesstoken',
sa.Column('code', sa.String(length=255), nullable=True))
op.create_index('accesstoken_code', 'accesstoken', ['code'], unique=True)
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
op.create_table('repomirrorrule',
sa.Column('id', sa.Integer(), nullable=False),
sa.Column('uuid', sa.String(length=36), nullable=False),
sa.Column('repository_id', sa.Integer(), nullable=False),
sa.Column('creation_date', sa.DateTime(), nullable=False),
sa.Column('rule_type', sa.Integer(), nullable=False),
sa.Column('rule_value', sa.Text(), nullable=False),
sa.Column('left_child_id', sa.Integer(), nullable=True),
sa.Column('right_child_id', sa.Integer(), nullable=True),
sa.ForeignKeyConstraint(['left_child_id'], ['repomirrorrule.id'], name=op.f('fk_repomirrorrule_left_child_id_repomirrorrule')),
sa.ForeignKeyConstraint(['repository_id'], ['repository.id'], name=op.f('fk_repomirrorrule_repository_id_repository')),
sa.ForeignKeyConstraint(['right_child_id'], ['repomirrorrule.id'], name=op.f('fk_repomirrorrule_right_child_id_repomirrorrule')),
sa.PrimaryKeyConstraint('id', name=op.f('pk_repomirrorrule')))
op.create_index('repomirrorrule_left_child_id', 'repomirrorrule', ['left_child_id'], unique=False)
op.create_index('repomirrorrule_repository_id', 'repomirrorrule', ['repository_id'], unique=False)
op.create_index('repomirrorrule_right_child_id', 'repomirrorrule', ['right_child_id'], unique=False)
op.create_index('repomirrorrule_rule_type', 'repomirrorrule', ['rule_type'], unique=False)
op.create_index('repomirrorrule_uuid', 'repomirrorrule', ['uuid'], unique=True)
op.create_table('repomirrorconfig',
sa.Column('id', sa.Integer(), nullable=False),
sa.Column('repository_id', sa.Integer(), nullable=False),
sa.Column('creation_date', sa.DateTime(), nullable=False),
sa.Column('is_enabled', sa.Boolean(), nullable=False),
sa.Column('mirror_type', sa.Integer(), nullable=False),
sa.Column('internal_robot_id', sa.Integer(), nullable=False),
sa.Column('external_registry', sa.String(length=255), nullable=False),
sa.Column('external_namespace', sa.String(length=255), nullable=False),
sa.Column('external_repository', sa.String(length=255), nullable=False),
sa.Column('external_registry_username', sa.String(length=2048), nullable=True),
sa.Column('external_registry_password', sa.String(length=2048), nullable=True),
sa.Column('external_registry_config', sa.Text(), nullable=False),
sa.Column('sync_interval', sa.Integer(), nullable=False, server_default='60'),
sa.Column('sync_start_date', sa.DateTime(), nullable=True),
sa.Column('sync_expiration_date', sa.DateTime(), nullable=True),
sa.Column('sync_retries_remaining', sa.Integer(), nullable=False, server_default='3'),
sa.Column('sync_status', sa.Integer(), nullable=False),
sa.Column('sync_transaction_id', sa.String(length=36), nullable=True),
sa.Column('root_rule_id', sa.Integer(), nullable=False),
sa.ForeignKeyConstraint(['repository_id'], ['repository.id'], name=op.f('fk_repomirrorconfig_repository_id_repository')),
sa.ForeignKeyConstraint(['root_rule_id'], ['repomirrorrule.id'], name=op.f('fk_repomirrorconfig_root_rule_id_repomirrorrule')),
sa.ForeignKeyConstraint(['internal_robot_id'], ['user.id'], name=op.f('fk_repomirrorconfig_internal_robot_id_user')),
sa.PrimaryKeyConstraint('id', name=op.f('pk_repomirrorconfig'))
)
op.create_index('repomirrorconfig_mirror_type', 'repomirrorconfig', ['mirror_type'], unique=False)
op.create_index('repomirrorconfig_repository_id', 'repomirrorconfig', ['repository_id'], unique=True)
op.create_index('repomirrorconfig_root_rule_id', 'repomirrorconfig', ['root_rule_id'], unique=False)
op.create_index('repomirrorconfig_sync_status', 'repomirrorconfig', ['sync_status'], unique=False)
op.create_index('repomirrorconfig_sync_transaction_id', 'repomirrorconfig', ['sync_transaction_id'], unique=False)
op.create_index('repomirrorconfig_internal_robot_id', 'repomirrorconfig', ['internal_robot_id'], unique=False)
op.add_column(u'repository', sa.Column('state', sa.Integer(), nullable=False, server_default='0'))
op.create_index('repository_state', 'repository', ['state'], unique=False)
op.bulk_insert(tables.logentrykind,
[
{'name': 'repo_mirror_enabled'},
{'name': 'repo_mirror_disabled'},
{'name': 'repo_mirror_config_changed'},
{'name': 'repo_mirror_sync_started'},
{'name': 'repo_mirror_sync_failed'},
{'name': 'repo_mirror_sync_success'},
{'name': 'repo_mirror_sync_now_requested'},
{'name': 'repo_mirror_sync_tag_success'},
{'name': 'repo_mirror_sync_tag_failed'},
{'name': 'repo_mirror_sync_test_success'},
{'name': 'repo_mirror_sync_test_failed'},
{'name': 'repo_mirror_sync_test_started'},
{'name': 'change_repo_state'}
])
tester.populate_table('repomirrorrule', [
('uuid', tester.TestDataType.String),
('repository_id', tester.TestDataType.Foreign('repository')),
('creation_date', tester.TestDataType.DateTime),
('rule_type', tester.TestDataType.Integer),
('rule_value', tester.TestDataType.String),
])
tester.populate_table('repomirrorconfig', [
('repository_id', tester.TestDataType.Foreign('repository')),
('creation_date', tester.TestDataType.DateTime),
('is_enabled', tester.TestDataType.Boolean),
('mirror_type', tester.TestDataType.Constant(1)),
('internal_robot_id', tester.TestDataType.Foreign('user')),
('external_registry', tester.TestDataType.String),
('external_namespace', tester.TestDataType.String),
('external_repository', tester.TestDataType.String),
('external_registry_username', tester.TestDataType.String),
('external_registry_password', tester.TestDataType.String),
('external_registry_config', tester.TestDataType.JSON),
('sync_start_date', tester.TestDataType.DateTime),
('sync_expiration_date', tester.TestDataType.DateTime),
('sync_retries_remaining', tester.TestDataType.Integer),
('sync_status', tester.TestDataType.Constant(0)),
('sync_transaction_id', tester.TestDataType.String),
('root_rule_id', tester.TestDataType.Foreign('repomirrorrule')),
])
def upgrade(tables, tester, progress_reporter):
op = ProgressWrapper(original_op, progress_reporter)
op.create_table(
"repomirrorrule",
sa.Column("id", sa.Integer(), nullable=False),
sa.Column("uuid", sa.String(length=36), nullable=False),
sa.Column("repository_id", sa.Integer(), nullable=False),
sa.Column("creation_date", sa.DateTime(), nullable=False),
sa.Column("rule_type", sa.Integer(), nullable=False),
sa.Column("rule_value", sa.Text(), nullable=False),
sa.Column("left_child_id", sa.Integer(), nullable=True),
sa.Column("right_child_id", sa.Integer(), nullable=True),
sa.ForeignKeyConstraint(
["left_child_id"],
["repomirrorrule.id"],
name=op.f("fk_repomirrorrule_left_child_id_repomirrorrule"),
),
sa.ForeignKeyConstraint(
["repository_id"],
["repository.id"],
name=op.f("fk_repomirrorrule_repository_id_repository"),
),
sa.ForeignKeyConstraint(
["right_child_id"],
["repomirrorrule.id"],
name=op.f("fk_repomirrorrule_right_child_id_repomirrorrule"),
),
sa.PrimaryKeyConstraint("id", name=op.f("pk_repomirrorrule")),
)
op.create_index("repomirrorrule_left_child_id",
"repomirrorrule", ["left_child_id"],
unique=False)
op.create_index("repomirrorrule_repository_id",
"repomirrorrule", ["repository_id"],
unique=False)
op.create_index("repomirrorrule_right_child_id",
"repomirrorrule", ["right_child_id"],
unique=False)
op.create_index("repomirrorrule_rule_type",
"repomirrorrule", ["rule_type"],
unique=False)
op.create_index("repomirrorrule_uuid",
"repomirrorrule", ["uuid"],
unique=True)
op.create_table(
"repomirrorconfig",
sa.Column("id", sa.Integer(), nullable=False),
sa.Column("repository_id", sa.Integer(), nullable=False),
sa.Column("creation_date", sa.DateTime(), nullable=False),
sa.Column("is_enabled", sa.Boolean(), nullable=False),
sa.Column("mirror_type", sa.Integer(), nullable=False),
sa.Column("internal_robot_id", sa.Integer(), nullable=False),
sa.Column("external_registry", sa.String(length=255), nullable=False),
sa.Column("external_namespace", sa.String(length=255), nullable=False),
sa.Column("external_repository", sa.String(length=255),
nullable=False),
sa.Column("external_registry_username",
sa.String(length=2048),
nullable=True),
sa.Column("external_registry_password",
sa.String(length=2048),
nullable=True),
sa.Column("external_registry_config", sa.Text(), nullable=False),
sa.Column("sync_interval",
sa.Integer(),
nullable=False,
server_default="60"),
sa.Column("sync_start_date", sa.DateTime(), nullable=True),
sa.Column("sync_expiration_date", sa.DateTime(), nullable=True),
sa.Column("sync_retries_remaining",
sa.Integer(),
nullable=False,
server_default="3"),
sa.Column("sync_status", sa.Integer(), nullable=False),
sa.Column("sync_transaction_id", sa.String(length=36), nullable=True),
sa.Column("root_rule_id", sa.Integer(), nullable=False),
sa.ForeignKeyConstraint(
["repository_id"],
["repository.id"],
name=op.f("fk_repomirrorconfig_repository_id_repository"),
),
sa.ForeignKeyConstraint(
["root_rule_id"],
["repomirrorrule.id"],
name=op.f("fk_repomirrorconfig_root_rule_id_repomirrorrule"),
),
sa.ForeignKeyConstraint(
["internal_robot_id"],
["user.id"],
name=op.f("fk_repomirrorconfig_internal_robot_id_user"),
),
sa.PrimaryKeyConstraint("id", name=op.f("pk_repomirrorconfig")),
)
op.create_index("repomirrorconfig_mirror_type",
"repomirrorconfig", ["mirror_type"],
unique=False)
op.create_index("repomirrorconfig_repository_id",
"repomirrorconfig", ["repository_id"],
unique=True)
op.create_index("repomirrorconfig_root_rule_id",
"repomirrorconfig", ["root_rule_id"],
unique=False)
op.create_index("repomirrorconfig_sync_status",
"repomirrorconfig", ["sync_status"],
unique=False)
op.create_index(
"repomirrorconfig_sync_transaction_id",
"repomirrorconfig",
["sync_transaction_id"],
unique=False,
)
op.create_index(
"repomirrorconfig_internal_robot_id",
"repomirrorconfig",
["internal_robot_id"],
unique=False,
)
op.add_column(
u"repository",
sa.Column("state", sa.Integer(), nullable=False, server_default="0"))
op.create_index("repository_state", "repository", ["state"], unique=False)
op.bulk_insert(
tables.logentrykind,
[
{
"name": "repo_mirror_enabled"
},
{
"name": "repo_mirror_disabled"
},
{
"name": "repo_mirror_config_changed"
},
{
"name": "repo_mirror_sync_started"
},
{
"name": "repo_mirror_sync_failed"
},
{
"name": "repo_mirror_sync_success"
},
{
"name": "repo_mirror_sync_now_requested"
},
{
"name": "repo_mirror_sync_tag_success"
},
{
"name": "repo_mirror_sync_tag_failed"
},
{
"name": "repo_mirror_sync_test_success"
},
{
"name": "repo_mirror_sync_test_failed"
},
{
"name": "repo_mirror_sync_test_started"
},
{
"name": "change_repo_state"
},
],
)
tester.populate_table(
"repomirrorrule",
[
("uuid", tester.TestDataType.String),
("repository_id", tester.TestDataType.Foreign("repository")),
("creation_date", tester.TestDataType.DateTime),
("rule_type", tester.TestDataType.Integer),
("rule_value", tester.TestDataType.String),
],
)
tester.populate_table(
"repomirrorconfig",
[
("repository_id", tester.TestDataType.Foreign("repository")),
("creation_date", tester.TestDataType.DateTime),
("is_enabled", tester.TestDataType.Boolean),
("mirror_type", tester.TestDataType.Constant(1)),
("internal_robot_id", tester.TestDataType.Foreign("user")),
("external_registry", tester.TestDataType.String),
("external_namespace", tester.TestDataType.String),
("external_repository", tester.TestDataType.String),
("external_registry_username", tester.TestDataType.String),
("external_registry_password", tester.TestDataType.String),
("external_registry_config", tester.TestDataType.JSON),
("sync_start_date", tester.TestDataType.DateTime),
("sync_expiration_date", tester.TestDataType.DateTime),
("sync_retries_remaining", tester.TestDataType.Integer),
("sync_status", tester.TestDataType.Constant(0)),
("sync_transaction_id", tester.TestDataType.String),
("root_rule_id", tester.TestDataType.Foreign("repomirrorrule")),
],
)
