c^dZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z m Z mZmZmZmZmZmZejeZde jd efd Z dd ejdededed eeddff dZd efdZd eded eefdZ ddededed eeddffdZded e ded efdZ dS)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|tjr tjSt|tjr tjSt|tjr tjStjS)zGet the lsp type of a node.) isinstancerClassDefr Class FunctionDefFunctionAssignVariable)rs =lib/python3.11/site-packages/rope/contrib/autoimport/parse.py get_type_astrsa$ %%~$((!  $ ##!   Fmodule package_name underlinedprocess_importsc#jK tj|}n'#t$r}t |Yd}~dSd}~wwxYwtj|D]}t |tjr|jD]v} t |tj sJ|s|j ds$t|j t|V`#ttf$rYswxYwt |tjtjfrB|s|j ds$t|jt|V|rt |tjr|jdkr-|jr"||jddurX|jD]k}t |tjr|jr|j} n |j} n|} |s| dst| t|VldS)z.Get all the names from a given file using ast.N_r.)rparse read_bytes SyntaxErrorprintiter_child_nodesrrtargetsr id startswithrrAttributeErrorAssertionErrorrrname ImportFromlevelrsplitnamesaliasasname) rrr r! root_nodeerrorrtargetr/ real_names rget_names_from_filer:(sIf//1122  e $Y//"E"E dCJ ' '! E,   %fch77777!)=)=c)B)B)"I(..'7D s = > > E !5!5c!:!: !I && ED#.!A!A EzQ ; ,$+2C2CC2H2H2K"K  E EdCI..%{.$(K $(I $IEY%9%9#%>%>E%id1C1CDDDDDE"E"Es'&+ AA  A AC**C>=C>ctj|r tjStj|stj|r tjStjS)z Determine the type of an object.)inspectisclassr r isfunction isbuiltinrr)imported_objects rget_type_objectrAYsS''~/**!g.?.P.P!   rpackagec ttr-ttjjjSttr4fdtj jjj DSgS)z(Get all names from a module and package.c2g|]}t|S)combine).0 partial_namerrBs r zget_names..is5    GV\ 2 2   r)r r!) rrlistget_names_from_compiledr/sourcer r r:filepathr!)rrBs``r get_namesrNbs&.))  #GL'.&BS T T   &*%%       3 !, & 6 !!!     IrrLc #RKddg}||vs|dr|sdS|tjtjfvrdS t t |}t j|D]\}}|s|dskt j|s(t j |st j |r/tt ||||t|VdS#t$r!t|dYdSwxYw)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 rA ImportErrorloggerr7)rBrLr bannedrr/values rrKrKusi$- (F&W//44Z fnfo66s7||,, #-f55  KD% !5!5 OE**)%00(// D 7GV_U=S=S    OOOPPPsC;;'D&%D&r/cZt|j|j|j|j|jS)z(Combine information to form a full name.)r r/modnamerL name_type)rBrr/s rrFrFs"  6>7< X XXr)rFF)F)!__doc__r<loggingpathlib importlibrtypingrr rope.baserdefsrr r r r r rr getLogger__name__rWASTrPathrTboolr:rArNrKrFrErrrisJ ######""""""""                      8 $ $sw8! .E.E L.E.E.E .E {D$&' .E.E.E.Ebj7tDz,&& & &&tT4 &&&&RYWYjY YYYYYYYr