ÆebdZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z m Z mZmZmZmZmZmZej(eZde j.d efd Z dd ej2d ed eded eeddff dZd efdZd eded eefdZ ddeded ed eeddffdZded e ded efdZ y)zz Functions to find importable names. Can extract names from source code of a python file, .so object, or builtin module. N) import_module) GeneratorList)ast)ModuleCompiled ModuleFile ModuleInfoNameNameTypePackage PartialNameSourcenodereturnct|tjrtjSt|tj rtj St|tjrtjStjS)zGet the lsp type of a node.) isinstancerClassDefr Class FunctionDefFunctionAssignVariable)rs ?/usr/lib/python3/dist-packages/rope/contrib/autoimport/parse.py get_type_astrsZ$ %~~$(   $ #     module package_name underlinedprocess_importsc#K tj|j}tj |D]}t |tjrm|jD]]} t |tjsJ|s|jjds!t|jt|_t |tj tj"fr@|s|j$jdrt|j$t||st |tj&s|j(dk(r%|j*r ||j*j-ddurS|j.D]o}t |tj0r&|j2r |j2} n|j$} n|} |s| jdrYt| t|qy#t$r}t |Yd}~yd}~wwxYw#ttf$rYwxYww)z.Get all the names from a given file using ast.N_r.)rparse read_bytes SyntaxErrorprintiter_child_nodesrrtargetsr id startswithrrAttributeErrorAssertionErrorrrname ImportFromlevelrsplitnamesaliasasname) rrrr root_nodeerrorrtargetr. real_names rget_names_from_filer9(sIIf//12 $$Y/"E dCJJ ',,  %fchh777!)=)=c)B)"II(. s = >!5!5c!:!II &D#..!AzzQ;;,$++2C2CC2H2K"K EdCII.{{$(KK $(II $IY%9%9#%>%id1CDD E3"E  e '7smI#HAI*AH2A I$I4IB#I4I H/ H*%I*H//I2IIIIctj|rtjStj|stj |rtj StjS)z Determine the type of an object.)inspectisclassr r isfunction isbuiltinrr)imported_objects rget_type_objectr@YsK'~~/*g.?.?.P     rpackagecXt|tr4tt|j|j |j St|trPt|j|j|j |jDcgc]}t|||c}SgScc}w)z(Get all names from a module and package.)rr ) rrlistget_names_from_compiledr.sourcerr r9filepathr combine)rrA partial_names r get_namesrIbs&.) #GLL'..&BSBS T  &*%!4 !,, & 6 6 !  GV\ 2   I sB'rEc #*Kddg}||vs|jdr|sy|tjtjfvry t t |}t j|D]|\}}|s|jdrt j|s+t j|st j|sZtt ||||t|~y#t$rtj|dYywxYww)z Get the names from a compiled module. Instead of using ast, it imports the module. Parameters ---------- package : str package to import. Must be in sys.path underlined : bool include underlined names builtins python_crunr"Nz. could not be imported for autoimport analysis)r+rBUILTINSTANDARDrstrr; getmembersr<r=r>r r@ ImportErrorloggerr6)rArErbannedrr.values rrDrDus$- (F&W//4Z fnnfoo66s7|, #--f5 KD%!5OOE*))%0((/D 7GV_U=S   y NOPs6ADC,.D?D&D,!D DDDr.ct|j|j|j|j|jS)z(Combine information to form a full name.)r r.modnamerE name_type)rArr.s rrGrGs)  6>>7<< XXr)FF)F)!__doc__r;loggingpathlib importlibrtypingrr rope.baserdefsrr r r r r rr getLogger__name__rRASTrPathrOboolr9r@rIrDrGrrrfs8 #"      8 $sww8! .E LL.E.E.E .E {D$&' .Ebj7tDz,& & &&tT4 &RYWYjY YYr