diff --git a/src/keymanager.py b/src/keymanager.py index 3798b463..b2ecad97 100755 --- a/src/keymanager.py +++ b/src/keymanager.py @@ -1,9 +1,11 @@ -''' - Onionr - Private P2P Communication +"""Onionr - Private P2P Communication. - Load, save, and delete the user's public key pairs (does not handle peer keys) -''' -''' +Load, save, and delete the user's public key pairs (does not handle peer keys) +""" +from onionrutils import bytesconverter +from onionrcrypto import generate +import filepaths +""" This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or @@ -16,18 +18,18 @@ You should have received a copy of the GNU General Public License along with this program. If not, see . -''' -from onionrutils import bytesconverter -from onionrcrypto import generate -import os -import filepaths +""" + class KeyManager: def __init__(self): self.keyFile = filepaths.keys_file def addKey(self, pubKey=None, privKey=None): - '''Add a new key pair, either specified or none to generate a new pair automatically''' + """Add a new key pair. + + either specified or None to generate a new pair automatically + """ if type(pubKey) is type(None) and type(privKey) is type(None): pubKey, privKey = generate.generate_pub_key() pubKey = bytesconverter.bytes_to_str(pubKey) @@ -43,7 +45,7 @@ class KeyManager: return (pubKey, privKey) def removeKey(self, pubKey): - '''Remove a key pair by pubkey''' + """Remove a key pair by pubkey""" keyList = self.getPubkeyList() keyData = '' try: @@ -56,7 +58,7 @@ class KeyManager: keyFile.write(keyData) def getPubkeyList(self): - '''Return a list of the user's keys''' + """Return a list of the user's keys""" keyList = [] try: with open(self.keyFile, "r") as keyFile: @@ -65,9 +67,10 @@ class KeyManager: keyData = '' keyData = keyData.split('\n') for pair in keyData: - if len(pair) > 0: keyList.append(pair.split(',')[0]) + if len(pair) > 0: + keyList.append(pair.split(',')[0]) return keyList - + def getPrivkey(self, pubKey): privKey = None with open(self.keyFile, "r") as keyFile: @@ -75,4 +78,4 @@ class KeyManager: for pair in keyData.split('\n'): if pubKey in pair or pubKey.replace('=', '') in pair: privKey = pair.split(',')[1] - return privKey \ No newline at end of file + return privKey