B A!p\!@sndZddlZddlmZmZddlmZdZGdddZGdd d eeZ Gd d d eZ Gd d d Z dS)zHhandle diagram generation options for class diagram or default diagrams N)PackageDiagram ClassDiagram) LocalsVisitorbuiltinsc@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)DiaDefGeneratorz!handle diagram generation optionscCs |j|_|||_d|_dS)z%common Diagram Handler initializationN)config_set_default_optionslinker classdiagram)selfr handlerr :lib/python3.7/site-packages/pylint/pyreverse/diadefslib.py__init__!szDiaDefGenerator.__init__cCs"|j}|jrd|j|f}|S)zget title for objectsz%s.%s)name module_namesroot)r nodetitler r r get_title(szDiaDefGenerator.get_titlecCs|dkrt|jjS|S)z3activate some options if not explicitly deactivatedN)boolrclasses)r Zoptionr r r _set_option/s zDiaDefGenerator._set_optioncCs~||jj|_||jj}||jj}d\}}|rsr>c@s eZdZdZddZddZdS)DiadefsHandlerzYhandle diagram definitions : get it from user (i.e. xml files) or generate them cCs ||_dS)N)r)r rr r rrszDiadefsHandler.__init__cCs`g}t||}x"|jjD]}||||qW|sFt|||}x|D] }|qLW|S)avGet the diagrams configuration data :param project:The pyreverse project :type project: pyreverse.utils.Project :param linker: The linker :type linker: pyreverse.inspector.Linker(IdGeneratorMixIn, LocalsVisitor) :returns: The list of diagram definitions :rtype: list(:class:`pylint.pyreverse.diagrams.ClassDiagram`) )r>rrappendrFr3r!Zextract_relationships)r rCr Zdiagrams generatorrDZdiagramr r r get_diadefss    zDiadefsHandler.get_diadefsN)r/r0r1r2rrJr r r rrGsrG) r2r*Zpylint.pyreverse.diagramsrrZpylint.pyreverse.utilsrrrr3r>rGr r r rs \8