From 38251fdc8f2f9b72a991032456ed9717ff613718 Mon Sep 17 00:00:00 2001
From: dbarzin <dbarzin@gmail.com>
Date: Fri, 24 Feb 2023 14:12:14 +0100
Subject: [PATCH] work on messages

---
 pandora-box.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/pandora-box.py b/pandora-box.py
index ea926fc..a1cf443 100755
--- a/pandora-box.py
+++ b/pandora-box.py
@@ -548,17 +548,22 @@ class PandoraBox:
                 self._log('PRESS KEY TO CLEAN')
                 self.screen.getch()
             # Remove infected files
+            files_removed = 0
             for file in self.infected_files:
                 try :
                     os.remove(file)
                     self._log(f"{file} removed")
                     logging.info(f'removed="{file}"')
+                    files_removed += 1
                 except Exception as ex :
                     self._log(f"Unexpected error: {str(ex)}")
                     logging.info(f'error="{str(ex)}"', exc_info=True)
             os.system("sync")
             if not self.has_curses:
                 self.display_image("OK")
+            else
+                self._log('Device cleaned !')
+            logging.info(f'cleaned="{files_removed}/{len(self.infected_files)}"')
         else:
             if not self.has_curses:
                 self.display_image("OK")