add warning & abort if tljh-config is called as non-root user

This commit is contained in:
anyushevai
2018-10-21 23:55:47 +03:00
parent 544d0b3f4b
commit 35309d13d0

View File

@@ -223,7 +223,11 @@ def main(argv=None):
argv = sys.argv[1:] argv = sys.argv[1:]
from .log import init_logging from .log import init_logging
init_logging() try:
init_logging()
except Exception as e:
print(str(e))
print("Perhaps you didn't use `sudo -E`?")
argparser = argparse.ArgumentParser() argparser = argparse.ArgumentParser()
argparser.add_argument( argparser.add_argument(
@@ -291,18 +295,22 @@ def main(argv=None):
args = argparser.parse_args(argv) args = argparser.parse_args(argv)
if args.action == 'show': try:
show_config(args.config_path) if args.action == 'show':
elif args.action == 'set': show_config(args.config_path)
set_config_value(args.config_path, args.key_path, parse_value(args.value)) elif args.action == 'set':
elif args.action == 'add-item': set_config_value(args.config_path, args.key_path, parse_value(args.value))
add_config_value(args.config_path, args.key_path, parse_value(args.value)) elif args.action == 'add-item':
elif args.action == 'remove-item': add_config_value(args.config_path, args.key_path, parse_value(args.value))
remove_config_value(args.config_path, args.key_path, parse_value(args.value)) elif args.action == 'remove-item':
elif args.action == 'reload': remove_config_value(args.config_path, args.key_path, parse_value(args.value))
reload_component(args.component) elif args.action == 'reload':
else: reload_component(args.component)
argparser.print_help() else:
argparser.print_help()
except Exception as e:
print(str(e))
print("Perhaps you didn't use `sudo -E`?")
if __name__ == '__main__': if __name__ == '__main__':