class CurlResponse(object): def __init__(self): self.headers = {} self.buffer = StringIO() def getheader(self, header, default=None): return self.headers.get(header.lower(), default) def getheaders(self): return self.headers.items() def header_function(self, header): if ':' in header: k, v = header.split(':', 1) self.headers[k.lower().strip()] = v.strip() def read(self): return self.buffer.getvalue()
def reduce_calls(self): out = StringIO() self.job.profile_stats(stream=out).print_stats('worker', 'reduce') for line in out.getvalue().splitlines(): if 'worker.py' in line: return int(line.strip().split()[0])
def encode_netstring_str(d): msg = StringIO() for k, v in d: msg.write("{0} {1} {2} {3}\n" .format(len(k), str(k), len(v), str(v))) return msg.getvalue()
def encode_netstring_str(d): msg = StringIO() for k, v in d: msg.write("{0} {1} {2} {3}\n".format(len(k), str(k), len(v), str(v))) return msg.getvalue()