Commitf3dd9ab

Paul Sokolovsky

committed

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

1 parent94b2afd commitf3dd9abCopy full SHA for f3dd9ab

File tree

+16

-9

lines changed

+16

-9

lines changed

Lines changed: 16 additions & 9 deletions

Original file line number	Diff line number	Diff line change
`@@ -9,13 +9,20 @@ def rmtree(top):`
`9`	`9`	`os.rmdir(path)`
`10`	`10`
`11`	`11`	`defcopyfileobj(src,dest,length=512):`
`12`		`-buf=bytearray(length)`
`13`		`-whileTrue:`
`14`		`-sz=src.readinto(buf)`
`15`		`-ifnotsz:`
`16`		`-break`
`17`		`-ifsz==length:`
	`12`	`+ifhasattr(src,"readinto"):`
	`13`	`+buf=bytearray(length)`
	`14`	`+whileTrue:`
	`15`	`+sz=src.readinto(buf)`
	`16`	`+ifnotsz:`
	`17`	`+break`
	`18`	`+ifsz==length:`
	`19`	`+dest.write(buf)`
	`20`	`+else:`
	`21`	`+b=memoryview(buf)[:sz]`
	`22`	`+dest.write(b)`
	`23`	`+else:`
	`24`	`+whileTrue:`
	`25`	`+buf=src.read(length)`
	`26`	`+ifnotbuf:`
	`27`	`+break`
`18`	`28`	`dest.write(buf)`
`19`		`-else:`
`20`		`-b=memoryview(buf)[:sz]`
`21`		`-dest.write(b)`

Comments

(0)