00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023 import sys, getopt, colorsys, imp, md5
00024
00025 class pydepgraphdot:
00026
00027 def main(self,argv):
00028 opts,args = getopt.getopt(argv,'',['mono'])
00029 self.colored = 1
00030 for o,v in opts:
00031 if o=='--mono':
00032 self.colored = 0
00033 self.render()
00034
00035 def fix(self,s):
00036
00037 return s.replace('.','_')
00038
00039 def render(self):
00040 p,t = self.get_data()
00041
00042
00043 for k,d in p.items():
00044 for v in d.keys():
00045 if not p.has_key(v):
00046 p[v] = {}
00047
00048 f = self.get_output_file()
00049
00050 f.write('digraph G {\n')
00051
00052
00053 f.write('ranksep=1.0;\n')
00054 f.write('node [style=filled,fontname=Helvetica,fontsize=10];\n')
00055 allkd = p.items()
00056 allkd.sort()
00057 for k,d in allkd:
00058 tk = t.get(k)
00059 if self.use(k,tk):
00060 allv = d.keys()
00061 allv.sort()
00062 for v in allv:
00063 tv = t.get(v)
00064 if self.use(v,tv) and not self.toocommon(v,tv):
00065 f.write('%s -> %s' % ( self.fix(k),self.fix(v) ) )
00066 self.write_attributes(f,self.edge_attributes(k,v))
00067 f.write(';\n')
00068 f.write(self.fix(k))
00069 self.write_attributes(f,self.node_attributes(k,tk))
00070 f.write(';\n')
00071 f.write('}\n')
00072
00073 def write_attributes(self,f,a):
00074 if a:
00075 f.write(' [')
00076 f.write(','.join(a))
00077 f.write(']')
00078
00079 def node_attributes(self,k,type):
00080 a = []
00081 a.append('label="%s"' % self.label(k))
00082 if self.colored:
00083 a.append('fillcolor="%s"' % self.color(k,type))
00084 else:
00085 a.append('fillcolor=white')
00086 if self.toocommon(k,type):
00087 a.append('peripheries=2')
00088 return a
00089
00090 def edge_attributes(self,k,v):
00091 a = []
00092 weight = self.weight(k,v)
00093 if weight!=1:
00094 a.append('weight=%d' % weight)
00095 length = self.alien(k,v)
00096 if length:
00097 a.append('minlen=%d' % length)
00098 return a
00099
00100 def get_data(self):
00101 t = eval(sys.stdin.read())
00102 return t['depgraph'],t['types']
00103
00104 def get_output_file(self):
00105 return sys.stdout
00106
00107 def use(self,s,type):
00108
00109
00110 if s in ('os','sys','qt','time','__future__','types','re','string'):
00111
00112
00113 return 0
00114 if s.startswith('encodings.'):
00115 return 0
00116 if s=='__main__':
00117 return 1
00118 if self.toocommon(s,type):
00119
00120
00121 return 0
00122 return 1
00123
00124 def toocommon(self,s,type):
00125
00126
00127
00128 if s=='__main__':
00129
00130
00131 return 1
00132 if type==imp.PKG_DIRECTORY:
00133
00134 return 1
00135 return 0
00136
00137 def weight(self,a,b):
00138
00139
00140
00141
00142
00143 if b.split('.')[-1].startswith('_'):
00144
00145
00146
00147 return 4
00148 return 1
00149
00150 def alien(self,a,b):
00151
00152
00153
00154
00155 return 0
00156
00157 def label(self,s):
00158
00159
00160 return '\\.\\n'.join(s.split('.'))
00161
00162 def color(self,s,type):
00163
00164
00165
00166
00167 t = self.normalise_module_name_for_hash_coloring(s,type)
00168 return self.color_from_name(t)
00169
00170 def normalise_module_name_for_hash_coloring(self,s,type):
00171 if type==imp.PKG_DIRECTORY:
00172 return s
00173 else:
00174 i = s.rfind('.')
00175 if i<0:
00176 return ''
00177 else:
00178 return s[:i]
00179
00180 def color_from_name(self,name):
00181 n = md5.md5(name).digest()
00182 hf = float(ord(n[0])+ord(n[1])*0xff)/0xffff
00183 sf = float(ord(n[2]))/0xff
00184 vf = float(ord(n[3]))/0xff
00185 r,g,b = colorsys.hsv_to_rgb(hf, 0.3+0.6*sf, 0.8+0.2*vf)
00186 return '#%02x%02x%02x' % (r*256,g*256,b*256)
00187
00188
00189 def main():
00190 pydepgraphdot().main(sys.argv[1:])
00191
00192 if __name__=='__main__':
00193 main()
00194
00195
00196