From e528ca3f26bc1a35eae98a64716be8bf0dcf2b00 Mon Sep 17 00:00:00 2001 From: JustAnotherArchivist Date: Sat, 18 May 2019 01:08:49 +0000 Subject: [PATCH] Dump locals only for snscrape modules (closes #39) --- snscrape/cli.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/snscrape/cli.py b/snscrape/cli.py index 80286f1..e817ca8 100644 --- a/snscrape/cli.py +++ b/snscrape/cli.py @@ -19,8 +19,10 @@ def _dump_locals_on_exception(): trace = inspect.trace() if len(trace) >= 3: with tempfile.NamedTemporaryFile('w', prefix = 'snscrape_locals_', delete = False) as fp: - for i in range(2, len(trace)): - frameRecord = trace[i] + for frameRecord in trace[2:]: + module = inspect.getmodule(frameRecord[0]) + if not module.__name__.startswith('snscrape.') and module.__name__ != 'snscrape': + continue locals_ = frameRecord[0].f_locals fp.write(f'Locals from file "{frameRecord.filename}", line {frameRecord.lineno}, in {frameRecord.function}:\n') fp.write(repr(locals_))