Add code for saving and displaying the size of images

data/database.py

@@ -168,6 +168,7 @@ class Image(BaseModel):
   created = DateTimeField(null=True)
   comment = TextField(null=True)
   repository = ForeignKeyField(Repository)
+  image_size = BigIntegerField(null=True)
 
   # '/' separated list of ancestory ids, e.g. /1/2/6/7/10/
   ancestors = CharField(index=True, default='/', max_length=64535)

data/model.py

@@ -706,6 +706,21 @@ def set_image_checksum(docker_image_id, repository, checksum):
   return fetched
 
 
+def set_image_size(docker_image_id, namespace_name, repository_name, image_size):
+  joined = Image.select().join(Repository)
+  image_list = list(joined.where(Repository.name == repository_name,
+                                 Repository.namespace == namespace_name,
+                                 Image.docker_image_id == docker_image_id))
+
+  if not image_list:
+    raise DataModelException('No image with specified id and repository')
+
+  fetched = image_list[0]
+  fetched.image_size = image_size
+  fetched.save()
+  return fetched
+
+
 def set_image_metadata(docker_image_id, namespace_name, repository_name,
                        created_date_str, comment, parent=None):
   joined = Image.select().join(Repository)