Exemple #1
0
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()
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()
Exemple #3
0
 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])
Exemple #4
0
 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])
Exemple #5
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()