quay/data/migrations/versions/3f4fe1194671_backfill_the_namespace_user_fields.py

"""Backfill the namespace_user fields.

Revision ID: 3f4fe1194671
Revises: 6f2ecf5afcf
Create Date: 2014-09-24 14:29:45.192179

"""

# revision identifiers, used by Alembic.
revision = '3f4fe1194671'
down_revision = '6f2ecf5afcf'

from alembic import op
import sqlalchemy as sa


def upgrade(tables):
    conn = op.get_bind()
    user_table_name_escaped = conn.dialect.identifier_preparer.format_table(tables['user'])
    conn.execute('update repository set namespace_user_id = (select id from {0} where {0}.username = repository.namespace) where namespace_user_id is NULL'.format(user_table_name_escaped))
    op.create_index('repository_namespace_user_id_name', 'repository', ['namespace_user_id', 'name'], unique=True)


def downgrade(tables):
    op.drop_constraint('fk_repository_namespace_user_id_user', table_name='repository', type_='foreignkey')
    op.drop_index('repository_namespace_user_id_name', table_name='repository')
Phase 2 of migrating repo namespaces to referencing user objects, backfilling the rows without a value for namespace_user, and changing all accesses to go through the namespace_user object. All tests are passing, manual testing still required. 2014-09-24 22:01:35 +00:00			`"""Backfill the namespace_user fields.`

			`Revision ID: 3f4fe1194671`
			`Revises: 6f2ecf5afcf`
			`Create Date: 2014-09-24 14:29:45.192179`

			`"""`

			`# revision identifiers, used by Alembic.`
			`revision = '3f4fe1194671'`
			`down_revision = '6f2ecf5afcf'`

			`from alembic import op`
			`import sqlalchemy as sa`


			`def upgrade(tables):`
			`conn = op.get_bind()`
Fix all of the upgrades and downgrades to work on both mysql and postgres. 2014-10-06 20:40:44 +00:00			`user_table_name_escaped = conn.dialect.identifier_preparer.format_table(tables['user'])`
			`conn.execute('update repository set namespace_user_id = (select id from {0} where {0}.username = repository.namespace) where namespace_user_id is NULL'.format(user_table_name_escaped))`
Phase 2 of migrating repo namespaces to referencing user objects, backfilling the rows without a value for namespace_user, and changing all accesses to go through the namespace_user object. All tests are passing, manual testing still required. 2014-09-24 22:01:35 +00:00			`op.create_index('repository_namespace_user_id_name', 'repository', ['namespace_user_id', 'name'], unique=True)`


			`def downgrade(tables):`
Allow the namespace column to be null, and also non-unique. Fix the uncompressed size clobbering the size on the wire field. Add metadata constraints so that foreign key constraints get predictable names. Fix all downgrade migrations. 2014-10-02 14:46:20 +00:00			`op.drop_constraint('fk_repository_namespace_user_id_user', table_name='repository', type_='foreignkey')`
Phase 2 of migrating repo namespaces to referencing user objects, backfilling the rows without a value for namespace_user, and changing all accesses to go through the namespace_user object. All tests are passing, manual testing still required. 2014-09-24 22:01:35 +00:00			`op.drop_index('repository_namespace_user_id_name', table_name='repository')`