From 151949a2f0264ebeeb8263bc8bd5094021009eb3 Mon Sep 17 00:00:00 2001 From: Binwei Hu Date: Wed, 24 Apr 2019 18:51:25 +0800 Subject: [PATCH] python2 collections --- pysnooper/tracer.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/pysnooper/tracer.py b/pysnooper/tracer.py index 31804a5..d265c5c 100644 --- a/pysnooper/tracer.py +++ b/pysnooper/tracer.py @@ -5,6 +5,10 @@ import types import sys import re import collections +if sys.version.startswith("3"): + from collections import ChainMap +else: + from ConfigParser import _Chainmap as ChainMap import datetime as datetime_module import itertools @@ -25,7 +29,7 @@ def get_local_reprs(frame, variables=()): result[key] = get_shortish_repr(value) except Exception: continue - locals_and_globals = collections.ChainMap(frame.f_locals, frame.f_globals) + locals_and_globals = ChainMap(frame.f_locals, frame.f_globals) for variable in variables: steps = variable.split('.') step_iterator = iter(steps)