diff options
author | Claude Paroz <claude@2xlibre.net> | 2012-05-05 14:01:38 +0200 |
---|---|---|
committer | Claude Paroz <claude@2xlibre.net> | 2012-05-05 14:06:36 +0200 |
commit | 865cd35c9b357e20994f6c6a51f2ae000ba0a3ee (patch) | |
tree | 9db42053673c856a2527d5573d7f8fd682334182 /extras | |
parent | ec5423df05dedeee6651c36dd4d90fec0d8cca7c (diff) | |
download | django-865cd35c9b357e20994f6c6a51f2ae000ba0a3ee.tar.gz |
Made more extensive usage of context managers with open.
Diffstat (limited to 'extras')
-rwxr-xr-x | extras/csrf_migration_helper.py | 20 |
1 files changed, 9 insertions, 11 deletions
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 |