From 5d2b11e60fa244c1f8af78d4238848f45a4c0d2c Mon Sep 17 00:00:00 2001 From: demostanis Date: Fri, 23 Oct 2020 15:53:31 +0200 Subject: Ran autopep8 --- archinstall/__main__.py | 73 +++++++++++++++++++++++++++---------------------- 1 file changed, 41 insertions(+), 32 deletions(-) diff --git a/archinstall/__main__.py b/archinstall/__main__.py index d2ef22cb..de7cf994 100644 --- a/archinstall/__main__.py +++ b/archinstall/__main__.py @@ -1,59 +1,68 @@ from urllib.parse import urlparse -import archinstall, sys, os, glob +import archinstall +import sys +import os +import glob import urllib.request # TODO: Learn the dark arts of argparse... # (I summon thee dark spawn of cPython) + class ProfileNotFound(BaseException): - pass + pass + def find_examples(): - """ - Used to locate the examples, bundled with the module or executable. + """ + Used to locate the examples, bundled with the module or executable. - :return: {'guided.py' : './examples/guided.py', '' : ''} - :rtype: dict - """ - cwd = os.path.abspath(f'{os.path.dirname(__file__)}') - examples = f"{cwd}/examples" + :return: {'guided.py' : './examples/guided.py', '' : ''} + :rtype: dict + """ + cwd = os.path.abspath(f'{os.path.dirname(__file__)}') + examples = f"{cwd}/examples" + + return {os.path.basename(path): path for path in glob.glob(f'{examples}/*.py')} - return {os.path.basename(path): path for path in glob.glob(f'{examples}/*.py')} def find(url): parsed_url = urlparse(url) if not parsed_url.scheme: examples = find_examples() if f"{url}.py" in examples: - return open(examples[f"{url}.py"]).read() + return open(examples[f"{url}.py"]).read() try: - return open(url, 'r').read() + return open(url, 'r').read() except FileNotFoundError: - return ProfileNotFound(f"File {url} does not exist") + return ProfileNotFound(f"File {url} does not exist") elif parsed_url.scheme in ('https', 'http'): return urllib.request.urlopen(url).read().decode('utf-8') else: return ProfileNotFound(f"Cannot handle scheme {parsed_url.scheme}") + def run_as_a_module(): - """ - Since we're running this as a 'python -m archinstall' module OR - a nuitka3 compiled version of the project. - This function and the file __main__ acts as a entry point. - """ - if len(sys.argv) == 1: sys.argv.append('guided') - - try: - profile = find(sys.argv[1]) - except ProfileNotFound as err: - print(f"Couldn't find file: {err}") - sys.exit(1) - - try: - exec(profile) # Is this is very safe? - except Exception as err: - print(f"Failed to run profile... {err}") - sys.exit(1) # Should prompt for another profile path instead + """ + Since we're running this as a 'python -m archinstall' module OR + a nuitka3 compiled version of the project. + This function and the file __main__ acts as a entry point. + """ + if len(sys.argv) == 1: + sys.argv.append('guided') + + try: + profile = find(sys.argv[1]) + except ProfileNotFound as err: + print(f"Couldn't find file: {err}") + sys.exit(1) + + try: + exec(profile) # Is this is very safe? + except Exception as err: + print(f"Failed to run profile... {err}") + sys.exit(1) # Should prompt for another profile path instead + if __name__ == '__main__': - run_as_a_module() + run_as_a_module() -- cgit v1.2.3-54-g00ecf