def download(filename): with open(keyfile_name, 'r') as keyfile_handle: keyfile = json.loads(keyfile_handle.read()) metadata = keyfile[filename] data = coding.get_chunks(metadata) strIO = StringIO() strIO.write(data) strIO.seek(0) return send_file(strIO, attachment_filename=filename, as_attachment=True)
args = parser.parse_args() if not args.keyfile: keyfile_name = keyfilelib.create_new_keyfile() else: keyfile_name = args.keyfile with open(keyfile_name, 'r') as keyfile_handle: keyfile = json.loads(keyfile_handle.read()) if args.upload: with open(args.upload, 'r') as f: data = f.read() metadata = coding.send_chunks_get_metadata(data) keyfile[args.upload] = metadata keyfilelib.save(keyfile_name, keyfile) if args.download: if not args.download in keyfile: print "Unknown file. Exiting" sys.exit(1) if not args.output: print "Please supply a destination file for writing" else: file_metadata = keyfile[args.download] data = coding.get_chunks(file_metadata) output_file = open(args.output, "w") output_file.write(data) output_file.close()