856bb454ba
Fix typo in Zun Change-Id: I584e9052baed5bcbd07ae417c7b7832c9cd3c1fd Signed-off-by: Andy Yan <yanchao3@lenovo.com>
106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
# Copyright 2016 Intel.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import six
|
|
import sys
|
|
|
|
from oslo_log import log as logging
|
|
from oslo_utils import importutils
|
|
|
|
from zun.common import exception
|
|
from zun.common.i18n import _
|
|
from zun.common.i18n import _LE
|
|
from zun.common.i18n import _LI
|
|
from zun.common.utils import parse_image_name
|
|
import zun.conf
|
|
|
|
CONF = zun.conf.CONF
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
def load_image_driver(image_driver=None):
|
|
"""Load an image driver module.
|
|
|
|
Load the container image driver module specified by the image_driver
|
|
configuration option or, if supplied, the driver name supplied as an
|
|
argument.
|
|
:param image_driver: container image driver name to override config opt
|
|
:returns: a ContainerImageDriver instance
|
|
"""
|
|
if not image_driver:
|
|
LOG.error(_LE("Container image driver option required, "
|
|
"but not specified"))
|
|
sys.exit(1)
|
|
|
|
LOG.info(_LI("Loading container image driver '%s'"), image_driver)
|
|
try:
|
|
driver = importutils.import_object(
|
|
'zun.image.%s' % image_driver)
|
|
if not isinstance(driver, ContainerImageDriver):
|
|
raise Exception(_('Expected driver of type: %s') %
|
|
str(ContainerImageDriver))
|
|
|
|
return driver
|
|
except ImportError:
|
|
LOG.exception(_LE("Unable to load the container image driver"))
|
|
sys.exit(1)
|
|
|
|
|
|
def pull_image(context, repo, tag, image_pull_policy):
|
|
image_driver_list = CONF.image_driver_list
|
|
for driver in image_driver_list:
|
|
try:
|
|
image_driver = load_image_driver(driver)
|
|
image = image_driver.pull_image(context, repo,
|
|
tag, image_pull_policy)
|
|
if image:
|
|
break
|
|
except exception.ImageNotFound:
|
|
image = None
|
|
except Exception as e:
|
|
LOG.exception(_LE('Unknown exception occurred while loading '
|
|
'image: %s'), six.text_type(e))
|
|
raise exception.ZunException(six.text_type(e))
|
|
if not image:
|
|
raise exception.ImageNotFound("Image %s not found" % repo)
|
|
return image
|
|
|
|
|
|
def search_image(context, image_name, exact_match):
|
|
images = []
|
|
repo, tag = parse_image_name(image_name)
|
|
for driver in CONF.image_driver_list:
|
|
try:
|
|
image_driver = load_image_driver(driver)
|
|
imgs = image_driver.search_image(context, repo, tag,
|
|
exact_match=exact_match)
|
|
images.extend(imgs)
|
|
except Exception as e:
|
|
LOG.exception(_LE('Unknown exception occurred while searching '
|
|
'for image: %s'), six.text_type(e))
|
|
raise exception.ZunException(six.text_type(e))
|
|
return images
|
|
|
|
|
|
class ContainerImageDriver(object):
|
|
'''Base class for container image driver.'''
|
|
|
|
def pull_image(self, context, repo, tag):
|
|
"""Pull an image."""
|
|
raise NotImplementedError()
|
|
|
|
def search_image(self, context, repo, tag):
|
|
"""Search an image."""
|
|
raise NotImplementedError()
|