clean up unnecessary files

harris21 · Jul 21, 2016 · 2c75725 · 2c75725
1 parent 732756a
commit 2c75725
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/afot.py b/afot.py
@@ -252,6 +252,16 @@ def main():
 
         time.sleep(2)
 
+        print colored('Cleaning up unnecessary files...', 'yellow')
+
+        # Delete unnecessary files
+        deleteFile('windows.csv')
+        deleteFile('hashes.txt')
+        deleteFile('nist.csv')
+        deleteFile('infected.txt')
+
+        time.sleep(2)
+
         print colored('\nreport.csv', 'green') + colored(' file has been created containing a full report!', 'yellow')
 
         sys.exit(1)