Merge "Revert " Removed unused scaling rpc handler code""
[osm/SO.git] / rwlaunchpad / plugins / rwlaunchpadtasklet / rift / tasklets / rwlaunchpad / image.py
1
2 #
3 # Copyright 2016 RIFT.IO Inc
4 #
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at
8 #
9 # http://www.apache.org/licenses/LICENSE-2.0
10 #
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
16 #
17
18 import asyncio
19 import itertools
20 import glanceclient
21
22 import gi
23 gi.require_version('RwcalYang', '1.0')
24
25 from rift.imagemgr import client
26
27
28 class ImageUploadError(Exception):
29 pass
30
31
32 class ImageUploader(object):
33 """ This class is responsible for uploading package images to cloud accounts """
34 def __init__(self, log, loop, dts):
35 """ Create an instance of ImageUploader
36
37 Arguments:
38 log - A logger
39 """
40 self._log = log
41 self._loop = loop
42 self._dts = dts
43
44 self._client = client.UploadJobClient(self._log, self._loop, self._dts)
45
46 def upload_image(self, image_name, image_checksum, image_hdl):
47 endpoint = "http://127.0.0.1:9292"
48 glance_client = glanceclient.Client('1', endpoint, token="asdf")
49
50 try:
51 for image in itertools.chain(
52 glance_client.images.list(is_public=False),
53 glance_client.images.list(is_public=True),
54 ):
55 if image.name == image_name and image_checksum == image_checksum:
56 self._log.debug("Found existing image in catalog, not re-uploading")
57 return
58
59 self._log.debug('Uploading image to catalog: {}'.format(image_name))
60
61 image = glance_client.images.create(name=image_name, data=image_hdl, is_public="False",
62 disk_format="qcow2", container_format="bare",
63 checksum=image_checksum)
64 self._log.debug('Image upload complete: %s', image)
65 except Exception as e:
66 raise ImageUploadError("Failed to upload image to catalog: %s" % str(e)) from e
67
68 def upload_image_to_cloud_accounts(self, image_name, image_checksum, cloud_accounts=None):
69 self._log.debug("uploading image %s to all cloud accounts", image_name)
70 upload_job = self._client.create_job_threadsafe(image_name, image_checksum, cloud_accounts)
71 try:
72 upload_job.wait_until_complete_threadsafe()
73 except client.UploadJobError as e:
74 raise ImageUploadError("Failed to upload image " + image_name + " to cloud accounts") from e
75