quay/data/oci_model/manifest_list.py

import logging
import hashlib
import json

from data.database import ManifestList, ManifestListManifest, db_transaction


logger = logging.getLogger(__name__)


def _ensure_sha256_header(digest):
  if digest.startswith('sha256:'):
    return digest
  return 'sha256:' + digest


def _digest(manifestjson):
  return _ensure_sha256_header(hashlib.sha256(json.dumps(manifestjson, sort_keys=True)).hexdigest())


def get_manifest_list(digest):
  return ManifestList.select().where(ManifestList.digest == _ensure_sha256_header(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