Rooba
/
boltons
mirror of https://github.com/mahmoud/boltons.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

guard against linecache non-safety

This commit is contained in:
Mahmoud Hashemi 2015-02-08 19:00:09 -08:00
parent 6c2296283b
commit 75616c7b94
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
boltons/tbutils.py
View File

@ -112,11 +112,14 @@ class _DeferredLine(object):
def __str__(self): def __str__(self):
if hasattr(self, '_line'): if hasattr(self, '_line'):
return self._line return self._line
linecache.checkcache(self.filename) try:
line = linecache.getline(self.filename, linecache.checkcache(self.filename)
self.lineno, line = linecache.getline(self.filename,
self.module_globals) self.lineno,
line = line.rstrip() self.module_globals)
line = line.rstrip()
except KeyError:
line = ''
self._line = line self._line = line
return line return line