kevinveenbirkenbach/split-secret
1
0
Fork 0
mirror of https://github.com/kevinveenbirkenbach/splitted-secret.git synced 2024-11-22 18:21:05 +01:00
Code Issues Packages Projects Releases Wiki Activity

Implemented basic cleanup

Browse Source
This commit is contained in:
Kevin Veen-Birkenbach 2022-12-09 13:22:57 +01:00
parent f547b79675
commit fa82f4d5da
5 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/classes/AbstractSplittedSecret.py
View File

@ -1,3 +1,6 @@
class AbstractSplittedSecret: from .Cli import Cli
class AbstractSplittedSecret(Cli):
def __init__(self): def __init__(self):
super(Cli, self).__init__()
self.encrypted_splitted_password_files_folder = "data/encrypted/splitted_password_files/" self.encrypted_splitted_password_files_folder = "data/encrypted/splitted_password_files/"

9
scripts/classes/Cleanup.py
View File

@ -1,7 +1,10 @@
from .AbstractSplittedSecret import AbstractSplittedSecret from .AbstractSplittedSecret import AbstractSplittedSecret
class Cleanup(AbstractSplittedSecret): class Cleanup(AbstractSplittedSecret):
def __init__(self): def __init__(self):
pass super(Cleanup, self).__init__()
def deleteAllEncryptedFiles(self):
self.executeCommand('rm -v ' + self.encrypted_splitted_password_files_folder + '*')
print(self.getCommandString())
print(self.getOutputString())
def deleteAll(self): def deleteAll(self):
self.deleteAllEncryptedFiles()
pass

2
scripts/classes/Cli.py
View File

@ -24,7 +24,7 @@ class Cli(object):
return self.output return self.output
def getOutputString(self): def getOutputString(self):
return str(' '.join(self.output)) return str('\n'.join(self.output))
def getCommandString(self): def getCommandString(self):
return self.command return self.command

4
scripts/classes/Generate.py
View File

@ -3,14 +3,12 @@ import string
import math import math
import numpy import numpy
import re import re
from .Cli import Cli
from .AbstractSplittedSecret import AbstractSplittedSecret from .AbstractSplittedSecret import AbstractSplittedSecret
class Generate(AbstractSplittedSecret,Cli): class Generate(AbstractSplittedSecret):
def __init__(self, amount_of_secret_holders, decryption_quota): def __init__(self, amount_of_secret_holders, decryption_quota):
super(Generate, self).__init__() super(Generate, self).__init__()
super(Cli, self).__init__()
self.amount_of_secret_holders = amount_of_secret_holders self.amount_of_secret_holders = amount_of_secret_holders
self.decryption_quota = decryption_quota self.decryption_quota = decryption_quota
self.decrypted_master_password_file_path="data/decrypted/password_files/master-password.txt" self.decrypted_master_password_file_path="data/decrypted/password_files/master-password.txt"

7
scripts/main.py
View File

@ -4,13 +4,14 @@ from classes.Cleanup import Cleanup
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('-m', '--mode',type=str, dest='mode',required=True,choices=['cleanup','generate']) parser.add_argument('--mode',type=str, dest='mode',required=True,choices=['cleanup','generate'])
parser.add_argument('-a', '--amount',type=int, dest='amount_of_secret_holders',required=False,choices=range(1,9)) parser.add_argument('--amount',type=int, dest='amount_of_secret_holders',required=False,choices=range(1,9))
parser.add_argument('-q', '--quota', type=int, dest='decryption_quota', choices=range(1,101),required=False) parser.add_argument('--quota', type=int, dest='decryption_quota', choices=range(1,101),required=False)
args = parser.parse_args() args = parser.parse_args()
if args.mode == 'cleanup': if args.mode == 'cleanup':
cleanup = Cleanup() cleanup = Cleanup()
cleanup.deleteAll()
exit() exit()
if args.mode == 'generate': if args.mode == 'generate':