Mirrors/PySnooper
1
0
Fork 0
mirror of https://github.com/cool-RR/PySnooper.git synced 2026-07-18 00:45:19 +00:00
Code Issues Projects Releases Wiki Activity

Implement IPython code fetching

Browse source
This commit is contained in:
Ram Rachum 2019-04-24 22:28:43 +03:00
parent 61fd0a5262
commit 4d943e9dbb
1 changed files with 18 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

23
pysnooper/tracer.py
View file

@ -15,6 +15,7 @@ import itertools
import six
MAX_VARIABLE_LENGTH = 100
ipython_filename_pattern = re.compile('^<ipython-input-([0-9]+)-.*>$')
def get_shortish_repr(item):
@ -79,11 +80,23 @@ def get_source_from_frame(frame):
if source is not None:
source = source.splitlines()
if source is None:
try:
with open(file_name, 'rb') as fp:
source = fp.read().splitlines()
except (OSError, IOError):
pass
ipython_filename_match = ipython_filename_pattern.match(file_name)
if ipython_filename_match:
entry_number = int(ipython_filename_match.group(1))
try:
import IPython
ipython_shell = IPython.get_ipython()
((_, _, source_chunk),) = ipython_shell.history_manager. \
get_range(0, entry_number, entry_number + 1)
source = source_chunk.splitlines()
except Exception:
pass
else:
try:
with open(file_name, 'rb') as fp:
source = fp.read().splitlines()
except (OSError, IOError):
pass
if source is None:
source = UnavailableSource()