Update setup.py to look for empty apt_packages.txt

Signed-off-by: jayofelony <oudshoorn.jeroen@gmail.com>
This commit is contained in:
jayofelony
2024-03-25 11:58:46 +01:00
parent 8fe503c67d
commit f140fe1a2d

View File

@ -34,11 +34,12 @@ def install_file(source_filename, dest_filename):
def install_system_files(): def install_system_files():
f = open("apt_packages.txt", "r") data_path = None
if os.stat("apt_packages.txt").st_size != 0:
for x in f: f = open("apt_packages.txt", "r")
os.system(f"apt-get install -y {x}") for x in f:
f.close() os.system(f"apt-get install -y {x}")
f.close()
setup_path = os.path.dirname(__file__) setup_path = os.path.dirname(__file__)
if platform.machine().startswith('arm'): if platform.machine().startswith('arm'):
data_path = os.path.join(setup_path, "builder/data/32bit") data_path = os.path.join(setup_path, "builder/data/32bit")
@ -63,9 +64,7 @@ class CustomInstall(install):
def run(self): def run(self):
super().run() super().run()
if os.geteuid() != 0: if os.geteuid() != 0:
warnings.warn( warnings.warn("Not running as root, can't install pwnagotchi system files!")
"Not running as root, can't install pwnagotchi system files!"
)
return return
install_system_files() install_system_files()
restart_services() restart_services()