[Python-checkins] cpython (3.3): Issue #18755: Allow imp.load_*() loaders to have get_data() called
brett.cannon
python-checkins at python.org
Fri Aug 23 17:52:54 CEST 2013
http://hg.python.org/cpython/rev/ddd610cb65ef
changeset: 85337:ddd610cb65ef
branch: 3.3
parent: 85335:7506a072ebdb
user: Brett Cannon <brett at python.org>
date: Fri Aug 23 11:45:57 2013 -0400
summary:
Issue #18755: Allow imp.load_*() loaders to have get_data() called
multiple times.
files:
Lib/imp.py | 9 +++++++--
Lib/test/test_imp.py | 7 +++++++
2 files changed, 14 insertions(+), 2 deletions(-)
diff --git a/Lib/imp.py b/Lib/imp.py
--- a/Lib/imp.py
+++ b/Lib/imp.py
@@ -90,13 +90,18 @@
def get_data(self, path):
"""Gross hack to contort loader to deal w/ load_*()'s bad API."""
if self.file and path == self.path:
- with self.file:
+ if not self.file.closed:
+ file = self.file
+ else:
+ self.file = file = open(self.path, 'r')
+
+ with file:
# Technically should be returning bytes, but
# SourceLoader.get_code() just passed what is returned to
# compile() which can handle str. And converting to bytes would
# require figuring out the encoding to decode to and
# tokenize.detect_encoding() only accepts bytes.
- return self.file.read()
+ return file.read()
else:
return super().get_data(path)
diff --git a/Lib/test/test_imp.py b/Lib/test/test_imp.py
--- a/Lib/test/test_imp.py
+++ b/Lib/test/test_imp.py
@@ -248,6 +248,13 @@
return
imp.load_module(name, None, *found[1:])
+ def test_multiple_calls_to_get_data(self):
+ # Issue #18755: make sure multiple calls to get_data() can succeed.
+ loader = imp._LoadSourceCompatibility('imp', imp.__file__,
+ open(imp.__file__))
+ loader.get_data(imp.__file__) # File should be closed
+ loader.get_data(imp.__file__) # Will need to create a newly opened file
+
class ReloadTests(unittest.TestCase):
--
Repository URL: http://hg.python.org/cpython
More information about the Python-checkins
mailing list