v2/manifest: refactor to use types

data/model/v2.py

@@ -0,0 +1,102 @@
+from data.types import (
+  Repository,
+  Tag,
+  ManifestJSON,
+  DockerV1Metadata,
+)
+
+def get_repository(namespace_name, repo_name):
+  repo = model.repository.get_repository(namespace_name, repo_name)
+  if repo is None:
+    return None
+
+  return Repository(
+    id=repo.id,
+    name=repo.name,
+    namespace_name=repo.namespace_user.username,
+  )
+
+
+def get_active_tag(namespace_name, repo_name, tag_name):
+  try:
+    return model.tag.get_active_tag(namespace_name, repo_name, tag_name)
+  except RepositoryTag.DoesNotExist:
+    return None
+
+
+def get_manifest_by_tag(namespace_name, repo_name, tag_name):
+  try:
+    manifest = model.tag.load_tag_manifest(namespace_name, repo_name, manifest_ref)
+    return ManifestJSON(digest=digest, json=manifest.json_data)
+  except model.InvalidManifestException:
+    return None
+
+
+def get_manifest_by_digest(namespace_name, repo_name, digest):
+  try:
+    manifest = model.tag.load_manifest_by_digest(namespace_name, repo_name, manifest_ref)
+    return ManifestJSON(digest=digest, json=manifest.json_data)
+  except model.InvalidManifestException:
+    return None
+
+
+def get_tag_by_manifest_digest(namespace_name, repo_name, digest):
+  return Tag()
+
+
+def delete_tag(namespace_name, repo_name, tag_name):
+  model.tag.delete_tag(namespace_name, repo_name, tag.name)
+  return True
+
+
+def docker_v1_metadata_by_tag(namespace_name, repo_name, tag_name):
+  if not repo_image:
+    return None
+
+  return DockerV1Metadata(
+    namespace_name=namespace_name,
+    repo_name=repo_name,
+    image_id=image_id,
+    checksum=repo_image.v1_checksum,
+    content_checksum=repo_image.content_checksum,
+    compat_json=repo_image.v1_json_metadata,
+  )
+
+
+def docker_v1_metadata_by_image_id(namespace_name, repo_name, image_ids):
+  images_query = model.image.lookup_repository_images(repo, all_image_ids)
+  return [DockerV1Metadata(
+    namespace_name=namespace_name,
+    repo_name=repo_name,
+    image_id=image.docker_image_id,
+    checksum=image.v1_checksum,
+    content_checksum=image.content_checksum,
+    compat_json=image.v1_json_metadata,
+  ) for image in images_query]
+
+
+def get_parents_docker_v1_metadata(namespace_name, repo_name, image_id):
+  # Old implementation:
+  # parents = model.image.get_parent_images(namespace_name, repo_name, image)
+
+  # desired:
+  # return a list of the AttrDict in docker_v1_metadata
+  return []
+
+
+def create_manifest_and_update_tag(namespace_name, repo_name, tag_name, manifest_digest, manifest_bytes):
+  try:
+    model.tag.associate_generated_tag_manifest(namespace_name, repo_name, tag_name,
+                                               manifest.digest, manifest.bytes)
+  except IntegrityError:
+    # It's already there!
+    pass
+
+
+def synthesize_v1_image(repo, storage, image_id, created, comment, command, compat_json, parent_image_id):
+  model.image.synthesize_v1_image(repo, storage, image_id, created, comment, command, compat_json, parent_image_id)
+
+
+def save_manifest(namespace_name, repo_name, tag_name, leaf_layer_id, manifest_digest, manifest_bytes):
+  model.tag.store_tag_manifest(namespace_name, repo_name, tag_name, leaf_layer_id, manifest_digest,
+                               manifest_bytes)