From 865cd35c9b357e20994f6c6a51f2ae000ba0a3ee Mon Sep 17 00:00:00 2001 From: Claude Paroz Date: Sat, 5 May 2012 14:01:38 +0200 Subject: Made more extensive usage of context managers with open. --- extras/csrf_migration_helper.py | 20 +++++++++----------- 1 file changed, 9 insertions(+), 11 deletions(-) (limited to 'extras') diff --git a/extras/csrf_migration_helper.py b/extras/csrf_migration_helper.py index 067580ca81..62a8fb92af 100755 --- a/extras/csrf_migration_helper.py +++ b/extras/csrf_migration_helper.py @@ -170,14 +170,13 @@ class Template(object): try: return self._content except AttributeError: - fd = open(self.absolute_filename) - try: - content = fd.read().decode(TEMPLATE_ENCODING) - except UnicodeDecodeError as e: - message = '%s in %s' % ( - e[4], self.absolute_filename.encode('UTF-8', 'ignore')) - raise UnicodeDecodeError(*(e.args[:4] + (message,))) - fd.close() + with open(self.absolute_filename) as fd: + try: + content = fd.read().decode(TEMPLATE_ENCODING) + except UnicodeDecodeError as e: + message = '%s in %s' % ( + e[4], self.absolute_filename.encode('UTF-8', 'ignore')) + raise UnicodeDecodeError(*(e.args[:4] + (message,))) self._content = content return content content = property(content) @@ -271,9 +270,8 @@ def get_python_code(paths): for f in filenames: if len([True for e in PYTHON_SOURCE_EXTENSIONS if f.endswith(e)]) > 0: fn = os.path.join(dirpath, f) - fd = open(fn) - content = [l.decode(PYTHON_ENCODING) for l in fd.readlines()] - fd.close() + with open(fn) as fd: + content = [l.decode(PYTHON_ENCODING) for l in fd.readlines()] retval.append((fn, content)) return retval -- cgit v1.2.1