split-secret/scripts/classes/Decryption.py

from .AbstractSplittedSecret import AbstractSplittedSecret
import json
class Decryption(AbstractSplittedSecret):
    
    def __init__(self):
        self.user_id='0';
        self.user_password=''
        super(Decryption, self).__init__()
    
    def setUserId(self,user_id):
        self.user_id=str(user_id)
        self.user_file_decrypted_path = self.getUserFilePath(self.user_id,"decrypted")
    
    def setUserPassword(self,user_password):
        self.user_password = str(user_password)
    
    def loadJsonFile(self,file_path):
        file = open(file_path)
        data = json.load(file)
        file.close()
        return data
    
    def getNeededEncryptersAmount(self):
        return len(str(list(self.user_data['groups'].keys())[0]))-1
    
    def decryptFile(self,password,input_file_path,output_file_path):
        self.executeCommand('gpg --batch --passphrase "'+ password + '" -o "' + output_file_path +'" "'+ input_file_path+'"')
    
    def decryptUserFile(self):
        input_file_path = self.getUserFilePath(self.user_id,"encrypted")
        self.decryptFile(self.user_password, input_file_path, self.user_file_decrypted_path)
        
    def decryptAccumulatedFile(self):
        input_file_path = self.getAccumulatedFilePath("encrypted")
        output_file_path = self.getAccumulatedFilePath("decrypted")
        self.decryptFile(self.user_password, input_file_path, output_file_path)
        
    def setUserData(self):
        self.decryptUserFile()
        self.user_data = self.loadJsonFile(self.user_file_decrypted_path)