mirror of https://github.com/cool-RR/PySnooper.git
Use .items() to get rid of a call to getattr
This commit is contained in:
parent
b4b425c652
commit
7222d78a83
|
@ -233,8 +233,7 @@ class Tracer:
|
||||||
return self._wrap_function(function_or_class)
|
return self._wrap_function(function_or_class)
|
||||||
|
|
||||||
def _wrap_class(self, cls):
|
def _wrap_class(self, cls):
|
||||||
for attr_name in cls.__dict__:
|
for attr_name, attr in cls.__dict__.items():
|
||||||
attr = getattr(cls, attr_name)
|
|
||||||
if inspect.isfunction(attr):
|
if inspect.isfunction(attr):
|
||||||
setattr(cls, attr_name, self._wrap_function(attr))
|
setattr(cls, attr_name, self._wrap_function(attr))
|
||||||
return cls
|
return cls
|
||||||
|
|
Loading…
Reference in New Issue