From f140fe1a2de41d4d7c7b197f1e5dd107a5f76ac3 Mon Sep 17 00:00:00 2001 From: jayofelony Date: Mon, 25 Mar 2024 11:58:46 +0100 Subject: [PATCH] Update setup.py to look for empty apt_packages.txt Signed-off-by: jayofelony --- setup.py | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/setup.py b/setup.py index fb183b9f..343725e8 100644 --- a/setup.py +++ b/setup.py @@ -34,11 +34,12 @@ def install_file(source_filename, dest_filename): def install_system_files(): - f = open("apt_packages.txt", "r") - - for x in f: - os.system(f"apt-get install -y {x}") - f.close() + data_path = None + if os.stat("apt_packages.txt").st_size != 0: + f = open("apt_packages.txt", "r") + for x in f: + os.system(f"apt-get install -y {x}") + f.close() setup_path = os.path.dirname(__file__) if platform.machine().startswith('arm'): data_path = os.path.join(setup_path, "builder/data/32bit") @@ -63,9 +64,7 @@ class CustomInstall(install): def run(self): super().run() if os.geteuid() != 0: - warnings.warn( - "Not running as root, can't install pwnagotchi system files!" - ) + warnings.warn("Not running as root, can't install pwnagotchi system files!") return install_system_files() restart_services()