dbe6293022
This patch adds the functionality to support image_driver during container create/run. If user provides a valid image_driver, that is defined in configuration he/she will be allowed to create the container otherwise the request will fail. This patch also add image_driver field in db, so that user can inspect which image driver he/she used during the container create. Partially-Implements: BP allow-specify-image-driver Change-Id: Id2299a613ed414b2df67c103fe6cda0397ab791d
118 lines
3.9 KiB
Python
118 lines
3.9 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
|
|
import stevedore
|
|
|
|
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 = stevedore.driver.DriverManager(
|
|
"zun.image.driver",
|
|
image_driver,
|
|
invoke_on_load=True).driver
|
|
|
|
if not isinstance(driver, ContainerImageDriver):
|
|
raise Exception(_('Expected driver of type: %s') %
|
|
str(ContainerImageDriver))
|
|
|
|
return driver
|
|
except Exception:
|
|
LOG.exception(_LE("Unable to load the container image driver"))
|
|
sys.exit(1)
|
|
|
|
|
|
def pull_image(context, repo, tag, image_pull_policy, image_driver):
|
|
if image_driver:
|
|
image_driver_list = [image_driver.lower()]
|
|
else:
|
|
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:
|
|
image['driver'] = driver.split('.')[0]
|
|
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, image_driver, exact_match):
|
|
images = []
|
|
repo, tag = parse_image_name(image_name)
|
|
if image_driver:
|
|
image_driver_list = [image_driver.lower()]
|
|
else:
|
|
image_driver_list = CONF.image_driver_list
|
|
for driver in 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()
|