def __init__(self):

self.db_api = get_db_api()

def db_sync(self, repo_path=None):

self.db_api.db_sync(CONF, repo_path=repo_path)

def db_upgrade(self, version=None, repo_path=None):

self.db_api.db_upgrade(CONF, version, repo_path=repo_path)

def db_downgrade(self, version, repo_path=None):

raise SystemExit(_("Database downgrade is no longer supported."))

def execute(self):

exec_method = getattr(self, CONF.action.name)

args = inspect.getargspec(exec_method)

args.args.remove('self')

kwargs = {}

for arg in args.args:

kwargs[arg] = getattr(CONF.action, arg)

exec_method(**kwargs)

def datastore_update(self, datastore_name, default_version):

try:

datastore_models.update_datastore(datastore_name,

default_version)

print("Datastore '%s' updated." % datastore_name)

except exception.DatastoreVersionNotFound as e:

print(e)

def datastore_version_update(self, datastore, version_name, manager,

image_id, packages, active):

try:

datastore_models.update_datastore_version(datastore,

version_name,

manager,

image_id,

packages, active)

print("Datastore version '%s' updated." % version_name)

except exception.DatastoreNotFound as e:

print(e)

def db_recreate(self, repo_path):

"""Drops the database and recreates it."""

self.db_api.drop_db(CONF)

self.db_sync(repo_path)

def db_load_datastore_config_parameters(self,

datastore,

datastore_version_name,

config_file_location):

print("Loading config parameters for datastore (%s) version (%s)"

% (datastore, datastore_version_name))

config_models.load_datastore_configuration_parameters(

datastore, datastore_version_name, config_file_location)

def db_remove_datastore_config_parameters(self, datastore,

datastore_version_name):

print("Removing config parameters for datastore (%s) version (%s)"

% (datastore, datastore_version_name))

config_models.remove_datastore_configuration_parameters(

datastore, datastore_version_name)

def datastore_version_flavor_add(self, datastore_name,

datastore_version_name, flavor_ids):

"""Adds flavors for a given datastore version id."""

try:

dsmetadata = datastore_models.DatastoreVersionMetadata

dsmetadata.add_datastore_version_flavor_association(

datastore_name, datastore_version_name, flavor_ids.split(","))

print("Added flavors '%s' to the '%s' '%s'."

% (flavor_ids, datastore_name, datastore_version_name))

except exception.DatastoreVersionNotFound as e:

print(e)

def datastore_version_flavor_delete(self, datastore_name,

datastore_version_name, flavor_id):

"""Deletes a flavor's association with a given datastore."""

try:

dsmetadata = datastore_models.DatastoreVersionMetadata

dsmetadata.delete_datastore_version_flavor_association(

datastore_name, datastore_version_name, flavor_id)

print("Deleted flavor '%s' from '%s' '%s'."

% (flavor_id, datastore_name, datastore_version_name))

except exception.DatastoreVersionNotFound as e:

print(e)

def datastore_version_volume_type_add(self, datastore_name,

datastore_version_name,

volume_type_ids):

"""Adds volume type assiciation for a given datastore version id."""

try:

dsmetadata = datastore_models.DatastoreVersionMetadata

dsmetadata.add_datastore_version_volume_type_association(

datastore_name, datastore_version_name,

volume_type_ids.split(","))

print("Added volume type '%s' to the '%s' '%s'."

% (volume_type_ids, datastore_name, datastore_version_name))

except exception.DatastoreVersionNotFound as e:

print(e)

def datastore_version_volume_type_delete(self, datastore_name,

datastore_version_name,

volume_type_id):

"""Deletes a volume type association with a given datastore."""

try:

dsmetadata = datastore_models.DatastoreVersionMetadata

dsmetadata.delete_datastore_version_volume_type_association(

datastore_name, datastore_version_name, volume_type_id)

print("Deleted volume type '%s' from '%s' '%s'."

% (volume_type_id, datastore_name, datastore_version_name))

except exception.DatastoreVersionNotFound as e:

print(e)

def datastore_version_volume_type_list(self, datastore_name,

datastore_version_name):

"""Lists volume type association with a given datastore."""

try:

dsmetadata = datastore_models.DatastoreVersionMetadata

vtlist = dsmetadata.list_datastore_volume_type_associations(

datastore_name, datastore_version_name)

if vtlist.count() > 0:

for volume_type in vtlist:

print("Datastore: %s, Version: %s, Volume Type: %s" %

(datastore_name, datastore_version_name,

volume_type.value))

else:

print("No Volume Type Associations found for Datastore: %s, "

"Version: %s." %

(datastore_name, datastore_version_name))

except exception.DatastoreVersionNotFound as e:

print(e)

def params_of(self, command_name):

if Commands.has(command_name):

def actions(subparser):

repo_path_help = 'SQLAlchemy Migrate repository path.'

parser = subparser.add_parser(

'db_sync', description='Populate the database structure')

parser.add_argument('--repo_path', help=repo_path_help)

parser = subparser.add_parser(

'db_upgrade', description='Upgrade the database to the '

'specified version.')

parser.add_argument(

'--version', help='Target version. Defaults to the '

'latest version.')

parser.add_argument('--repo_path', help=repo_path_help)

parser = subparser.add_parser(

'datastore_update', description='Add or update a datastore. '

'If the datastore already exists, the default version will be '

'updated.')

parser.add_argument(

'datastore_name', help='Name of the datastore.')

parser.add_argument(

'default_version', help='Name or ID of an existing datastore '

'version to set as the default. When adding a new datastore, use '

'an empty string.')

parser = subparser.add_parser(

'datastore_version_update', description='Add or update a '

'datastore version. If the datastore version already exists, all '

'values except the datastore name and version will be updated.')

parser.add_argument('datastore', help='Name of the datastore.')

parser.add_argument(

'version_name', help='Name of the datastore version.')

parser.add_argument(

'manager', help='Name of the manager that will administer the '

'datastore version.')

parser.add_argument(

'image_id', help='ID of the image used to create an instance of '

'the datastore version.')

parser.add_argument(

'packages', help='Packages required by the datastore version that '

'are installed on the guest image.')

parser.add_argument(

'active', type=int,

help='Whether the datastore version is active or not. '

'Accepted values are 0 and 1.')

parser = subparser.add_parser(

'db_recreate', description='Drop the database and recreate it.')

parser.add_argument('--repo_path', help=repo_path_help)

parser = subparser.add_parser(

'db_load_datastore_config_parameters',

description='Loads configuration group parameter validation rules '

'for a datastore version into the database.')

parser.add_argument(

'datastore',

help='Name of the datastore.')

parser.add_argument(

'datastore_version_name',

help='Name of the datastore version.')

parser.add_argument(

'config_file_location',

help='Fully qualified file path to the configuration group '

'parameter validation rules.')

parser = subparser.add_parser(

'db_remove_datastore_config_parameters',

description='Remove configuration group parameter validation '

'rules for a datastore version from the database.')

parser.add_argument(

'datastore',

help='Name of the datastore.')

parser.add_argument(

'datastore_version_name',

help='Name of the datastore version.')

parser = subparser.add_parser(

'datastore_version_flavor_add', help='Adds flavor association to '

'a given datastore and datastore version.')

parser.add_argument('datastore_name', help='Name of the datastore.')

parser.add_argument('datastore_version_name', help='Name of the '

'datastore version.')

parser.add_argument('flavor_ids', help='Comma separated list of '

'flavor ids.')

parser = subparser.add_parser(

'datastore_version_flavor_delete', help='Deletes a flavor '

'associated with a given datastore and datastore version.')

parser.add_argument('datastore_name', help='Name of the datastore.')

parser.add_argument('datastore_version_name', help='Name of the '

'datastore version.')

parser.add_argument('flavor_id', help='The flavor to be deleted for '

'a given datastore and datastore version.')

parser = subparser.add_parser(

'datastore_version_volume_type_add', help='Adds volume_type '

'association to a given datastore and datastore version.')

parser.add_argument('datastore_name', help='Name of the datastore.')

parser.add_argument('datastore_version_name', help='Name of the '

'datastore version.')

parser.add_argument('volume_type_ids', help='Comma separated list of '

'volume_type ids.')

parser = subparser.add_parser(

'datastore_version_volume_type_delete',

help='Deletes a volume_type '

'associated with a given datastore and datastore version.')

parser.add_argument('datastore_name', help='Name of the datastore.')

parser.add_argument('datastore_version_name', help='Name of the '

'datastore version.')

parser.add_argument('volume_type_id', help='The volume_type to be '

'deleted for a given datastore and datastore '

'version.')

parser = subparser.add_parser(

'datastore_version_volume_type_list',

help='Lists the volume_types '

'associated with a given datastore and datastore version.')

parser.add_argument('datastore_name', help='Name of the datastore.')

parser.add_argument('datastore_version_name', help='Name of the '

'datastore version.')

cfg.custom_parser('action', actions)

cfg.parse_args(sys.argv)

try:

logging.setup(CONF, None)

Commands().execute()

sys.exit(0)

except TypeError as e:

print(_("Possible wrong number of arguments supplied %s.") % e)

sys.exit(2)

except Exception:

print(_("Command failed, please check log for more info."))