shutil: copyfileobj(): Support file objects without readinto() method.

pfalcon · pfalcon · commit f3dd9abcda12 · 2015-02-04T02:50:20.000+02:00
diff --git a/shutil/shutil.py b/shutil/shutil.py
@@ -9,13 +9,20 @@ def rmtree(top):
         os.rmdir(path)
 
 def copyfileobj(src, dest, length=512):
-    buf = bytearray(length)
-    while True:
-        sz = src.readinto(buf)
-        if not sz:
-            break
-        if sz == length:
+    if hasattr(src, "readinto"):
+        buf = bytearray(length)
+        while True:
+            sz = src.readinto(buf)
+            if not sz:
+                break
+            if sz == length:
+                dest.write(buf)
+            else:
+                b = memoryview(buf)[:sz]
+                dest.write(b)
+    else:
+        while True:
+            buf = src.read(length)
+            if not buf:
+                break
             dest.write(buf)
-        else:
-            b = memoryview(buf)[:sz]
-            dest.write(b)
