o DDij@sUddlZddlmZddlmZddlmZmZm Z ddl m Z m Z m Z mZmZmZmZmZddlZde eejfddZeed d d Zeeejed <de eejfd dZeedd d Zeeejed<iZe eeje eejfed<eD]ZeeddeeDee<qdeejdeejfddZ iZ!e eejeejfed<eD]Ze ee!e<qeddGdddZ"ddeDZ#e eeje"fed<de$de%fd d!Z&de$de$fd"d#Z'de$de e$fd$d%Z(de$de%fd&d'Z)de$de%fd(d)Z*de$ddfd*d+Z+eD]Zeep gD]Z,e,j-d,kr+q!e+e,j.q!qee/Z0ee1ee1fed-<e!2D]\ZZ3ej45d.rSqFee3fD] Z6e0e6j78e6j4qWqFdS)/N) defaultdict)Sequence) dataclassfieldsreplace)DictIteratorListMappingrSetTypeUnionreturnccs,ttD]}|ds qtt|VqdS)z Get all base classes that are subclasses of CSTNode but not an actual node itself. This allows us to keep our types sane by refering to the base classes themselves. BaseN)dircst startswithgetattr)namerN/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/codegen/gather.py _get_basess   rcC|jSN__name__)baserrrr)key typeclassesc csfttD]+}|dr|drq|dkrqtt|}z t|tjr&|VWqty0YqwdS)zz Grab all CSTNodes that are not a superclass. Basically, anything that a person might use to generate a tree. __CSTNodeN)rrrendswithr issubclassr" TypeError)rnoderrr _get_nodes"s    r'cCrrr)r&rrrr8rall_libcst_nodes node_to_basescCsg|] }t|tjr|qSr)r$rr".0brrr >sr-r&cCsddt|D}|dS)NcSsg|]}|tvr|qSrr)r*rrrr-Esz3_get_most_generic_base_for_node..rr.)r&exportable_basesrrr_get_most_generic_base_for_nodeBsr0 nodebasesT)frozenc@s2eZdZUdZeed<dZeed<dZeed<dS)UsageFmaybeoptionalsequenceN)r __module__ __qualname__r4bool__annotations__r5r6rrrrr3Os  r3cCsi|]}|tqSr)r3)r+r&rrr Vsr;nodeusestypeobjcCs$zt|tjWStyYdSw)NF)r$r MaybeSentinelr%r=rrr _is_maybeYs  r@cCsz|jWStyYdSwr) __origin__AttributeErrorr?rrr _get_originas  rCcCs z|jWStygYSwr)__args__rBr?rrr _get_argsjs  rEcCst|}|tup |tuSr)rCr ABCSequence)r=originrrr _is_sequencessrHcCs t|tuSr)rCr r?rrr _is_unionxs rIcCst|r?tddt|D}tddt|D}t|D] }|tvr:tt|t|jp.|t|jp4|dt|<qt|qt |r]t|D]}|tvrXtt|ddt|<qGt|qGdSdS)Ncss|]}t|VqdSr)r@r+nrrr ~sz#_calc_node_usage..css|] }t|tdVqdSr) isinstancetyperJrrrrLs)r4r5T)r6) rIanyrEr(rr<r4r5_calc_node_usagerH)r= has_maybehas_noner&rrrrP|s$      rP _metadataimportsr)9inspect collectionsrcollections.abcrrF dataclassesrrrtypingrrr r r r r libcstrr"rsortedr r:r'r(r)r&listreversedgetmror0r1r3r<objectr9r@rCrErHrIrPfieldrrNsetrTstritemsrrrxr7addrrrrs\   (  $  $