close files after comparing them

This commit is contained in:
Benjamin Peterson 2009-03-19 03:04:31 +00:00
parent a61ae6922f
commit fcfa7ead4f
1 changed files with 9 additions and 9 deletions

View File

@ -11,6 +11,7 @@ Functions:
import os
import stat
import contextlib
from itertools import ifilter, ifilterfalse, imap, izip
__all__ = ["cmp","dircmp","cmpfiles"]
@ -62,15 +63,14 @@ def _sig(st):
def _do_cmp(f1, f2):
bufsize = BUFSIZE
fp1 = open(f1, 'rb')
fp2 = open(f2, 'rb')
while True:
b1 = fp1.read(bufsize)
b2 = fp2.read(bufsize)
if b1 != b2:
return False
if not b1:
return True
with contextlib.nested(open(f1, 'rb'), open(f2, 'rb')) as (fp1, fp2):
while True:
b1 = fp1.read(bufsize)
b2 = fp2.read(bufsize)
if b1 != b2:
return False
if not b1:
return True
# Directory comparison class.
#