python: Move check_root() from setup to lib

Signed-off-by: Nathan Chancellor <nathan@kernel.org>

python/lib/setup.py

@@ -147,11 +147,6 @@ def check_ip(ip_to_check):
     ipaddress.ip_address(ip_to_check)
 
 
-def check_root():
-    if os.geteuid() != 0:
-        raise RuntimeError("root access is required!")
-
-
 # Easier than os.walk() + shutil.chown()
 def chown(new_user, folder):
     lib.utils.run(['chown', '-R', f"{new_user}:{new_user}", folder])

python/lib/utils.py

@@ -31,6 +31,11 @@ def call_git_loud(directory, cmd, **kwargs):
     return call_git(directory, cmd, **kwargs, capture_output=False)
 
 
+def check_root():
+    if os.geteuid() != 0:
+        raise RuntimeError("root access is required!")
+
+
 def chronic(*args, **kwargs):
     kwargs.setdefault('capture_output', True)
 

python/setup/almalinux.py

@@ -18,7 +18,7 @@ def get_alma_version():
 
 
 def prechecks():
-    lib.setup.check_root()
+    lib.utils.check_root()
     alma_version = get_alma_version()
     if alma_version not in (9, ):
         raise RuntimeError(

python/setup/alpine.py

@@ -150,7 +150,7 @@ def setup_podman(user_name):
     if not (password := args.user_password):
         password = getpass.getpass(prompt='Password for Alpine user account: ')
 
-    lib.setup.check_root()
+    lib.utils.check_root()
     enable_community_repo()
     update_and_install_packages()
     setup_user(args.user_name, password)

python/setup/arch.py

@@ -649,7 +649,7 @@ def parse_arguments():
 
 
 def prechecks():
-    lib.setup.check_root()
+    lib.utils.check_root()
 
 
 def setup_doas(username):

python/setup/debian.py

@@ -64,7 +64,7 @@ def pi_setup(user_name):
 
 
 def prechecks():
-    lib.setup.check_root()
+    lib.utils.check_root()
 
     supported_versions = (
         'bullseye',

python/setup/equinix.py

@@ -111,7 +111,7 @@ def parse_arguments():
 if __name__ == '__main__':
     args = parse_arguments()
 
-    lib.setup.check_root()
+    lib.utils.check_root()
 
     drive = args.drive
     folder = Path(args.folder)

python/setup/fedora.py

@@ -111,7 +111,7 @@ def machine_is_trusted():
 
 
 def prechecks():
-    lib.setup.check_root()
+    lib.utils.check_root()
     if (fedora_version := get_fedora_version()) not in range(MIN_FEDORA_VERSION,
                                                              MAX_FEDORA_VERSION + 1):
         raise RuntimeError(

python/setup/rocky.py

@@ -18,7 +18,7 @@ def get_rocky_version():
 
 
 def prechecks():
-    lib.setup.check_root()
+    lib.utils.check_root()
     rocky_version = get_rocky_version()
     if rocky_version not in (9, ):
         raise RuntimeError(

python/setup/ubuntu.py

@@ -28,7 +28,7 @@ def parse_arguments():
 
 
 def prechecks():
-    lib.setup.check_root()
+    lib.utils.check_root()
 
     supported_versions = ('focal', 'jammy', 'kinetic')
     if (codename := lib.setup.get_version_codename()) not in supported_versions: