data.oci_model: init with app methods

data/oci_model/manifest_list.py

@@ -0,0 +1,55 @@
+import logging
+import hashlib
+import json
+
+from data.database import ManifestList, ManifestListManifest, db_transaction
+
+
+logger = logging.getLogger(__name__)
+
+
+def _digest(manifestjson):
+  return hashlib.sha256(json.dumps(manifestjson, sort_keys=True)).hexdigest()
+
+
+def get_manifest_list(digest):
+  return ManifestList.select().where(ManifestList.digest == digest).get()
+
+
+def get_or_create_manifest_list(manifest_list_json, media_type_name, schema_version):
+  digest = _digest(manifest_list_json)
+  media_type_id = ManifestList.media_type.get_id(media_type_name)
+
+  try:
+    return get_manifest_list(digest)
+  except ManifestList.DoesNotExist:
+    with db_transaction():
+      manifestlist = ManifestList.create(digest=digest, manifest_list_json=manifest_list_json,
+                                         schema_version=schema_version, media_type=media_type_id)
+  return manifestlist
+
+
+def create_manifestlistmanifest(manifestlist, manifest_ids, manifest_list_json):
+  """ From a manifestlist, manifests, and the manifest list blob,
+  create if doesn't exist the manfiestlistmanifest for each manifest """
+  for pos in xrange(len(manifest_ids)):
+    manifest_id = manifest_ids[pos]
+    manifest_json = manifest_list_json[pos]
+    get_or_create_manifestlistmanifest(manifest=manifest_id,
+                                       manifestlist=manifestlist,
+                                       media_type_name=manifest_json['mediaType'])
+
+
+def get_or_create_manifestlistmanifest(manifest, manifestlist, media_type_name):
+  media_type_id = ManifestListManifest.media_type.get_id(media_type_name)
+  try:
+    ml = (ManifestListManifest
+          .select()
+          .where(ManifestListManifest.manifest == manifest,
+                 ManifestListManifest.media_type == media_type_id,
+                 ManifestListManifest.manifest_list == manifestlist)).get()
+
+  except ManifestListManifest.DoesNotExist:
+    ml = ManifestListManifest.create(manifest_list=manifestlist, media_type=media_type_id,
+                                     manifest=manifest)
+    return ml