Change label creation function to return the label created

This commit is contained in:
Joseph Schorr 2018-08-21 14:27:10 -04:00
parent 23ff49f0c1
commit 5a997eb4e3
2 changed files with 15 additions and 3 deletions

View file

@ -5,7 +5,7 @@ from collections import defaultdict
from data import database
from data import model
from data.registry_model.interface import RegistryDataInterface
from data.registry_model.datatypes import Tag, RepositoryReference, Manifest, LegacyImage
from data.registry_model.datatypes import Tag, RepositoryReference, Manifest, LegacyImage, Label
class PreOCIModel(RegistryDataInterface):
@ -60,9 +60,11 @@ class PreOCIModel(RegistryDataInterface):
try:
tag_manifest = database.TagManifest.get(id=manifest._db_id)
except database.TagManifest.DoesNotExist:
return
return None
model.label.create_manifest_label(tag_manifest, key, value, source_type_name, media_type_name)
label = model.label.create_manifest_label(tag_manifest, key, value, source_type_name,
media_type_name)
return Label.for_label(label)
def get_legacy_images(self, repository_ref):
"""