Attribute support

Author: samuelcolvin

README.rst

@@ -33,9 +33,13 @@ modify ``/usr/lib/python3.6/sitecustomize.py`` making ``debug`` available in any
 
 .. code:: python
 
-   from devtools import debug
-   __builtins__['debug'] = debug
-
+   # add devtools debug to builtins
+   try:
+       from devtools import debug
+   except ImportError:
+       pass
+   else:
+       __builtins__['debug'] = debug
 
 .. |BuildStatus| image:: https://travis-ci.org/samuelcolvin/python-devtools.svg?branch=master
    :target: https://travis-ci.org/samuelcolvin/python-devtools

devtools/debug.py

@@ -67,7 +67,7 @@ class Debug:
     # 50 lines should be enough to make sure we always get the entire function definition
     frame_context_length = 50
     complex_nodes = (
-        ast.Call,
+        ast.Call, ast.Attribute,
         ast.IfExp, ast.BoolOp, ast.BinOp, ast.Compare,
         ast.DictComp, ast.ListComp, ast.SetComp, ast.GeneratorExp
     )

tests/test_main.py

@@ -103,6 +103,18 @@ def test_simple_vars():
     )
 
 
+def test_attributes():
+    class Foo:
+        x = 1
+
+    class Bar:
+        y = Foo()
+
+    b = Bar()
+    v = debug.format(b.y.x)
+    assert 'test_attributes: b.y.x = 1 (int)' in str(v)
+
+
 def test_eval():
     with pytest.warns(RuntimeWarning):
         v = eval('debug.format(1)')