From fb95159f322bcd057cfbad054d675d28da373d56 Mon Sep 17 00:00:00 2001 From: timeshifter Date: Wed, 31 May 2023 15:40:42 +0200 Subject: [PATCH] feat: verify we are root --- update.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/update.py b/update.py index 873340a..3daf308 100755 --- a/update.py +++ b/update.py @@ -3,6 +3,7 @@ from __future__ import annotations from functools import lru_cache +from os import geteuid from pathlib import Path from subprocess import call, check_output from datetime import datetime @@ -14,6 +15,9 @@ KEEP_DAYS = 3 def main(): + if not are_we_root(): + print("we must be root") + exit(1) setup_logger() build_images() update_containers() @@ -149,6 +153,10 @@ def setup_logger() -> None: logging.info('set up logger') +def are_we_root() -> bool: + return geteuid() == 0 + + if __name__ == "__main__": main()