Fix decorators

README.rst

@@ -66,6 +66,34 @@ A simple use of the API is::
     with PyCallGraph(output=GraphvizOutput()):
         code_to_profile()
 
+Use decorators for an even more simple use of the API::
+
+    from pycallgraph.decorators import trace
+
+    @trace("path/to/output.png")
+    def main():
+        code_to_profile()
+
+    main()
+
+Or decorate a specific function inside your code you want to profile::
+
+    from pycallgraph.decorators import trace
+
+    @trace("path/to/output.png")
+    def function_1():
+        do_stuff
+
+    def function_2():
+        do_stuff
+
+    def code_to_profile():
+        function_1()
+        function_2()
+
+    code_to_profile()
+
+
 Documentation
 =============
 

pycallgraph/__init__.py

@@ -14,7 +14,10 @@
 
 from .pycallgraph import PyCallGraph
 from .exceptions import PyCallGraphException
-from . import decorators
+try:
+    from . import decorators
+except Exception:
+    import decorators
 from .config import Config
 from .globbing_filter import GlobbingFilter
 from .grouper import Grouper

pycallgraph/decorators.py

@@ -1,13 +1,18 @@
 import functools
 
 from .pycallgraph import PyCallGraph
+from .output import GraphvizOutput
 
 
 def trace(output=None, config=None):
     def inner(func):
         @functools.wraps(func)
         def exec_func(*args, **kw_args):
-            with(PyCallGraph(output, config)):
+
+            graphviz = GraphvizOutput()
+            graphviz.output_file = output
+
+            with(PyCallGraph(output=graphviz, config=config)):
                 return func(*args, **kw_args)
 
         return exec_func