elif fmtype == 'method': tmp = name.split(' ') 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: