Check if root after processing args

Allows for the `-h/--help` argument to be used without needing to elevate privileges.
This commit is contained in:
Carl 2019-10-16 18:56:24 +10:00 committed by GitHub
parent c9846e8984
commit 1b8a5e9eec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -459,10 +459,6 @@ def find_installer_app(mountpoint):
def main():
'''Do the main thing here'''
if os.getuid() != 0:
sys.exit('This command requires root (to install packages), so please '
'run again with sudo or as root.')
parser = argparse.ArgumentParser()
parser.add_argument('--seedprogram', default='',
help='Which Seed Program catalog to use. Valid values '
@ -488,6 +484,10 @@ def main():
help='Ignore any previously cached files.')
args = parser.parse_args()
if os.getuid() != 0:
sys.exit('This command requires root (to install packages), so please '
'run again with sudo or as root.')
if args.catalogurl:
su_catalog_url = args.catalogurl
elif args.seedprogram: