diff --git a/src/__main__.py b/src/__main__.py index 61a1ace..8d34a8a 100644 --- a/src/__main__.py +++ b/src/__main__.py @@ -1,15 +1,14 @@ import os -import sys from src.packages import install_packages from src.util.user import Input, Print -if os.geteuid() != 0: - sys.exit("Please run this program as root user") - - def main(): + if os.geteuid() != 0: + Print.err("You need to run this program as root user") + return + if Input.yes_no("Do you wish to perform package install (from `packages.yaml`)?"): install_packages()