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.

data/migrations/versions/3f4fe1194671_backfill_the_namespace_user_fields.py

@@ -22,4 +22,5 @@ def upgrade(tables):
 
 
 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')

data/migrations/versions/51d04d0e7e6f_email_invites_for_joining_a_team.py

@@ -74,8 +74,5 @@ def downgrade(tables):
          .where(tables.notificationkind.c.name == op.inline_literal('org_team_invite')))
         )
 
-    op.drop_index('teammemberinvite_user_id', table_name='teammemberinvite')
-    op.drop_index('teammemberinvite_team_id', table_name='teammemberinvite')
-    op.drop_index('teammemberinvite_inviter_id', table_name='teammemberinvite')
     op.drop_table('teammemberinvite')
     ### end Alembic commands ###

data/migrations/versions/9a1087b007d_allow_the_namespace_column_to_be_.py

@@ -0,0 +1,28 @@
+"""Allow the namespace column to be nullable.
+
+Revision ID: 9a1087b007d
+Revises: 3f4fe1194671
+Create Date: 2014-10-01 16:11:21.277226
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '9a1087b007d'
+down_revision = '3f4fe1194671'
+
+from alembic import op
+import sqlalchemy as sa
+
+
+def upgrade(tables):
+    op.drop_index('repository_namespace_name', table_name='repository')
+    op.alter_column('repository', 'namespace', nullable=True, existing_type=sa.String(length=255),
+                    server_default=sa.text('NULL'))
+
+
+def downgrade(tables):
+    conn = op.get_bind()
+    conn.execute('update repository set namespace = (select username from user where user.id = repository.namespace_user_id) where namespace is NULL')
+
+    op.create_index('repository_namespace_name', 'repository', ['namespace', 'name'], unique=True)
+    op.alter_column('repository', 'namespace', nullable=False, existing_type=sa.String(length=255))

data/model/legacy.py

@@ -1223,8 +1223,7 @@ def get_storage_by_uuid(storage_uuid):
   return found
 
 
-def set_image_size(docker_image_id, namespace_name, repository_name,
-                   image_size):
+def set_image_size(docker_image_id, namespace_name, repository_name, image_size, uncompressed_size):
   try:
     image = (Image
              .select(Image, ImageStorage)
@@ -1233,18 +1232,15 @@ def set_image_size(docker_image_id, namespace_name, repository_name,
              .switch(Image)
              .join(ImageStorage, JOIN_LEFT_OUTER)
              .where(Repository.name == repository_name, Namespace.username == namespace_name,
-                   Image.docker_image_id == docker_image_id)
+                    Image.docker_image_id == docker_image_id)
              .get())
 
   except Image.DoesNotExist:
     raise DataModelException('No image with specified id and repository')
 
-  if image.storage and image.storage.id:
-    image.storage.image_size = image_size
-    image.storage.save()
-  else:
-    image.image_size = image_size
-    image.save()
+  image.storage.image_size = image_size
+  image.storage.uncompressed_size = uncompressed_size
+  image.storage.save()
 
   return image
 

data/model/sqlalchemybridge.py

@@ -17,7 +17,12 @@ OPTION_TRANSLATIONS = {
 
 
 def gen_sqlalchemy_metadata(peewee_model_list):
-  metadata = MetaData()
+  metadata = MetaData(naming_convention={
+    "ix": 'ix_%(column_0_label)s',
+    "uq": "uq_%(table_name)s_%(column_0_name)s",
+    "fk": "fk_%(table_name)s_%(column_0_name)s_%(referred_table_name)s",
+    "pk": "pk_%(table_name)s"
+  })
 
   for model in peewee_model_list:
     meta = model._meta