create_graph.py

#!/usr/bin/env python
'''
Create and format a dependency graph for the project.
'''
from subprocess import Popen, PIPE

# 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 'PyQt5' 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('"PyQt5" [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)