diff --git a/bpytop.py b/bpytop.py index a1c2b3a..b1cc28e 100755 --- a/bpytop.py +++ b/bpytop.py @@ -64,23 +64,27 @@ args.add_argument("-p" , "--proc" ,action="store_true" ,help ="Start in minimal args.add_argument("-s" , "--stat" ,action="store_true" ,help ="Start in minimal mode without process box") args.add_argument("-v" , "--version" ,action="store_true" ,help ="Show version info and exit") args.add_argument("--debug" ,action="store_true" ,help ="Start with loglevel set to DEBUG overriding value set in config") -stdargs = args.parse_args() - -if stdargs.version: - print(f'bpytop version: {VERSION}\n' - f'psutil version: {".".join(str(x) for x in psutil.version_info)}') - raise SystemExit(0) ARG_MODE: str = "" -if stdargs.full: - ARG_MODE = "full" -elif stdargs.proc: - ARG_MODE = "proc" -elif stdargs.stat: - ARG_MODE = "stat" +if __name__ == "__main__": + stdargs = args.parse_args() + + if stdargs.version: + print(f'bpytop version: {VERSION}\n' + f'psutil version: {".".join(str(x) for x in psutil.version_info)}') + raise SystemExit(0) -DEBUG = stdargs.debug + if stdargs.full: + ARG_MODE = "full" + elif stdargs.proc: + ARG_MODE = "proc" + elif stdargs.stat: + ARG_MODE = "stat" + + DEBUG = stdargs.debug +else: + DEBUG = False #? Variables ------------------------------------------------------------------------------------->