Merge remote-tracking branch 'upstream/v2-phase4' into python-registry-v2

data/database.py

@@ -3,9 +3,7 @@ import logging
 import uuid
 import time
 import toposort
-import base64
 import resumablehashlib
-import json
 
 from random import SystemRandom
 from datetime import datetime
@@ -15,7 +13,6 @@ from data.fields import ResumableSHAField, JSONField
 from sqlalchemy.engine.url import make_url
 from collections import defaultdict
 
-from data.read_slave import ReadSlaveModel
 from util.names import urn_generator
 
 

data/migrations/migration.sh

@@ -108,6 +108,16 @@ test_migrate $MYSQL_CONFIG_OVERRIDE
 set -e
 down_mysql
 
+# Test via Postgres.
+echo '> Starting Postgres'
+up_postgres
+
+echo '> Testing Migration (postgres)'
+set +e
+test_migrate $PGSQL_CONFIG_OVERRIDE
+set -e
+down_postgres
+
 # Test via MariaDB.
 echo '> Starting MariaDB'
 up_mariadb
@@ -127,13 +137,3 @@ set +e
 test_migrate $PERCONA_CONFIG_OVERRIDE
 set -e
 down_percona
-
-# Test via Postgres.
-echo '> Starting Postgres'
-up_postgres
-
-echo '> Testing Migration (postgres)'
-set +e
-test_migrate $PGSQL_CONFIG_OVERRIDE
-set -e
-down_postgres

data/migrations/versions/127905a52fdd_remove_the_deprecated_imagestorage_.py

@@ -12,7 +12,6 @@ down_revision = '2e0380215d01'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###
@@ -25,8 +24,8 @@ def upgrade(tables):
 
 def downgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###
-    op.add_column('imagestorage', sa.Column('created', mysql.DATETIME(), nullable=True))
+    op.add_column('imagestorage', sa.Column('created', sa.DateTime(), nullable=True))
-    op.add_column('imagestorage', sa.Column('command', mysql.TEXT(), nullable=True))
+    op.add_column('imagestorage', sa.Column('command', sa.Text(), nullable=True))
-    op.add_column('imagestorage', sa.Column('aggregate_size', mysql.BIGINT(display_width=20), autoincrement=False, nullable=True))
+    op.add_column('imagestorage', sa.Column('aggregate_size', sa.BigInteger(), nullable=True))
-    op.add_column('imagestorage', sa.Column('comment', mysql.TEXT(), nullable=True))
+    op.add_column('imagestorage', sa.Column('comment', sa.Text(), nullable=True))
     ### end Alembic commands ###

data/migrations/versions/14fe12ade3df_add_build_queue_item_reference_to_the_.py

@@ -12,7 +12,6 @@ down_revision = '5ad999136045'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###

data/migrations/versions/1594a74a74ca_add_metadata_field_to_external_logins.py

@@ -12,7 +12,6 @@ down_revision = 'f42b0ea7a4d'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###

data/migrations/versions/17f11e265e13_add_uuid_field_to_user.py

@@ -12,7 +12,6 @@ down_revision = '313d297811c4'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 
 def upgrade(tables):

data/migrations/versions/1c5b738283a5_backfill_user_uuids.py

@@ -12,7 +12,6 @@ down_revision = '2fb36d4be80d'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 from util.migrate.backfill_user_uuids import backfill_user_uuids
 
 def upgrade(tables):

data/migrations/versions/1d2d86d09fcd_actually_remove_the_column.py

@@ -12,7 +12,6 @@ down_revision = '14fe12ade3df'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 from sqlalchemy.exc import InternalError
 
 def upgrade(tables):
@@ -29,7 +28,7 @@ def upgrade(tables):
 def downgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###
     try:
-        op.add_column('logentry', sa.Column('access_token_id', mysql.INTEGER(display_width=11), autoincrement=False, nullable=True))
+        op.add_column('logentry', sa.Column('access_token_id', sa.Integer(), nullable=True))
         op.create_foreign_key(u'fk_logentry_access_token_id_accesstoken', 'logentry', 'accesstoken', ['access_token_id'], ['id'])
         op.create_index('logentry_access_token_id', 'logentry', ['access_token_id'], unique=False)
     except InternalError:

data/migrations/versions/201d55b38649_remove_fields_from_image_table_that_.py

@@ -12,7 +12,6 @@ down_revision = '5a07499ce53f'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###

data/migrations/versions/2088f2b81010_add_stars.py

@@ -12,7 +12,6 @@ down_revision = '707d5191eda'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     op.create_table('star',

data/migrations/versions/228d1af6af1c_mysql_max_index_lengths.py

@@ -12,7 +12,6 @@ down_revision = '5b84373e5db'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
   op.drop_index('queueitem_queue_name', table_name='queueitem')

data/migrations/versions/31288f79df53_make_resource_key_nullable.py

@@ -12,12 +12,11 @@ down_revision = '214350b6a8b1'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###
     op.alter_column('repositorybuild', 'resource_key',
-               existing_type=mysql.VARCHAR(length=255),
+               existing_type=sa.String(length=255),
                nullable=True)
     ### end Alembic commands ###
 
@@ -25,6 +24,6 @@ def upgrade(tables):
 def downgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###
     op.alter_column('repositorybuild', 'resource_key',
-               existing_type=mysql.VARCHAR(length=255),
+               existing_type=sa.String(length=255),
                nullable=False)
     ### end Alembic commands ###

data/migrations/versions/313d297811c4_add_an_index_to_the_docker_image_id_.py

@@ -12,7 +12,6 @@ down_revision = '204abf14783d'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###

data/migrations/versions/325a4d7c79d9_prepare_the_database_for_the_new_.py

@@ -12,7 +12,6 @@ down_revision = '4b7ef0c7bdb2'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###

data/migrations/versions/33bd39ef5ed6_backport_v2_db_changes.py

@@ -0,0 +1,43 @@
+"""Backport v2 db changes.
+
+Revision ID: 33bd39ef5ed6
+Revises: 127905a52fdd
+Create Date: 2015-10-23 12:34:22.776542
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '33bd39ef5ed6'
+down_revision = '127905a52fdd'
+
+from alembic import op
+import sqlalchemy as sa
+
+
+def upgrade(tables):
+    ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('blobupload',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('repository_id', sa.Integer(), nullable=False),
+    sa.Column('uuid', sa.String(length=255), nullable=False),
+    sa.Column('byte_count', sa.Integer(), nullable=False),
+    sa.Column('sha_state', sa.Text(), nullable=True),
+    sa.Column('location_id', sa.Integer(), nullable=False),
+    sa.Column('storage_metadata', sa.Text(), nullable=True),
+    sa.ForeignKeyConstraint(['location_id'], ['imagestoragelocation.id'], name=op.f('fk_blobupload_location_id_imagestoragelocation')),
+    sa.ForeignKeyConstraint(['repository_id'], ['repository.id'], name=op.f('fk_blobupload_repository_id_repository')),
+    sa.PrimaryKeyConstraint('id', name=op.f('pk_blobupload'))
+    )
+    op.create_index('blobupload_location_id', 'blobupload', ['location_id'], unique=False)
+    op.create_index('blobupload_repository_id', 'blobupload', ['repository_id'], unique=False)
+    op.create_index('blobupload_repository_id_uuid', 'blobupload', ['repository_id', 'uuid'], unique=True)
+    op.create_index('blobupload_uuid', 'blobupload', ['uuid'], unique=True)
+    op.add_column(u'imagestorage', sa.Column('cas_path', sa.Boolean(), nullable=False, server_default="0"))
+    ### end Alembic commands ###
+
+
+def downgrade(tables):
+    ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column(u'imagestorage', 'cas_path')
+    op.drop_table('blobupload')
+    ### end Alembic commands ###

data/migrations/versions/35f538da62_switch_manifest_text_to_a_longtext.py

@@ -0,0 +1,47 @@
+"""Switch manifest text to a longtext.
+
+Revision ID: 35f538da62
+Revises: 33bd39ef5ed6
+Create Date: 2015-10-23 15:31:27.353995
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '35f538da62'
+down_revision = '33bd39ef5ed6'
+
+from alembic import op
+import sqlalchemy as sa
+
+from sqlalchemy.types import TypeDecorator, Text
+from sqlalchemy.dialects.mysql import LONGTEXT
+import uuid
+
+class EngineLongText(TypeDecorator):
+    """Platform-independent LongText type.
+
+    Uses MySQL's LONGTEXT type, otherwise uses
+    Text, because other engines are not as limited
+    as MySQL.
+
+    """
+    impl = Text
+
+    def load_dialect_impl(self, dialect):
+        if dialect.name == 'mysql':
+            return dialect.type_descriptor(LONGTEXT())
+        else:
+            return dialect.type_descriptor(Text())
+
+def upgrade(tables):
+    ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column(u'tagmanifest', 'json_data')
+    op.add_column(u'tagmanifest', sa.Column('json_data', EngineLongText(), nullable=False))
+    ### end Alembic commands ###
+
+
+def downgrade(tables):
+    ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column(u'tagmanifest', 'json_data')
+    op.add_column(u'tagmanifest', sa.Column('json_data', sa.Text(), nullable=False))
+    ### end Alembic commands ###

data/migrations/versions/3fee6f979c2a_make_auth_token_nullable.py

@@ -12,12 +12,11 @@ down_revision = '31288f79df53'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###
     op.alter_column('repositorybuildtrigger', 'auth_token',
-               existing_type=mysql.VARCHAR(length=255),
+               existing_type=sa.String(length=255),
                nullable=True)
     ### end Alembic commands ###
 
@@ -25,6 +24,6 @@ def upgrade(tables):
 def downgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###
     op.alter_column('repositorybuildtrigger', 'auth_token',
-               existing_type=mysql.VARCHAR(length=255),
+               existing_type=sa.String(length=255),
                nullable=False)
     ### end Alembic commands ###

data/migrations/versions/4a0c94399f38_add_new_notification_kinds.py

@@ -12,7 +12,6 @@ down_revision = '1594a74a74ca'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     op.bulk_insert(tables.externalnotificationmethod,

data/migrations/versions/4fdb65816b8d_add_brute_force_prevention_metadata_to_.py

@@ -12,7 +12,6 @@ down_revision = '43e943c0639f'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###

data/migrations/versions/5ad999136045_add_signature_storage.py

@@ -12,7 +12,6 @@ down_revision = '228d1af6af1c'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###

data/migrations/versions/707d5191eda_change_build_queue_reference_from_.py

@@ -12,7 +12,6 @@ down_revision = '4ef04c61fcf9'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###
@@ -26,7 +25,7 @@ def upgrade(tables):
 
 def downgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###
-    op.add_column('repositorybuild', sa.Column('queue_item_id', mysql.INTEGER(display_width=11), autoincrement=False, nullable=True))
+    op.add_column('repositorybuild', sa.Column('queue_item_id', sa.Integer(), autoincrement=False, nullable=True))
     op.create_foreign_key(u'fk_repositorybuild_queue_item_id_queueitem', 'repositorybuild', 'queueitem', ['queue_item_id'], ['id'])
     op.create_index('repositorybuild_queue_item_id', 'repositorybuild', ['queue_item_id'], unique=False)
     op.drop_index('repositorybuild_queue_id', table_name='repositorybuild')

data/migrations/versions/82297d834ad_add_us_west_location.py

@@ -12,7 +12,6 @@ down_revision = '47670cbeced'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     op.bulk_insert(tables.imagestoragelocation,

data/migrations/versions/f42b0ea7a4d_remove_the_old_webhooks_table.py

@@ -12,7 +12,6 @@ down_revision = '4fdb65816b8d'
 
 from alembic import op
 import sqlalchemy as sa
-from sqlalchemy.dialects import mysql
 
 def upgrade(tables):
     ### commands auto generated by Alembic - please adjust! ###