diff --git a/install.py b/install.py index d68085f..23872b5 100644 --- a/install.py +++ b/install.py @@ -1,31 +1,22 @@ import os import platform -is_linux = False -is_macos = False -is_windows = False +def get_os_type(): return platform.uname().system.lower() -is_debian = False -is_rhel = False +def is_linux(): return get_os_type() == "linux" +def is_macos(): return get_os_type() == "darwin" +def is_windows(): return get_os_type() == "windows" + +def is_debian(): return os.system('apt --version > /dev/null 2>&1') >> 8 == 0 +def is_rhel(): return os.system('dnf --version > /dev/null 2>&1') >> 8 == 0 def check_os(): - os_type = platform.uname().system.lower() - - is_linux = os_type == "linux" - is_macos = os_type == "darwin" - is_windows = os_type == "windows" - - if is_linux or is_macos or is_windows: - print(f"Running on {os_type}") + if is_linux() or is_macos() or is_windows(): + print(f"Running on {get_os_type()}") else: - print(f"Unsupported OS family: {os_type}") + print(f"Unsupported OS family: {get_os_type()}") exit(1) - - is_debian = os.system('apt --version > /dev/null 2>&1') >> 8 == 0 - is_rhel = os.system('dnf --version > /dev/null 2>&1') >> 8 == 0 - print(f'is_windows: {is_windows}, is_macos: {is_macos}, is_linux: {is_linux}, is_debian: {is_debian}, is_rhel: {is_rhel}') - if __name__ == "__main__": check_os() - print(f'is_windows: {is_windows}, is_macos: {is_macos}, is_linux: {is_linux}, is_debian: {is_debian}, is_rhel: {is_rhel}') \ No newline at end of file + print(f'is_windows: {is_windows()}, is_macos: {is_macos()}, is_linux: {is_linux()}, is_debian: {is_debian()}, is_rhel: {is_rhel()}') \ No newline at end of file