o b#@s|dZddlZddlmZmZddlmZGdddZGdddeZGd d d eZ Gd d d eeZ Gd dde Z dS)zdiagram objects N) is_interface FilterMixIn)decorated_with_propertyc@seZdZdZdS)Figurezbase class for counter handlingN)__name__ __module__ __qualname____doc__r r X/var/www/secure340b-portal/env/lib/python3.10/site-packages/pylint/pyreverse/diagrams.pyrsrc@seZdZdZdddZdS) Relationshipz=a relation ship from an object in the diagram to another NcCs&t|||_||_||_||_dSN)r__init__ from_object to_objecttypename)selfrr relation_typerr r r rs  zRelationship.__init__r rrrr rr r r r r r c@seZdZdZdddZdS) DiagramEntityzAa diagram object, i.e. a label associated to an astroid node No nameNcCst|||_||_dSr )rrtitlenode)rrrr r r r's  zDiagramEntity.__init__)rNrr r r r r#rrc@seZdZdZdZddZddZd dd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZdS)! ClassDiagramz main class diagram handling classcCs8t||t|||_g|_i|_i|_g|_dSr )rrrrobjects relationships_nodesdepends)rrmoder r r r3s   zClassDiagram.__init__cCst|j|ddddS)Nr cSs|jj|jjfSr )rZfig_idr)xr r r @sz0ClassDiagram.get_relationships..key)sortedrget)rroler r r get_relationships<s zClassDiagram.get_relationshipsNcCs&t||||}|j|g|dS)zcreate a relation ship N)r r setdefaultappend)rrrrrrelr r r add_relationshipCszClassDiagram.add_relationshipcCs.|j|dD] }|j|ur|Sqt|)z'return a relation ship or None r )rr'rKeyError)rrrr,r r r get_relationshipIs  zClassDiagram.get_relationshipcCs~g}dd|D}t|jt|j|D]\}}||s%q||}|r5d|d|f}||qt|S)z3return visible attributes, possibly with class namecSs,g|]\}}t|tjrt|r||fqSr ) isinstanceastroid FunctionDefr).0nmr r r Ts z*ClassDiagram.get_attrs..z%s : %sz, ) itemslistinstance_attrs_type locals_type show_attr class_namesjoinr+r&)rrattrs properties node_nameZassociated_nodesnamesr r r get_attrsQs"     zClassDiagram.get_attrscs&fdd|D}t|dddS)zreturn visible methodscs0g|]}t|tjrt|s|jr|qSr )r0r1r2rr;r)r3r5rr r r6hs  z,ClassDiagram.get_methods..cSs|jSr )r)r4r r r r#osz*ClassDiagram.get_methods..r$)valuesr&)rrmethodsr rCr get_methodsfs zClassDiagram.get_methodscCs2||jvsJt||}||j|<|j|dS)z create a diagram object N)rrrr+)rrrentr r r add_objectqs  zClassDiagram.add_objectcCs^g}|D](}t|tjr|j}t|tjr,t|dr,||s,|j|vr,|j}||q|S)z'return class names if needed in diagramr) r0r1Instance_proxiedClassDefhasattrhas_noderr+)rnodesrArr@r r r r<ys    zClassDiagram.class_namescCs |jS)z,return the list of underlying nodes )rkeysrCr r r rN zClassDiagram.nodescCs ||jvS)zAreturn true if the given node is included in the diagram rrrr r r rMrPzClassDiagram.has_nodecCs |j|S)z1return the diagram object mapped to node rQrRr r r object_from_noderPzClassDiagram.object_from_nodecCdd|jDS)z%return all class nodes in the diagramcSg|] }t|jtjr|qSr )r0rr1rKr3or r r r6z(ClassDiagram.classes..rrCr r r classeszClassDiagram.classescC*|D] }|jj|kr|Sqt|)z@return a class by its name, raise KeyError if not found )rZrrr.)rrklassr r r classe  zClassDiagram.classec Cs>|D]}|j}|||_|||_t|rd|_nd|_|jddD]}z| |}| ||dWq&t y?Yq&w|j D]}z| |}| ||dWqCt y\YqCwt |jt |jD]0\}}|D])} | tjuryqqt| tjr| j} z| | } | | |d|Wqqt yYqqwqkqdS) rFrErshapeZ ancestorsrSr-r.rar8r9r7r:r1Z Uninferabler0rIrJ) robjrZpar_nodeZpar_objZ impl_nodeZimpl_objrrDvalueZassociated_objr r r extract_relationshipssL            z"ClassDiagram.extract_relationshipsr )rrrr TYPErr)r-r/rBrFrHr<rNrMrSrZr^rer r r r r-s"   rc@s<eZdZdZdZddZddZddZd d Zd d Z d S)PackageDiagramzpackage diagram handling packagecCrT)z&return all module nodes in the diagramcSrUr )r0rr1ModulerVr r r r6rXz*PackageDiagram.modules..rYrCr r r modulesr[zPackageDiagram.modulescCr\)zAreturn a module by its name, raise KeyError if not found )rjrrr.)rrmodr r r moduler_zPackageDiagram.modulecCst|D]1}|jj}||kr|S|j}|d||fkr#|S|d|ddd|fkr5|Sqt|)zlreturn a module by its name, looking also for relative imports; raise KeyError if not found z%s.%s.r)rjrrrootrsplitr.)rrrrkmod_namerhr r r get_modules  zPackageDiagram.get_modulecCs6|j}||}||jjvr|jj|dSdS)z1add dependencies created by from-imports N)rorrlrr r+)rrZ from_modulerqrcr r r add_from_depends   zPackageDiagram.add_from_dependc Cst||D]}z||j}|||dWq ty%Yq w|D]&}d|_ |jj D]}z | ||j}Wn tyGYq3w|||dq3q*dS)r`Z ownershiprhr N) rrerZrSrror-r.rjrbr rr)rrcrkdep_namedepr r r res&      z$PackageDiagram.extract_relationshipsN) rrrr rfrjrlrrrsrer r r r rgs rg) r r1Zpylint.pyreverse.utilsrrZpylint.checkers.utilsrrr rrrgr r r r s    "