Implemented cleanup for user

scripts/classes/Cleanup.py

@@ -20,6 +20,12 @@ class Cleanup(AbstractSplittedSecret):
     def deleteAllEncryptedFiles(self):
         for folder_path in self.encrypted_files_folders:
             self.deleteAllFilesInFolder(folder_path)
+            
+    def cleanupForUser(self,user):
+        self.executeCommand('find "' + self.encrypted_folder + '" -not -name "*' + str(user) +'*" -type f -print | xargs rm -v')
+        print(self.getCommandString())
+        print(self.getOutputString())
+        
 
     def deleteAll(self):
         self.deleteAllEncryptedFiles()

scripts/main.py

@@ -9,12 +9,15 @@ if __name__ == '__main__':
     parser.add_argument('--amount',type=int, dest='amount_of_secret_holders',required=False,choices=range(1,9))
     parser.add_argument('--quota', type=int, dest='decryption_quota', choices=range(1,101),required=False)
     parser.add_argument('--master-password',type=str, dest='master_password',required=False)
+    parser.add_argument('--user',type=int, dest='user',choices=range(1,9),required=False)
     args = parser.parse_args()
     
     if args.mode == 'cleanup':
         cleanup = Cleanup()
-        cleanup.deleteAll()
-        exit()
+        if args.user is None: 
+            cleanup.deleteAll()
+            exit()
+        cleanup.cleanupForUser(args.user)
         
     if args.mode == 'generate':
         if args.master_password is None: