create_graph.py

#!/usr/bin/env python
"""
Create and format a dependency graph for the project.
"""

from subprocess import PIPE, Popen

# create the graph file
result_1 = Popen(["sfood", "quicknxs"], stdout=PIPE).communicate()[0]
dot = Popen(["sfood-graph"], stdout=PIPE, stdin=PIPE).communicate(result_1)[0]

# process result dot file data

out_dot = []
ext_packages = []
first_add = True

for line in dot.splitlines():
    if (
        line.startswith('"quicknxs"')
        or "version" in line
        or '"logging"' in line
        or "sip" in line
        or "figureoptions" in line
        or "types" in line
        or "nxutils.so" in line
        or "output_templates" in line
        or "os" in line
        or "sys" in line
    ):
        continue
    elif line.startswith('"quicknxs/'):
        line = line.replace("style=filled", 'style=filled, color="#00aa00" shape=box, fontsize=24').replace(
            "quicknxs/", ""
        )
    if "PyQt4" in line or "email" in line or "scipy" in line or "matplotlib/" in line or "IPython" in line:
        line = line.split("/")[0] + '";'
        if first_add:
            first_add = False
            out_dot.append('"PyQt4" [style=filled, color="#aa0000", fontsize=36];')
            out_dot.append('"scipy" [style=filled, color="#aaaaff", fontsize=30];')
            out_dot.append('"numpy" [style=filled, color="#0000aa", fontsize=32, fontcolor=white];')
            out_dot.append('"matplotlib" [style=filled, color="#aa00aa", fontsize=32];')
            out_dot.append('"IPython" [style=filled, color="#aaaa00", fontsize=24];')
            out_dot.append('"h5py" [style=filled, color="#aaffaa", fontsize=32];')
            for sys_module in [
                "glob",
                "subprocess",
                "multiprocessing",
                "traceback",
                "zipfile",
                "StringIO",
                "cPickle",
                "atexit",
                "tempfile",
                "getpass",
                "time",
                "copy",
                "inspect",
                "email",
                "smtplib",
                "ConfigParser",
                "errno",
            ]:
                out_dot.append('"%s" [shape=hexagon, fontsize=18];' % sys_module)
    line = line.replace("pylab", "matplotlib").replace("cStringIO", "StringIO")
    out_dot.append(line.replace(".py", ""))

out_dot = "\n".join(out_dot)

Popen(["dot", "-Tps", "-o", "import_graph.ps"], stdout=PIPE, stdin=PIPE).communicate(out_dot)