split-secret/scripts/generate-hashtable.py

45 lines
2.0 KiB
Python

import argparse
import random
import string
def getPassword():
characters = string.ascii_letters + string.digits
return ''.join(random.choice(characters) for i in range(int(64*quota_factor))).upper();
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('-a', '--amount',type=int, dest='amount_of_secret_holders',required=True)
parser.add_argument('-q', '--quota', type=int, dest='decryption_quota', choices=range(1,101),required=True)
args = parser.parse_args()
amount_of_secret_holders = args.amount_of_secret_holders;
decryption_quota = args.decryption_quota;
quota_factor=decryption_quota/100;
password_group_members_amount=amount_of_secret_holders * quota_factor
amount_of_partner_secrets=(amount_of_secret_holders * password_group_members_amount)
#required_passwords=amount_of_partner_secrets * ( amount_of_secret_holders -1) ;
required_passwords=amount_of_partner_secrets * amount_of_secret_holders;
print(quota_factor);
print(amount_of_secret_holders);
print(decryption_quota);
print(required_passwords);
password_groups = {}
password_group_index = 0
secret_holder_index = 0
while password_group_index < required_passwords :
password_groups[password_group_index] = {};
password_groups[password_group_index]['members'] = {}
password_groups[password_group_index]['password'] = ''
group_members_count = 0;
while group_members_count < password_group_members_amount :
if secret_holder_index == amount_of_secret_holders:
secret_holder_index = 0;
password_groups[password_group_index]['members'][secret_holder_index] = getPassword();
password_groups[password_group_index]['password'] += ''.join(password_groups[password_group_index]['members'][secret_holder_index])
secret_holder_index += 1;
group_members_count += 1;
password_group_index += 1;
print(password_groups);