def format(request, data, res_name): """ """ try: p = Parser() p.read(data) result = p.getSEED() except: return data # set file name request.setHeader('Content-Disposition', 'attachment; filename=%s.dataless' \ % res_name) # set content type request.setHeader('content-type', 'application/octet-stream') return result
def format(request, data, res_name): """ """ channel = str(request.args0.get('channel', '')).upper() try: p = Parser() p.read(data) resp_list = p.getRESP() # Create a ZIP archive. zip_fh = StringIO() if channel == '': zip_file = zipfile.ZipFile(zip_fh, "w") for response in resp_list: response[1].seek(0, 0) zip_file.writestr(response[0], response[1].read()) zip_file.close() zip_fh.seek(0) data = zip_fh.read() res_name += os.extsep + "zip" else: for response in resp_list: if response[0][-3:] != channel: continue response[1].seek(0, 0) data = response[1].read() res_name = response[0] break except: return data if channel == '': # set content type request.setHeader('content-type', 'application/zip') # set file name request.setHeader('Content-Disposition', 'attachment; filename=%s' % res_name) return data