split-secret/scripts/classes/Cleanup.py

from .AbstractSplittedSecret import AbstractSplittedSecret
class Cleanup(AbstractSplittedSecret):
    def __init__(self):
        super(Cleanup, self).__init__()
        self.encrypted_files_folders = [self.decrypted_group_files_folder,self.decrypted_user_files_folder]
        self.decrypted_files_folders = [self.encrypted_group_files_folder,self.encrypted_user_files_folder]
    
    def deleteAllFilesInFolder(self,folder_path):
        try:
            self.executeCommand('rm -v ' + folder_path + '*')
            print(self.getCommandString())
            print(self.getOutputString())
        except:
            pass    
    
    def deleteAllDecryptedFiles(self):
        for folder_path in self.decrypted_files_folders:
            self.deleteAllFilesInFolder(folder_path)
        
    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()
        self.deleteAllDecryptedFiles()