class MessageBuilder(object):
def __init__(self):
self.envelope = Envelope()
self.terminator = self.envelope.delimiter() * 2
# Note that order is reversed so pop() gets first.
self.writers = [self.write_body, self.write_head]
super(MessageBuilder, self).__init__()
def write(self, bytes):
writemethod = self.writers.pop()
return writemethod(bytes)
def get_terminator(self):
return self.terminator
def write_head(self, head):
self.envelope.parse_head(head)
self.terminator = int(self.envelope.content_length())
def write_body(self, body):
self.envelope.parse_body(body)
self.terminator = 0
def message(self):
return self.envelope.getvalue()
def iscomplete(self):
return not self.writers
class MessageBuilder(object):
def __init__(self):
self.envelope = Envelope()
self.terminator = self.envelope.delimiter() * 2
# Note that order is reversed so pop() gets first.
self.writers = [self.write_body, self.write_head]
super(MessageBuilder, self).__init__()
def write(self, bytes):
writemethod = self.writers.pop()
return writemethod(bytes)
def get_terminator(self):
return self.terminator
def write_head(self, head):
self.envelope.parse_head(head)
self.terminator = int(self.envelope.content_length())
def write_body(self, body):
self.envelope.parse_body(body)
self.terminator = 0
def message(self):
return self.envelope.getvalue()
def iscomplete(self):
return not self.writers
