#!/bin/python3 import os import sys import logging import json import requests import argparse sys.path.append(os.path.abspath("../")) from subject import main # Identity attributes # {'username' : '', 'full_name' : '', 'email' : '', public_key : '' } logging.basicConfig(format='%(levelname)s\t- %(message)s') logger = logging.getLogger() logger.setLevel(logging.INFO) state = main(sys.argv) #get file #file handle - file(optonal) def getFile(args): parser = argparse.ArgumentParser() parser.add_argument("-k", '--key', nargs=1, help="Path to the key file") parser.add_argument("-r", '--repo', nargs=1, help="Address:Port of the repository") parser.add_argument("-v", '--verbose', help="Increase verbosity", action="store_true") parser.add_argument('filehandle', nargs='?', default=None) parser.add_argument('file', nargs='?', default=None) args = parser.parse_args() if not args.filehandle: logger.error("Need a file handle.") sys.exit(-1) else: if not os.path.isfile(args.filehandle): logger.error("File '" + args.filehandle + "' not found" ) try: file = requests.get(f'http://{state['REP_ADDRESS']}/file', json=json.dumps({"file_handle" : args.filehandle})) file.raise_for_status() except requests.exceptions.RequestException as errex: logger.error("Failed to obtain response from server.") sys.exit(1) file = json.loads(file) # decrypt file content = '' #decrypt(file.encode('utf-8')) if not args.file: sys.stdout.write(content) else: with open(args.file, "wb") as f: f.write(content) if __name__ == '__main__': getFile(sys.argv[1:])