data.oci_model: sloppily rewrite digest format

We expect digests to be in the form 'sha256:digest'
This commit is contained in:
Jimmy Zelinskie 2017-03-23 12:37:32 -04:00
parent 2c8930c912
commit 9c0cbbf57c
5 changed files with 41 additions and 10 deletions

View file

@ -11,12 +11,18 @@ from data.oci_model import tag as tag_model
logger = logging.getLogger(__name__)
def _ensure_sha256_header(digest):
if digest.startswith('sha256:'):
return digest
return 'sha256:' + digest
def _digest(manifestjson):
return hashlib.sha256(json.dumps(manifestjson, sort_keys=True)).hexdigest()
return _ensure_sha256_header(hashlib.sha256(json.dumps(manifestjson, sort_keys=True)).hexdigest())
def get_manifest_query(digest, media_type):
return Manifest.select().where(Manifest.digest == digest,
return Manifest.select().where(Manifest.digest == _ensure_sha256_header(digest),
Manifest.media_type == Manifest.media_type.get_id(media_type))