def test_upgrade_2dash_master_in_resource_create_with_complications(self):
self.assertEqual([
"-f", "path/to/file", "resource", "-V", "create", "master",
"second"
],
upgrade_args([
"-f", "path/to/file", "resource", "-V", "create",
"--master", "second"
]))
def test_upgrade_2dash_master_in_resource_create_with_complications(self):
self.assertEqual(
[
"-f", "path/to/file", "resource", "-V", "create", "master",
"second"
],
upgrade_args([
"-f", "path/to/file", "resource", "-V", "create", "--master",
"second"
])
)
def main(argv=None):
if completion.has_applicable_environment(os.environ):
print(completion.make_suggestions(
os.environ,
usage.generate_completion_tree_from_usage()
))
sys.exit()
argv = argv if argv else sys.argv[1:]
utils.subprocess_setup()
global filename, usefile
orig_argv = argv[:]
utils.pcs_options = {}
argv = parse_args.upgrade_args(argv)
# we want to support optional arguments for --wait, so if an argument
# is specified with --wait (ie. --wait=30) then we use them
waitsecs = None
new_argv = []
for arg in argv:
if arg.startswith("--wait="):
tempsecs = arg.replace("--wait=","")
if len(tempsecs) > 0:
waitsecs = tempsecs
arg = "--wait"
new_argv.append(arg)
argv = new_argv
try:
pcs_options, dummy_argv = getopt.gnu_getopt(
parse_args.filter_out_non_option_negative_numbers(argv),
parse_args.PCS_SHORT_OPTIONS,
parse_args.PCS_LONG_OPTIONS,
)
except getopt.GetoptError as err:
print(err)
usage.main()
sys.exit(1)
argv = parse_args.filter_out_options(argv)
for o, a in pcs_options:
if not o in utils.pcs_options:
if o == "--watchdog":
a = [a]
utils.pcs_options[o] = a
else:
# If any options are a list then they've been entered twice which isn't valid
if o != "--watchdog":
utils.err("%s can only be used once" % o)
else:
utils.pcs_options[o].append(a)
if o == "-h" or o == "--help":
if len(argv) == 0:
usage.main()
sys.exit()
else:
argv = [argv[0], "help" ] + argv[1:]
elif o == "-f":
usefile = True
filename = a
utils.usefile = usefile
utils.filename = filename
elif o == "--corosync_conf":
settings.corosync_conf_file = a
elif o == "--cluster_conf":
settings.cluster_conf_file = a
elif o == "--version":
print(settings.pcs_version)
sys.exit()
elif o == "--fullhelp":
usage.full_usage()
sys.exit()
elif o == "--wait":
utils.pcs_options[o] = waitsecs
elif o == "--request-timeout":
request_timeout_valid = False
try:
timeout = int(a)
if timeout > 0:
utils.pcs_options[o] = timeout
request_timeout_valid = True
except ValueError:
pass
if not request_timeout_valid:
utils.err(
(
"'{0}' is not a valid --request-timeout value, use "
"a positive integer"
).format(a)
)
if len(argv) == 0:
usage.main()
sys.exit(1)
# create a dummy logger
# we do not have a log file for cli (yet), but library requires a logger
logger = logging.getLogger("old_cli")
logger.propagate = 0
logger.handlers = []
command = argv.pop(0)
if (command == "-h" or command == "help"):
usage.main()
return
cmd_map = {
"resource": resource.resource_cmd,
"cluster": cluster.cluster_cmd,
"stonith": stonith.stonith_cmd,
"property": prop.property_cmd,
"constraint": constraint.constraint_cmd,
"acl": lambda argv: acl.acl_cmd(
utils.get_library_wrapper(),
argv,
utils.get_modificators()
),
"status": status.status_cmd,
"config": config.config_cmd,
"pcsd": pcsd.pcsd_cmd,
"node": lambda argv: node.node_cmd(
utils.get_library_wrapper(),
argv,
utils.get_modificators()
),
"quorum": lambda argv: quorum.quorum_cmd(
utils.get_library_wrapper(),
argv,
utils.get_modificators()
),
"qdevice": lambda argv: qdevice.qdevice_cmd(
utils.get_library_wrapper(),
argv,
utils.get_modificators()
),
"alert": lambda args: alert.alert_cmd(
utils.get_library_wrapper(),
args,
utils.get_modificators()
),
"booth": lambda argv: booth.booth_cmd(
utils.get_library_wrapper(),
argv,
utils.get_modificators()
),
}
if command not in cmd_map:
usage.main()
sys.exit(1)
# root can run everything directly, also help can be displayed,
# working on a local file also do not need to run under root
if (os.getuid() == 0) or (argv and argv[0] == "help") or usefile:
cmd_map[command](argv)
return
# specific commands need to be run under root account, pass them to pcsd
# don't forget to allow each command in pcsd.rb in "post /run_pcs do"
root_command_list = [
['cluster', 'auth', '...'],
['cluster', 'corosync', '...'],
['cluster', 'destroy', '...'],
['cluster', 'disable', '...'],
['cluster', 'enable', '...'],
['cluster', 'node', '...'],
['cluster', 'pcsd-status', '...'],
['cluster', 'setup', '...'],
['cluster', 'start', '...'],
['cluster', 'stop', '...'],
['cluster', 'sync', '...'],
# ['config', 'restore', '...'], # handled in config.config_restore
['pcsd', 'sync-certificates'],
['status', 'nodes', 'corosync-id'],
['status', 'nodes', 'pacemaker-id'],
['status', 'pcsd', '...'],
]
argv_cmd = argv[:]
argv_cmd.insert(0, command)
for root_cmd in root_command_list:
if (
(argv_cmd == root_cmd)
or
(
root_cmd[-1] == "..."
and
argv_cmd[:len(root_cmd)-1] == root_cmd[:-1]
)
):
# handle interactivity of 'pcs cluster auth'
if argv_cmd[0:2] == ["cluster", "auth"]:
if "-u" not in utils.pcs_options:
username = utils.get_terminal_input('Username: ')
orig_argv.extend(["-u", username])
if "-p" not in utils.pcs_options:
password = utils.get_terminal_password()
orig_argv.extend(["-p", password])
# call the local pcsd
err_msgs, exitcode, std_out, std_err = utils.call_local_pcsd(
orig_argv, True
)
if err_msgs:
for msg in err_msgs:
utils.err(msg, False)
sys.exit(1)
if std_out.strip():
print(std_out)
if std_err.strip():
sys.stderr.write(std_err)
sys.exit(exitcode)
return
cmd_map[command](argv)
def test_dont_upgrade_2dash_master_outside_of_resource_create(self):
self.assertEqual(
["first", "--master", "second"],
upgrade_args(["first", "--master", "second"])
)
def test_upgrade_2dash_master_in_resource_create(self):
self.assertEqual(
["resource", "create", "master", "second"],
upgrade_args(["resource", "create", "--master", "second"])
)
def test_upgrade_2dash_cloneopt_with_value(self):
self.assertEqual(
["first", "clone", "1", "second"],
upgrade_args(["first", "--cloneopt=1", "second"])
)
def test_upgrade_2dash_clone(self):
self.assertEqual(
["first", "clone", "second"],
upgrade_args(["first", "--clone", "second"])
)
def test_returns_the_same_args_when_no_older_versions_detected(self):
args = ["first", "second"]
self.assertEqual(args, upgrade_args(args))
def main(argv=None):
if completion.has_applicable_environment(os.environ):
print(
completion.make_suggestions(
os.environ, usage.generate_completion_tree_from_usage()))
sys.exit()
argv = argv if argv else sys.argv[1:]
utils.subprocess_setup()
global filename, usefile
orig_argv = argv[:]
utils.pcs_options = {}
argv = parse_args.upgrade_args(argv)
# we want to support optional arguments for --wait, so if an argument
# is specified with --wait (ie. --wait=30) then we use them
waitsecs = None
new_argv = []
for arg in argv:
if arg.startswith("--wait="):
tempsecs = arg.replace("--wait=", "")
if len(tempsecs) > 0:
waitsecs = tempsecs
arg = "--wait"
new_argv.append(arg)
argv = new_argv
try:
pcs_options, dummy_argv = getopt.gnu_getopt(
parse_args.filter_out_non_option_negative_numbers(argv),
parse_args.PCS_SHORT_OPTIONS,
parse_args.PCS_LONG_OPTIONS,
)
except getopt.GetoptError as err:
print(err)
usage.main()
sys.exit(1)
argv = parse_args.filter_out_options(argv)
full = False
for option, dummy_value in pcs_options:
if option == "--full":
full = True
break
for o, a in pcs_options:
if not o in utils.pcs_options:
if o in ["--watchdog", "--device"]:
a = [a]
utils.pcs_options[o] = a
else:
# If any options are a list then they've been entered twice which isn't valid
if o not in ["--watchdog", "--device"]:
utils.err("%s can only be used once" % o)
else:
utils.pcs_options[o].append(a)
if o == "-h" or o == "--help":
if len(argv) == 0:
usage.main()
sys.exit()
else:
argv = [argv[0], "help"] + argv[1:]
elif o == "-f":
usefile = True
filename = a
utils.usefile = usefile
utils.filename = filename
elif o == "--corosync_conf":
settings.corosync_conf_file = a
elif o == "--cluster_conf":
settings.cluster_conf_file = a
elif o == "--version":
print(settings.pcs_version)
if full:
print(" ".join(
sorted([
feat["id"]
for feat in capabilities.get_pcs_capabilities()
])))
sys.exit()
elif o == "--fullhelp":
usage.full_usage()
sys.exit()
elif o == "--wait":
utils.pcs_options[o] = waitsecs
elif o == "--request-timeout":
request_timeout_valid = False
try:
timeout = int(a)
if timeout > 0:
utils.pcs_options[o] = timeout
request_timeout_valid = True
except ValueError:
pass
if not request_timeout_valid:
utils.err(("'{0}' is not a valid --request-timeout value, use "
"a positive integer").format(a))
if len(argv) == 0:
usage.main()
sys.exit(1)
# create a dummy logger
# we do not have a log file for cli (yet), but library requires a logger
logger = logging.getLogger("old_cli")
logger.propagate = 0
logger.handlers = []
command = argv.pop(0)
if (command == "-h" or command == "help"):
usage.main()
return
cmd_map = {
"resource":
resource.resource_cmd,
"cluster":
cluster.cluster_cmd,
"stonith":
stonith.stonith_cmd,
"property":
prop.property_cmd,
"constraint":
constraint.constraint_cmd,
"acl":
lambda argv: acl.acl_cmd(utils.get_library_wrapper(), argv,
utils.get_modificators()),
"status":
status.status_cmd,
"config":
config.config_cmd,
"pcsd":
pcsd.pcsd_cmd,
"node":
lambda argv: node.node_cmd(utils.get_library_wrapper(), argv,
utils.get_modificators()),
"quorum":
lambda argv: quorum.quorum_cmd(utils.get_library_wrapper(), argv,
utils.get_modificators()),
"qdevice":
lambda argv: qdevice.qdevice_cmd(utils.get_library_wrapper(), argv,
utils.get_modificators()),
"alert":
lambda args: alert.alert_cmd(utils.get_library_wrapper(), args,
utils.get_modificators()),
"booth":
lambda argv: booth.booth_cmd(utils.get_library_wrapper(), argv,
utils.get_modificators()),
}
if command not in cmd_map:
usage.main()
sys.exit(1)
# root can run everything directly, also help can be displayed,
# working on a local file also do not need to run under root
if (os.getuid() == 0) or (argv and argv[0] == "help") or usefile:
cmd_map[command](argv)
return
# specific commands need to be run under root account, pass them to pcsd
# don't forget to allow each command in pcsd.rb in "post /run_pcs do"
root_command_list = [
['cluster', 'auth', '...'],
['cluster', 'corosync', '...'],
['cluster', 'destroy', '...'],
['cluster', 'disable', '...'],
['cluster', 'enable', '...'],
['cluster', 'node', '...'],
['cluster', 'pcsd-status', '...'],
['cluster', 'setup', '...'],
['cluster', 'start', '...'],
['cluster', 'stop', '...'],
['cluster', 'sync', '...'],
# ['config', 'restore', '...'], # handled in config.config_restore
['pcsd', 'sync-certificates'],
['status', 'nodes', 'corosync-id'],
['status', 'nodes', 'pacemaker-id'],
['status', 'pcsd', '...'],
]
argv_cmd = argv[:]
argv_cmd.insert(0, command)
for root_cmd in root_command_list:
if ((argv_cmd == root_cmd)
or (root_cmd[-1] == "..."
and argv_cmd[:len(root_cmd) - 1] == root_cmd[:-1])):
# handle interactivity of 'pcs cluster auth'
if argv_cmd[0:2] == ["cluster", "auth"]:
if "-u" not in utils.pcs_options:
username = utils.get_terminal_input('Username: ')
orig_argv.extend(["-u", username])
if "-p" not in utils.pcs_options:
password = utils.get_terminal_password()
orig_argv.extend(["-p", password])
# call the local pcsd
err_msgs, exitcode, std_out, std_err = utils.call_local_pcsd(
orig_argv, True)
if err_msgs:
for msg in err_msgs:
utils.err(msg, False)
sys.exit(1)
if std_out.strip():
print(std_out)
if std_err.strip():
sys.stderr.write(std_err)
sys.exit(exitcode)
return
cmd_map[command](argv)
def test_dont_upgrade_2dash_master_outside_of_resource_create(self):
self.assertEqual(
["first", "--master", "second"],
upgrade_args(["first", "--master", "second"])
)
def test_upgrade_2dash_master_in_resource_create(self):
self.assertEqual(
["resource", "create", "master", "second"],
upgrade_args(["resource", "create", "--master", "second"])
)
def test_upgrade_2dash_cloneopt_with_value(self):
self.assertEqual(
["first", "clone", "1", "second"],
upgrade_args(["first", "--cloneopt=1", "second"])
)
def test_upgrade_2dash_clone(self):
self.assertEqual(
["first", "clone", "second"],
upgrade_args(["first", "--clone", "second"])
)
def test_returns_the_same_args_when_no_older_versions_detected(self):
args = ["first", "second"]
self.assertEqual(args, upgrade_args(args))
