Update read-n-characters-given-read4.py

kamyu104 · web-flow · commit b63b1b912aaa · 2018-08-04T14:04:52.000+08:00
diff --git a/Python/read-n-characters-given-read4.py b/Python/read-n-characters-given-read4.py
@@ -40,9 +40,9 @@ def read(self, buf, n):
         for i in xrange(n / 4 + 1):
             size = read4(buffer)
             if size:
-                to_read = min(n-read_bytes, size)
-                buf[read_bytes:read_bytes+to_read] = buffer[:to_read]
-                read_bytes += to_read
+                size = min(size, n-read_bytes)
+                buf[read_bytes:read_bytes+size] = buffer[:size]
+                read_bytes += size
             else:
                 break
         return read_bytes
