Re-adding pycallgraph control script

README.md

@@ -56,4 +56,17 @@ Features
 -   Modules can be visually grouped together.
 -   Easily extendable to create your own output formats.
 
+Command-line interface
+----------------------
 
+The following commands shall collect the call graph into the specified output file in .dot format, which can then be visualized as an SVG image:
+
+```
+pycallgraph graphviz --output-format=dot --output-file=script.dot -- ./script.py
+dot -Tsvg script.dot > script.svg
+```
+
+Development
+-----------
+
+Clone this repository and issue a `poetry install` command, in order to install the current version of package.

pycallgraph/pycallgraph.py

@@ -98,3 +98,24 @@ def prepare_output(self, output):
         output.sanity_check()
         output.set_processor(self.tracer.processor)
         output.reset()
+
+def main():
+    import os, sys
+    # Inject the current working directory so modules can be imported.
+    sys.path.insert(0, os.getcwd())
+
+    import pycallgraph as __pycallgraph
+
+    __config = __pycallgraph.Config()
+    __config.parse_args()
+    __config.strip_argv()
+
+    globals()['__file__'] = __config.command
+
+    __file_content = open(__config.command).read()
+
+    with __pycallgraph.PyCallGraph(config=__config):
+        exec(__file_content)
+
+if __name__ == '__main__':
+    main()

pycallgraph/test/test_script.py

@@ -2,7 +2,7 @@
 
 
 def execute(arguments):
-    command = "PYTHONPATH=. scripts/pycallgraph " + arguments
+    command = "pycallgraph " + arguments
     return subprocess.check_output(command, shell=True).decode("utf-8")
 
 

pyproject.toml

@@ -20,6 +20,9 @@ name = 'default'
 url = 'https://pypi.python.org/simple'
 secondary = true
 
+[tool.poetry.scripts]
+pycallgraph = "pycallgraph.pycallgraph:main"
+
 [tool.poetry.dependencies]
 python = "^3.6.1"
 memory_profiler="^0.58.0"
@@ -37,4 +40,4 @@ sphinx = "^3.4.1"
 
 [build-system]
 requires = ["poetry>=0.12"]
-build-backend = "poetry.masonry.api"
+build-backend = "poetry.masonry.api"