add support for read() (without arguments) to comm_curl

jflatow · Feb 8, 2010 · 4b31f71 · 4b31f71
1 parent 732119f
commit 4b31f71
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/pydisco/disco/comm_curl.py b/pydisco/disco/comm_curl.py
@@ -131,9 +131,15 @@ def write(self, buf):
  self.body = True
  self.buf += buf
 
- def read(self, bytes):
- while len(self.buf) < bytes and self.cont:
- self.perform()
+ def read(self, bytes = None):
+ if bytes == None:
+ while self.cont:
+ self.perform()
+ bytes = len(self.buf)
+ else:
+ while len(self.buf) < bytes and self.cont:
+ self.perform()
+
  r = self.buf[:bytes]
  self.buf = self.buf[bytes:]
  return r