kevinveenbirkenbach
/
split-secret
mirror of https://github.com/kevinveenbirkenbach/splitted-secret.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Implemented loading of json file

This commit is contained in:
Kevin Veen-Birkenbach 2022-12-09 22:55:33 +01:00
parent 0831ab3448
commit 72765e280d
2 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
scripts/classes/Decryption.py
View File

@ -1,4 +1,5 @@
from .AbstractSplittedSecret import AbstractSplittedSecret
import json
class Decryption(AbstractSplittedSecret):
def __init__(self):
@ -8,19 +9,29 @@ class Decryption(AbstractSplittedSecret):
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 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")
output_file_path = self.getUserFilePath(self.user_id,"decrypted")
self.decryptFile(self.user_password, input_file_path, output_file_path)
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)
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)

6
scripts/main.py
View File

@ -39,11 +39,13 @@ if __name__ == '__main__':
decrypt.setUserPassword(getpass())
print("Decrypting User File...")
try:
decrypt.decryptUserFile();
decrypt.setUserData();
break;
except:
print("Wrong password :(")
print("File encrypted :) ")
print("File decrypted :) ")
print("Please contact ")
print(decrypt.user_data)
exit()
print("Decrypting accumulated file...")
decrypt.setUserPassword(args.master_password)