zhangyiss/the-littlest-jupyterhub
1
0
Fork 0
mirror of https://github.com/jupyterhub/the-littlest-jupyterhub.git synced 2025-12-18 21:54:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

early check if it was called as non-root user

Browse Source
This commit is contained in:
anyushevai
2018-10-23 23:45:33 +03:00
parent 35309d13d0
commit 12aaf382aa
1 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
tljh/config.py
View File

@@ -219,6 +219,10 @@ def _is_list(item):
def main(argv=None):
if os.geteuid() != 0:
print("Perhaps you didn't use `sudo -E`?")
raise(SystemExit)
if argv is None:
argv = sys.argv[1:]
@@ -295,22 +299,18 @@ def main(argv=None):
args = argparser.parse_args(argv)
try:
if args.action == 'show':
show_config(args.config_path)
elif args.action == 'set':
set_config_value(args.config_path, args.key_path, parse_value(args.value))
elif args.action == 'add-item':
add_config_value(args.config_path, args.key_path, parse_value(args.value))
elif args.action == 'remove-item':
remove_config_value(args.config_path, args.key_path, parse_value(args.value))
elif args.action == 'reload':
reload_component(args.component)
else:
argparser.print_help()
except Exception as e:
print(str(e))
print("Perhaps you didn't use `sudo -E`?")
if args.action == 'show':
show_config(args.config_path)
elif args.action == 'set':
set_config_value(args.config_path, args.key_path, parse_value(args.value))
elif args.action == 'add-item':
add_config_value(args.config_path, args.key_path, parse_value(args.value))
elif args.action == 'remove-item':
remove_config_value(args.config_path, args.key_path, parse_value(args.value))
elif args.action == 'reload':
reload_component(args.component)
else:
argparser.print_help()
if __name__ == '__main__':