1
0
Fork 0
mirror of https://github.com/dbarzin/pandora-box.git synced 2025-07-19 05:19:40 +02:00

fix bug in remove key

This commit is contained in:
dbarzin 2023-03-03 15:57:11 +01:00
parent 5dfd28ca6e
commit bda89e2bfe

View file

@ -577,19 +577,33 @@ class PandoraBox:
"""Remove infected files""" """Remove infected files"""
if len(self.infected_files) > 0: if len(self.infected_files) > 0:
# display message # display message
self._log(f"{len(self.infected_files)} infected_files detecetd :") self._log(f"{len(self.infected_files)} infected files detecetd:")
logging.info(f"infeted_files={len(self.infected_files)}") logging.info(f"infeted_files={len(self.infected_files)}")
# print list of files
for file in self.infected_files:
self._log(file)
if not self.has_curses: if not self.has_curses:
self.display_image("BAD") self.display_image("BAD")
self.wait_mouse_click() self.wait_mouse_click()
else: else:
# print list of files
cnt = 0
for file in self.infected_files:
self._log(file)
cnt = cnt + 1
if (cnt >= 10):
self._log('...')
break
# wait for clean
self._log('PRESS KEY TO CLEAN') self._log('PRESS KEY TO CLEAN')
self.screen.getch() self.screen.getch()
# check key is still here
try:
os.statvfs(self.mount_point)
except Exception :
self._log("Device not cleaned !")
logging.info('device_not_cleaned')
return "WAIT"
# Remove infected files # Remove infected files
files_removed = 0 files_removed = 0
for file in self.infected_files: for file in self.infected_files: