def main(argv=None):
'''
Parses command line
'''
parser = argparse.ArgumentParser()
parser.add_argument('-f', '--file', nargs=1, help="specify keys file")
command = parser.parse_args(argv)
if command.file:
keyfile.set_file(command.file[0])
else:
print("MyKeys key file: %s" % (keyfile.get_file(), ))
if not os.path.exists(keyfile.get_file()):
print("Warning: keyfile not found")
def test_key_file(self):
'''
Test get_file/set_file APIs
'''
keyfile.set_file("dummy")
expected = os.path.abspath("dummy")
self.assertEquals(expected, keyfile.get_file())
def test_command_line(self):
'''
Test updating keyfile via command line
'''
cmdline.main(["-f", "dummy"])
expected = os.path.abspath("dummy")
self.assertEquals(expected, keyfile.get_file())
def parse_keys():
'''
Parse keys file
'''
keys = configparser.ConfigParser()
keys.read(keyfile.get_file())
keys['DEFAULT']['key'] = ""
keys['DEFAULT']['secret'] = ""
attributes = dict()
for section in keys.sections():
attributes[section] = KeySecretPair(keys[section]['key'], keys[section]['secret'])
return type('Keys', (), attributes)
def setUp(self):
self.save_keyfile = keyfile.get_file()
