data.oci_model: init with app methods

data/oci_model/manifest.py

@@ -0,0 +1,55 @@
+import logging
+import hashlib
+import json
+
+from cnr.models.package_base import get_media_type
+
+from data import oci_model
+from data.database import db_transaction, Manifest, ManifestListManifest, MediaType, Blob, Tag
+
+
+logger = logging.getLogger(__name__)
+
+
+def _digest(manifestjson):
+  return hashlib.sha256(json.dumps(manifestjson, sort_keys=True)).hexdigest()
+
+
+def get_manifest_query(digest, media_type):
+  return Manifest.select().where(Manifest.digest == digest,
+                                 Manifest.media_type == Manifest.media_type.get_id(media_type))
+
+
+def get_manifest_with_blob(digest, media_type):
+  query = get_manifest_query(digest, media_type)
+  return query.join(Blob).get()
+
+
+def get_or_create_manifest(manifest_json, media_type_name):
+  digest = _digest(manifest_json)
+  try:
+    manifest = get_manifest_query(digest, media_type_name).get()
+  except Manifest.DoesNotExist:
+    with db_transaction():
+      manifest = Manifest.create(digest=digest,
+                                 manifest_json=manifest_json,
+                                 media_type=Manifest.media_type.get_id(media_type_name))
+  return manifest
+
+def get_manifest_types(repo, release=None):
+  """ Returns an array of MediaTypes.name for a repo, can filter by tag """
+  query = oci_model.tag.tag_alive_oci(Tag
+                                      .select(MediaType.name)
+                                      .join(ManifestListManifest,
+                                            on=(ManifestListManifest.manifest_list == Tag.manifest_list))
+                                      .join(MediaType,
+                                            on=(ManifestListManifest.media_type == MediaType.id))
+                                      .where(Tag.repository == repo,
+                                             Tag.tag_kind == Tag.tag_kind.get_id('release')))
+  if release:
+    query = query.where(Tag.name == release)
+
+  manifests = set()
+  for m in query.distinct().tuples():
+    manifests.add(get_media_type(m[0]))
+  return manifests