gistfile1.py

"""
Generate graph of prerequisites.

http://www.csupomona.edu/~cs/student/undergrad.shtml

"""
import urllib
import re

data = urllib.urlopen('http://www.csupomona.edu/~cs/student/undergrad.shtml').read()

d = {}
for c in re.finditer(r'name="(.*?)"(.*?)</block', data, re.M|re.S):
	d[c.group(1)] = re.findall(r'#(.*?)"', c.group(2))

# generate graph
outf = open('classes.dot', 'w')
print >> outf, 'digraph classes {'
for k, v in d.items():
	if v:
		for preq in v: print >> outf, '%s -> %s;' %(preq, k)
	else:
		print >> outf, '%s;' % k
print >> outf, '}'
outf.close()