Tony Mack | bf39d9f | 2014-05-06 21:42:36 -0400 | [diff] [blame] | 1 | import os |
| 2 | import base64 |
| 3 | from collections import defaultdict |
| 4 | from django.db.models import F, Q |
| 5 | from planetstack.config import Config |
| 6 | from observer.openstacksyncstep import OpenStackSyncStep |
Tony Mack | 91463d9 | 2014-06-19 20:42:03 -0400 | [diff] [blame^] | 7 | from core.models import Deployment |
| 8 | from core.models import Image, ImageDeployments |
Scott Baker | e3969b2 | 2014-06-02 22:59:08 -0700 | [diff] [blame] | 9 | from util.logger import Logger, logging |
| 10 |
|
| 11 | logger = Logger(level=logging.INFO) |
Tony Mack | bf39d9f | 2014-05-06 21:42:36 -0400 | [diff] [blame] | 12 | |
| 13 | class SyncImageDeployments(OpenStackSyncStep): |
| 14 | provides=[ImageDeployments] |
| 15 | requested_interval=0 |
| 16 | |
| 17 | def fetch_pending(self): |
Scott Baker | b3a7bca | 2014-06-11 11:35:40 -0700 | [diff] [blame] | 18 | # smbaker: commented out automatic creation of ImageDeployments as |
| 19 | # as they will now be configured in GUI. Not sure if this is |
| 20 | # sufficient. |
Scott Baker | e3969b2 | 2014-06-02 22:59:08 -0700 | [diff] [blame] | 21 | |
Scott Baker | b3a7bca | 2014-06-11 11:35:40 -0700 | [diff] [blame] | 22 | # # ensure images are available across all deployments |
| 23 | # image_deployments = ImageDeployments.objects.all() |
| 24 | # image_deploy_lookup = defaultdict(list) |
| 25 | # for image_deployment in image_deployments: |
| 26 | # image_deploy_lookup[image_deployment.image].append(image_deployment.deployment) |
| 27 | # |
| 28 | # all_deployments = Deployment.objects.all() |
| 29 | # for image in Image.objects.all(): |
| 30 | # expected_deployments = all_deployments |
| 31 | # for expected_deployment in expected_deployments: |
| 32 | # if image not in image_deploy_lookup or \ |
| 33 | # expected_deployment not in image_deploy_lookup[image]: |
| 34 | # id = ImageDeployments(image=image, deployment=expected_deployment) |
| 35 | # id.save() |
Scott Baker | e3969b2 | 2014-06-02 22:59:08 -0700 | [diff] [blame] | 36 | |
Tony Mack | bf39d9f | 2014-05-06 21:42:36 -0400 | [diff] [blame] | 37 | # now we return all images that need to be enacted |
Scott Baker | e3969b2 | 2014-06-02 22:59:08 -0700 | [diff] [blame] | 38 | return ImageDeployments.objects.filter(Q(enacted__lt=F('updated')) | Q(enacted=None)) |
| 39 | |
Tony Mack | bf39d9f | 2014-05-06 21:42:36 -0400 | [diff] [blame] | 40 | def sync_record(self, image_deployment): |
Scott Baker | e3969b2 | 2014-06-02 22:59:08 -0700 | [diff] [blame] | 41 | logger.info("Working on image %s on deployment %s" % (image_deployment.image.name, image_deployment.deployment.name)) |
Tony Mack | bf39d9f | 2014-05-06 21:42:36 -0400 | [diff] [blame] | 42 | driver = self.driver.admin_driver(deployment=image_deployment.deployment.name) |
Tony Mack | 8b85d9a | 2014-05-06 23:43:14 -0400 | [diff] [blame] | 43 | images = driver.shell.glance.get_images() |
| 44 | glance_image = None |
| 45 | for image in images: |
| 46 | if image['name'] == image_deployment.image.name: |
| 47 | glance_image = image |
| 48 | break |
| 49 | if glance_image: |
Scott Baker | e3969b2 | 2014-06-02 22:59:08 -0700 | [diff] [blame] | 50 | logger.info("Found image %s on deployment %s" % (image_deployment.image.name, image_deployment.deployment.name)) |
Tony Mack | 8b85d9a | 2014-05-06 23:43:14 -0400 | [diff] [blame] | 51 | image_deployment.glance_image_id = glance_image['id'] |
| 52 | elif image_deployment.image.path: |
Scott Baker | 3949bc6 | 2014-05-12 10:42:16 -0700 | [diff] [blame] | 53 | image = { |
| 54 | 'name': image_deployment.image.name, |
| 55 | 'is_public': True, |
Scott Baker | e3969b2 | 2014-06-02 22:59:08 -0700 | [diff] [blame] | 56 | 'disk_format': 'raw', |
Scott Baker | 3949bc6 | 2014-05-12 10:42:16 -0700 | [diff] [blame] | 57 | 'container_format': 'bare', |
Scott Baker | e3969b2 | 2014-06-02 22:59:08 -0700 | [diff] [blame] | 58 | 'file': image_deployment.image.path, |
| 59 | } |
| 60 | |
| 61 | logger.info("Creating image %s on deployment %s" % (image_deployment.image.name, image_deployment.deployment.name)) |
| 62 | |
Tony Mack | 8b85d9a | 2014-05-06 23:43:14 -0400 | [diff] [blame] | 63 | glance_image = driver.shell.glanceclient.images.create(name=image_deployment.image.name, |
| 64 | is_public=True, |
| 65 | disk_format='raw', |
| 66 | container_format='bare') |
| 67 | glance_image.update(data=open(image_deployment.image.path, 'rb')) |
Scott Baker | e3969b2 | 2014-06-02 22:59:08 -0700 | [diff] [blame] | 68 | |
| 69 | # While the images returned by driver.shell.glance.get_images() |
| 70 | # are dicts, the images returned by driver.shell.glanceclient.images.create |
| 71 | # are not dicts. We have to use getattr() instead of [] operator. |
| 72 | if not glance_image or not getattr(glance_image,"id",None): |
Scott Baker | 3949bc6 | 2014-05-12 10:42:16 -0700 | [diff] [blame] | 73 | raise Exception, "Add image failed at deployment %s" % image_deployment.deployment.name |
Scott Baker | e3969b2 | 2014-06-02 22:59:08 -0700 | [diff] [blame] | 74 | image_deployment.glance_image_id = getattr(glance_image, "id") |
Tony Mack | bf39d9f | 2014-05-06 21:42:36 -0400 | [diff] [blame] | 75 | image_deployment.save() |