From a4975a911d17d8baa96570794fa6db19c0676a2a Mon Sep 17 00:00:00 2001 From: Brett Cannon Date: Fri, 23 Aug 2013 11:45:57 -0400 Subject: Issue #18755: Allow imp.load_*() loaders to have get_data() called multiple times. --- Lib/imp.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'Lib/imp.py') diff --git a/Lib/imp.py b/Lib/imp.py index 30c343f62e..408838387b 100644 --- a/Lib/imp.py +++ b/Lib/imp.py @@ -90,13 +90,18 @@ class _HackedGetData: 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) -- cgit v1.2.1