def test_add_content(monkeypatch): gpg = get_GPG('{"One": {"user":"******","password":"******","url":"url1","other":"other1"},"example": {"user":"******","password":"******","url":"example","other":"example"}}', 'secrets_tmp5', '12345') gpg2 = get_GPG('{"One": {"user":"******","password":"******","url":"url1","other":"other1"}}', 'secrets_tmp6', '12345') json_msg = '{"example": {"user":"******","password":"******","url":"example","other":"example"}}' cmdc = CommandControler(gpg2) cmdc.add_content(json_msg) json_gpg = json.loads(str(gpg.decrypt_content())) json_gpg2 = json.loads(str(gpg2.decrypt_content())) assert json_gpg == json_gpg2 remove_files(['secrets_tmp5', 'secrets_tmp6'])
class InteractiveCMD: cmdc = '' FILE = '' def __init__(self, gpg): self.cmdc = CommandControler(gpg) self.FILE = gpg.get_file() def id_or_list(self): keys_list = self.cmdc.get_keys() id = raw_input( "Introduce the identifier name or 'list' for list all identifiers: " ).replace(' ', '_') while id.lower() == 'list' or id not in keys_list: if id.lower() == 'list': self.show_keys() id = raw_input( "Introduce the identifier name or 'list' for list all identifiers: " ).replace(' ', '_') if id not in keys_list: id = raw_input( "Introduce a valid identifier name or 'list' for list all identifiers: " ).replace(' ', '_') return id def add_content(self): id = raw_input('Please Introduce an Identifier: ').replace(' ', '_') if (os.path.isfile(self.FILE)): while (id in self.cmdc.get_keys()): id = raw_input('Please Introduce an Identifier: ').replace( ' ', '_') for el in KEYS: KEYS[el] = raw_input("Please introduce a value for '" + str(el.lower()) + "' field, or leave it empty: ") new_json_content = {id: dict(KEYS)} self.cmdc.add_content(json.dumps(new_json_content)) def modify_content(self): id = self.id_or_list() json_content = json.loads(self.cmdc.get_json()) json_content.pop(id, None) new_json = {} print('Leave all elements without value for delete the entry') for e in KEYS: element = raw_input('New ' + str(e) + ': ') new_json[e] = element if all(values == '' for key, values in new_json.items()): jkv = json.dumps(json_content, sort_keys=True) self.cmdc.set_json(jkv) print('Done! Identifier ' + id + ' has been deleted') else: json_content[id] = new_json jkv = json.dumps(json_content, sort_keys=True) self.cmdc.set_json(jkv) print('Done! Identifier ' + id + ' has been modified') def show_keys(self): self.cmdc.show_keys() def update_keys(self): self.cmdc.update_keys() def print_decrypt_content(self): id = self.id_or_list() json_content = json.loads(self.cmdc.get_json()) output = raw_input('Show values? (Y/n): ') while output.lower() != 'y' and output.lower() != 'n' and output.lower( ) != 'yes' and output.lower() != 'no' and output.lower() != '': output = raw_input('Show values? (Y/n): ') if output.lower() == '' or output.lower() == 'y' or output.lower( ) == 'yes': for e in KEYS: print(str(e) + ': ' + str(json_content[id][e])) output = raw_input( 'Copy any elemento to clipboard? (N/element name): ') while output.lower() not in KEYS and output.lower( ) != '' and output.lower() != 'n' and output.lower() != 'no': output = raw_input( "Please choose 'no' for leave. For copy and element 'user', 'password', 'url' or 'other': " ) if output.lower() != '' and output.lower() != 'no' and output.lower( ) != 'n': if platform.system() == 'Darwin': pyperclip.copy(json_content[id][output.lower()]) else: print('Only Darwin platforms') def exit(self): sys.exit(0) def raw_input_menu(self, option, switcher): while True: try: option = int(option) if option not in range(1, switcher): raise ValueError break except: option = raw_input('Please, choose correct option: ') return option def interactive_menu(self): if os.path.isfile(self.FILE): switcher = { 0: lambda: '', 1: self.add_content, 2: self.modify_content, 3: self.print_decrypt_content, 4: self.show_keys, 5: self.update_keys, 6: self.exit } option = raw_input( '\t1: Add Key/Value Pair\n\t2: Modify/Delete Key/Value Pair\n\t3: Decrypt Key/Value Pair\n\t4: Show Keys\n\t5: Update public keys\n\t6: Exit\nChoose: ' ) option = self.raw_input_menu(option, len(switcher)) func = switcher.get(option, lambda: 'nothing') return func() else: print('The file' + self.FILE + 'has not been found, using -i/--interactive argument.') switcher = {0: lambda: '', 1: self.add_content, 2: self.exit} option = raw_input('\t1: Add\n\t2: Exit\nChoose: ') option = self.raw_input_menu(option, len(switcher)) func = switcher.get(option, lambda: 'nothing') return func() def main(self): pass