admin: Add some basic protection against running the CLI as root

lkhq · Sep 24, 2023 · b9a9cd0 · b9a9cd0
1 parent 73ea575
commit b9a9cd0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/lkadmin/cli.py b/src/lkadmin/cli.py
@@ -85,7 +85,7 @@ def _register_commands():
 def run(mainfile, args):
     from rich.traceback import install
 
-    from laniakea.utils import set_process_title
+    from laniakea.utils.misc import set_process_title, ensure_laniakea_master_user
 
     set_process_title('lk-admin')
     if len(args) == 0:
@@ -95,6 +95,9 @@ def run(mainfile, args):
     global __mainfile
     __mainfile = mainfile
 
+    # some safeguarding against running as root or any other "wrong" user
+    ensure_laniakea_master_user(warn_only=True)
+
     install(show_locals=True, suppress=[click])
     _register_commands()
     cli()  # pylint: disable=no-value-for-parameter