def test00_file(self):
HEADING()
try:
self.key_store = cm_keys_yaml("wrong file")
except:
pass
assert True
def test00_file(self):
HEADING()
try:
self.key_store = cm_keys_yaml("wrong file")
except:
pass
assert True
def _load_keys(self):
try:
filename = "$HOME/.futuregrid/cloudmesh.yaml"
if self.echo:
log.info("Reading keys information from -> {0}"
.format(filename))
self.keys = cm_keys_yaml(filename)
self.keys_loaded = True
except:
print "ERROR: could not find the keys in %s" % filename
def _load_keys_from_yaml(self):
try:
filename = config_file("/cloudmesh.yaml")
if self.echo:
log.info(
"Reading keys information from -> {0}".format(filename))
self.keys = cm_keys_yaml(filename)
self.keys_loaded = True
except:
print("ERROR: could not find the keys in %s" % filename)
def _load_keys_from_yaml(self):
try:
filename = config_file("/cloudmesh.yaml")
if self.echo:
log.info("Reading keys information from -> {0}"
.format(filename))
self.keys = cm_keys_yaml(filename)
self.keys_loaded = True
except:
print("ERROR: could not find the keys in %s" % filename)
def _load_keys(self):
try:
filename = "$HOME/.futuregrid/cloudmesh.yaml"
if self.echo:
log.info(
"Reading keys information from -> {0}".format(filename))
self.keys = cm_keys_yaml(filename)
self.keys_loaded = True
except:
print "ERROR: could not find the keys in %s" % filename
def __init__(self):
self.key_store = cm_keys_yaml(self.filename)
self.username = os.environ['USER']
self.mykey = "{0}-key".format(self.username)
def do_key(self, args, arguments):
"""
::
Usage:
key -h|--help
key list [--source=SOURCE] [--dir=DIR] [--format=FORMAT]
key add [--keyname=KEYNAME] FILENAME
key default [KEYNAME]
key delete KEYNAME
Manages the keys
Arguments:
SOURCE mongo, yaml, ssh
KEYNAME The name of a key
FORMAT The format of the output (table, json, yaml)
FILENAME The filename with full path in which the key
is located
Options:
--dir=DIR the directory with keys [default: ~/.ssh]
--format=FORMAT the format of the output [default: table]
--source=SOURCE the source for the keys [default: mongo]
--keyname=KEYNAME the name of the keys
Description:
key list --source=ssh [--dir=DIR] [--format=FORMAT]
lists all keys in the directory. If the directory is not
specified the default will be ~/.ssh
key list --source=yaml [--dir=DIR] [--format=FORMAT]
lists all keys in cloudmesh.yaml file in the specified directory.
dir is by default ~/.cloudmesh
key list [--format=FORMAT]
list the keys in mongo
key add [--keyname=keyname] FILENAME
adds the key specifid by the filename to mongodb
key list
Prints list of keys. NAME of the key can be specified
key default [NAME]
Used to set a key from the key-list as the default key if NAME
is given. Otherwise print the current default key
key delete NAME
deletes a key. In yaml mode it can delete only key that
are not saved in mongo
"""
# print arguments
def _find_keys(directory):
return [
file for file in listdir(expanduser(directory))
if file.lower().endswith(".pub")
]
#
# DIR (OK)
#
directory = path_expand(arguments["--dir"])
source = arguments["--source"]
if source not in ["ssh", "yaml", "mongo"]:
print("ERROR: source is not defined")
return
#
# PRINT DICT (OK)
#
def _print_dict(d, header=None):
print_format = arguments['--format']
if print_format == "json":
return json.dumps(d, indent=4)
elif print_format == "yaml":
return yaml.dump(d, default_flow_style=False)
else:
return two_column_table(keys, header)
#
# PRINT SYSTEM (OK)
#
if arguments["list"] and source == "ssh":
files = _find_keys(directory)
keys = {}
for key in files:
keys[key] = directory + "/" + key
print(_print_dict(keys, header=["Key", "Location"]))
return
#
# PRINT YAML (OK)
#
if arguments["list"] and source == "yaml":
key_store = cm_keys_yaml()
keynames = key_store.names()
keys = {}
for key in keynames:
keys[key] = get_fingerprint(key_store[key])
print(_print_dict(keys, header=["Key", "Fingerprint"]))
return
#
# FROM HERE ON BROKEN
#
if arguments["list"] and source == "mongo":
username = cm_config().username()
key_store = cm_keys_mongo(username)
keynames = key_store.names()
keys = {}
for key in keynames:
keys[key] = key_store[key]
if keytype(keys[key]) == 'string':
keys[key] = get_fingerprint(keys[key])
print(_print_dict(keys, header=["Key", "Fingerprint"]))
return
# self._load_keys_from_yaml()
# self._load_mongo()
# self._load_keys_mongo()
if arguments["add"] and arguments["FILENAME"]:
username = cm_config().username()
key_store = cm_keys_mongo(username)
filename = arguments["FILENAME"]
# print filename
# file existence check moved to the util function
# if file does not exist it will return None
keystring = get_key_from_file(filename)
if keystring:
keystring = keystring.strip()
keysegments = key_parse(keystring)
# key valid?
if key_validate("string", keystring):
if arguments["--keyname"]:
keyname = arguments["--keyname"]
else:
# get keyname from the comment field of key
keyname = keysegments[2]
# in case a comment field is really missing
# use a timestamp for uniqueness
if keyname is None:
tstamp = datetime.now().strftime("%Y%m%d%H%M%S")
keyname = tstamp
# sanitize the key name for mongo use
else:
keyname = re.sub(r'@|-|\.', '_', keysegments[2])
key_store[keyname] = keystring
else:
print("ERROR: INVALID key. Please verify!")
else:
print("ERROR: INVALID filename provided!")
return
"""
def func():
if arguments["KEY"]:
key_store.__setitem__(
arguments["NAME"], arguments["KEY"])
else:
files = _find_keys(directory)
result = menu_return_num(
title="Select a public key", menu_list=files, tries=3)
if result == 'q':
return
else:
key_store.__setitem__(arguments["NAME"],
"{0}/{1}".format(directory,files[result]))
if arguments["NAME"] in key_store.names():
if yn_choice("key {0} exists, update?"
.format(arguments["NAME"]), default='n'):
print "Updating key {0} ...".format(arguments["NAME"])
func()
else:
return
else:
print "adding key {0} ...".format(arguments["NAME"])
func()
"""
if arguments["default"]:
username = cm_config().username()
key_store = cm_keys_mongo(username)
# print key_store.names()
# no name provided, will get current default key
if not arguments["KEYNAME"]:
defaultkey = key_store.default()
print("Current default key is: {0}".format(defaultkey))
# name provided, check if it exists in the db
elif arguments["KEYNAME"] in key_store.names():
key_store.setdefault(arguments["KEYNAME"])
# Update mongo db defaults with new default key
print('The default key was successfully set to: ',
arguments['KEYNAME'])
else:
print("ERROR: Specified key is not registered.")
return
if arguments["delete"]:
username = cm_config().username()
key_store = cm_keys_mongo(username)
print("Attempting to delete key: {0}".format(arguments["KEYNAME"]))
if self.use_yaml:
print("WARNING: This will only remove the keys that"
"have not been written to the databse already when "
"'keys save' is"
"called. If your key is already in the database, "
"you should use mongo mode\n")
key_store.delete(arguments["KEYNAME"])
return
# deprecating...
"""
def do_key(self, args, arguments):
"""
Usage:
key -h|--help
key list [--source=SOURCE] [--dir=DIR] [--format=FORMAT]
key add [--keyname=KEYNAME] FILENAME
key default [KEYNAME]
key delete KEYNAME
Manages the keys
Arguments:
SOURCE mongo, yaml, ssh
KEYNAME The name of a key
FORMAT The format of the output (table, json, yaml)
FILENAME The filename with full path in which the key is located
Options:
--dir=DIR the directory with keys [default: ~/.ssh]
--format=FORMAT the format of the output [default: table]
--source=SOURCE the source for the keys [default: mongo]
--keyname=KEYNAME the name of the keys
Description:
key list --source=ssh [--dir=DIR] [--format=FORMAT]
lists all keys in the directory. If the directory is not
specified the defualt will be ~/.ssh
key list --source=yaml [--dir=DIR] [--format=FORMAT]
lists all keys in cloudmesh.yaml file in the specified directory.
dir is by default ~/.cloudmesh
key list [--format=FORMAT]
list the keys in mongo
key add [--keyname=keyname] FILENAME
adds the key specifid by the filename to mongodb
key list
Prints list of keys. NAME of the key can be specified
key default [NAME]
Used to set a key from the key-list as the default key if NAME
is given. Otherwise print the current default key
key delete NAME
deletes a key. In yaml mode it can delete only key that
are not saved in mongo
"""
# print arguments
def _find_keys(directory):
return [file for file in listdir(expanduser(directory))
if file.lower().endswith(".pub")]
#
# DIR (OK)
#
directory = path_expand(arguments["--dir"])
source = arguments["--source"]
if source not in ["ssh", "yaml", "mongo"]:
print("ERROR: source is not defined")
return
#
# PRINT DICT (OK)
#
def _print_dict(d, header=None):
print_format = arguments['--format']
if print_format == "json":
return json.dumps(d, indent=4)
elif print_format == "yaml":
return yaml.dump(d, default_flow_style=False)
else:
return two_column_table(keys, header)
#
# PRINT SYSTEM (OK)
#
if arguments["list"] and source == "ssh":
files = _find_keys(directory)
keys = {}
for key in files:
keys[key] = directory + "/" + key
print(_print_dict(keys, header=["Key", "Location"]))
return
#
# PRINT YAML (OK)
#
if arguments["list"] and source == "yaml":
key_store = cm_keys_yaml()
keynames = key_store.names()
keys = {}
for key in keynames:
keys[key] = get_fingerprint(key_store[key])
print(_print_dict(keys, header=["Key", "Fingerprint"]))
return
#
# FROM HERE ON BROKEN
#
if arguments["list"] and source == "mongo":
username = cm_config().username()
key_store = cm_keys_mongo(username)
keynames = key_store.names()
keys = {}
for key in keynames:
keys[key] = key_store[key]
if keytype(keys[key]) == 'string':
keys[key] = get_fingerprint(keys[key])
print(_print_dict(keys, header=["Key", "Fingerprint"]))
return
# self._load_keys_from_yaml()
# self._load_mongo()
# self._load_keys_mongo()
if arguments["add"] and arguments["FILENAME"]:
username = cm_config().username()
key_store = cm_keys_mongo(username)
filename = arguments["FILENAME"]
# print filename
# file existence check moved to the util function
# if file does not exist it will return None
keystring = get_key_from_file(filename)
if keystring:
keystring = keystring.strip()
keysegments = key_parse(keystring)
# key valid?
if key_validate("string", keystring):
if arguments["--keyname"]:
keyname = arguments["--keyname"]
else:
# get keyname from the comment field of key
keyname = keysegments[2]
# in case a comment field is really missing
# use a timestamp for uniqueness
if keyname is None:
tstamp = datetime.now().strftime("%Y%m%d%H%M%S")
keyname = tstamp
# sanitize the key name for mongo use
else:
keyname = re.sub(r'@|-|\.', '_', keysegments[2])
key_store[keyname] = keystring
else:
print("ERROR: INVALID key. Please verify!")
else:
print("ERROR: INVALID filename provided!")
return
"""
def func():
if arguments["KEY"]:
key_store.__setitem__(
arguments["NAME"], arguments["KEY"])
else:
files = _find_keys(directory)
result = menu_return_num(
title="Select a public key", menu_list=files, tries=3)
if result == 'q':
return
else:
key_store.__setitem__(arguments["NAME"],
"{0}/{1}".format(directory,files[result]))
if arguments["NAME"] in key_store.names():
if yn_choice("key {0} exists, update?"
.format(arguments["NAME"]), default='n'):
print "Updating key {0} ...".format(arguments["NAME"])
func()
else:
return
else:
print "adding key {0} ...".format(arguments["NAME"])
func()
"""
if arguments["default"]:
username = cm_config().username()
key_store = cm_keys_mongo(username)
# print key_store.names()
# no name provided, will get current default key
if not arguments["KEYNAME"]:
defaultkey = key_store.default()
print("Current default key is: {0}".format(defaultkey))
# name provided, check if it exists in the db
elif arguments["KEYNAME"] in key_store.names():
key_store.setdefault(arguments["KEYNAME"])
# Update mongo db defaults with new default key
print('The default key was successfully set to: ', arguments['KEYNAME'])
else:
print("ERROR: Specified key is not registered.")
return
if arguments["delete"]:
username = cm_config().username()
key_store = cm_keys_mongo(username)
print("Attempting to delete key: {0}".format(arguments["KEYNAME"]))
if self.use_yaml:
print("WARNING: This will only remove the keys that"
"have not been written to the databse already when "
"'keys save' is"
"called. If your key is already in the database, "
"you should use mongo mode\n")
key_store.delete(arguments["KEYNAME"])
return
# deprecating...
"""
def __init__(self):
self.key_store = cm_keys_yaml(self.filename)
self.username = os.environ['USER']
self.mykey = "{0}-key".format(self.username)