diff --git a/install.py b/install.py index b40ea14..c06fe6e 100644 --- a/install.py +++ b/install.py @@ -5,6 +5,10 @@ linux_packages = [ 'zsh' ] +linux_after_command = { + 'zsh': 'chsh -s $(which zsh)' +} + os_type = platform.uname().system.lower() is_linux = os_type == "linux" @@ -31,6 +35,8 @@ def install(package): else: print(f"Install not supported for: {os_type}") exit(1) + if is_linux and package in after_command: + os.system(linux_after_command[package]) def check_os(): if is_linux or is_macos or is_windows: