From f7f58d88fc1ad7c16208e7e0dad598aab88ccbee Mon Sep 17 00:00:00 2001 From: Michael Basnight Date: Thu, 24 May 2012 12:23:33 -0500 Subject: [PATCH] Adding image_update call to reddwarf-manage --- bin/reddwarf-manage | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/bin/reddwarf-manage b/bin/reddwarf-manage index 3d52f8ac2a..4a53e5bb3a 100755 --- a/bin/reddwarf-manage +++ b/bin/reddwarf-manage @@ -37,6 +37,7 @@ from reddwarf import version from reddwarf.common import config from reddwarf.common import utils from reddwarf.db import db_api +from reddwarf.instance import models as instance_models def create_options(parser): @@ -71,7 +72,18 @@ class Commands(object): if self.has(command_name): return getattr(self, command_name)(*args) - _commands = ['db_sync', 'db_upgrade', 'db_downgrade'] + def image_update(self, service_name, image_id): + db_api.configure_db(self.conf) + image = db_api.find_by(instance_models.ServiceImage, service_name=service_name) + if image is None: + # Create a new one + image = instance_models.ServiceImage() + image.id = utils.generate_uuid() + image.service_name = service_name + image.image_id = image_id + db_api.save(image) + + _commands = ['db_sync', 'db_upgrade', 'db_downgrade', 'image_update'] @classmethod def has(cls, command_name):