name = tmp[0]
params = tmp[1:]
current_class.add_method(name,*params)
elif fmtype == 'TODO':
current_class.add_todo(name)
diagram.add(current_class)
diagram.link()
return diagram
if __name__ == "__main__":
import os
parg = PPParse()
parg.add_flag('-p','use color','color',lambda arg : True,default=True)
parg.add_flag('-u','do not use color','color',lambda arg : False)
parg.add_flag('-m','show methods','methods',lambda arg : True,default=True)
parg.add_flag('-n','do not show methods','methods',lambda arg : False)
parg.add_flag('-f=','set the filename','filename',lambda arg : arg[3:],default='docs/diagram.ppmd')
parg.add_flag('-c=','set the module','module',lambda arg : arg[3:],default=None)
parg.add_flag('-e','echo instead of parse','echo',lambda arg : True,default=False)
parg.add_flag('-d=','set the depth','depth',lambda arg : int(arg[3:]),default=None)
parg.add_flag('-a','show all modules','depth',lambda arg : None)
parg.parse()
parg.color = 'color' in os.environ['TERM'] and parg.color
diagram = read_diagram(parg.filename,parg.color)
if parg.echo:
(diagram.echo())
else:
diagram.draw(module=parg.module,depth=parg.depth,methods=parg.methods)
