ue3UddlZddlZddlmZmZmZmZmZmZm Z ddl Z ddl m Z ddl mZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZmZd d lmZej@e!Z"dZ#de#zZ$dZ%dZ&GddZ'edee(ee(ee(effffdZ)dejTdejVde,fdZ-dedejTdejVde,fdZ.dejVdejTde,fdZ/dejVdejTde,fdZ0dedejVfdZ1 de'Z?e'e@d;<y)=N)AnyDict GeneratorListOptionalSetUnion)Script)tree) NodeOrLeaf)Resource) SearchResult) AutoImport)hookimpl)Config)Document Workspace)PylspTaskHandle ic peZdZdZdZ d dededeee dee fdZ dded e d eee fd Zd Zy)AutoimportCachezHandles the cache creation.cd|_yN)threadselfs ?/usr/lib/python3/dist-packages/pylsp/plugins/rope_autoimport.py__init__zAutoimportCache.__init__ s  Nconfig workspacefiles single_threadc|jry|jdjdd}|jjdi}|j ||}|dn|Dcgc]}|j |c}} |r|j ||| ytj|j ||| f|_ |jjycc}w)Nrope_autoimportmemoryFrope)targetargs) is_blockedplugin_settingsgetsettings_rope_autoimport_rope_resource _reload_cache threadingThreadrstart) rr"r#r$r%r( rope_config autoimportdocument resourcess r reload_cachezAutoimportCache.reload_cache#s ??  --.?@DDXuUoo'++FB7 // VD } GLM8())+6M     y*i @$**))J 0RDK KK   Ns-C"r7r9cdt|}|j|||j|y)N) task_handler9)r<)rgenerate_cachegenerate_modules_cache)rr#r7r9r<s rr2zAutoimportCache._reload_cache@s2 &i0 !!kY!O))k)Br!cR|jxr|jjSr)ris_alivers rr,zAutoimportCache.is_blockedJs{{5t{{3355r!)NFr)__name__ __module__ __qualname____doc__r rrrrrboolr:rr r2r,r!rrrs%+/(-    X'   ~  B/3 CCCDN+ C6r!rreturnc ddddddiddidiiS)Npluginsr'FenabledT)rJr( completions code_actionsrFrFr!rpylsp_settingsrMNs:   t t!     r!expr word_nodec|syt|jdk(ry|jd}t|tjrd|j vry||k(ryt|jdkDrt d|jDryt|tjtjfr t||St|||S)z Check if we should insert the word_node on the given expr. Works for both correct and incorrect code. This is because the user is often working on the code as they write it. FrT#rc3|K|]4}|jdk(xrd|jvxs|jdk(6yw)operator.trailerN)typevalue).0nodes r z!_should_insert..ss@  II # 9tzz(9 STYY)=S S s:<) lenchildren isinstancer EndMarkerprefixanyPythonErrorNode PythonNode_should_insert_handle_first_child)rNrO first_childs rrcrcas  4==Q--"K+t~~. +$$ $i 4==A     + 4 4dooFGk955 {D) <F)r\_handle_argumentr]r rtrWrus rrorosVI  D) , i /$&9$  r!rYct|tjrP|jdk(r|jd|k(ry|jdk(r|jD]}t ||syy)NtfpdefT parametersF)r]r rbrVr\rx)rYrO parameters rrxrxsa$( 99 }}Q9, 99 $!]] #Iy9  r! suggestionsdoc_uriwordr7r8featurec#K|D]}|j|jdz }|dd}||d} | |jdzd} t|j|j|j|} | t kDrs|dk(r?|j|j t| d|it|j| gd |d k(r$|jd d || giid t| idtd|yw)Nrr)line character)r5end )rangenewTextrKr)labelkindsortTextdatadetailadditionalTextEditsrLquickfixchangesr)titlereditrzUnknown feature: ) find_insertion_linesourceimport_statement _get_scorename _score_maxitemkind _sort_import _document ValueError) r~rrr7r8r suggestion insert_liner5 edit_rangerscores r_process_statementsrs!"<  44X__EI $15$U3 # 0K0Kd0RS   z::JOOT  :   m ##"++(/"G,#J$?$?@(,v   &#44""Wtf$56$\%%89  0 :; ;=z#pylsp_completions..s i 3r!keyN)r-r.cacher,linesparsoparseget_leaf_for_positionrcrWrrr/pathr jedi_scriptr0list search_fullsortedrurir[MAX_RESULTS_COMPLETIONS) r"r#r8positionrrrNrOrr6r7r~resultss rpylsp_completionsrsc  " "#4 5 ]B  Y       >>(6* +D ;;t D**Ax /D+EFI $ * ??DII078// />BB62NK+yt40M++K8Jz--d--PQK tZ=  4 G  7|--223 Nr!rc d|zS)Nz# Auto-Import rF)rs rrrs #3 33r!rfull_statementsuggested_namectd}t||z }t|t|z dz}d|z}||z|zS)Nrgr{)r[)rrr desired_name import_lengthfull_statement_scoresuggested_name_score source_scores rrrsQMM~.>/#l2CCI;L "6 6 EEr!rctt|tdz d}dt|j t dzS)Nrrz[z0)maxminrstrrjust _score_pow)rs rrrs8 EJ!+,a 0E #e*"":s3 33r!c|dd}tj|j|djd|ddzfjS)Nrr5rrr)rrrrrW)r8 diagnosticr5s rget_name_or_moduler$sN w  (E HNN5=12  5#5#9: ; r!rcontextc |jdjdijddrtjrgStj d|d|d|g}|jdgD] }d|jd d j vr't||}tj d ||j|j jd i}|j|} t| j|} tj d| tt| |j|| |dd} t!| t"kDr | dt"} |j%|  |S)a Provide code actions through rope. Parameters ---------- config : pylsp.config.config.Config Current config. workspace : pylsp.workspace.Workspace Current workspace. document : pylsp.workspace.Document Document to apply code actions on. range : Dict Range argument given by pylsp. Not used here. context : Dict CodeActionContext given as dict. Returns ------- List of dicts containing the code actions. r'rLrJTztextDocument/codeAction:  diagnosticszundefined namemessagerrr)zautoimport: suggestions: %sc|ddS)NrrrFrs rrz$pylsp_code_actions..es)F"3J"?r!rN)r-r.rr,rrlowerrr/rr0rrrrrr[MAX_RESULTS_CODE_ACTIONSextend) r"r#r8rrrLrrr6r7r~rs rpylsp_code_actionsr-sw:  " "#4 5 ^R Y       II)(1UG1WIFGLkk-4% :>>)R#@#F#F#H H !(J7 4TF;<ooHMMoBFFvrR // < :11$78  /=    @   w<2 2778GG$1%4 r!c0tj||yzPInitialize AutoImport. Generates the cache for local and global items. Nrr:r"r#s rpylsp_initializero  vy)r!c0tj||yrrrs rpylsp_document_did_openrxrr!c4tj|||gy)z/Update the names associated with this document.Nr)r"r#r8s rpylsp_document_did_savers vy8*5r!c|jdjddrtj||ytj dy)z Initialize autoimport if it has been enabled through a workspace/didChangeConfiguration message from the frontend. Generates the cache for local and global items. r'rJFz"autoimport: Skipping cache reload.N)r-r.rr:rrrs r%pylsp_workspace_configuration_changedrs=/044YF 69- 67r!r)rK)Aloggingr3typingrrrrrrr rjedir parso.pythonr parso.treer rope.base.resourcesr rope.contrib.autoimport.defsrrope.contrib.autoimport.sqliterpylsprpylsp.config.configrpylsp.workspacerr_rope_task_handler getLoggerrArrrrrrrrMBaseNodeLeafrErcrdrprorxrrrrintrrrrrrrrr__annotations__rFr!rrs]CCC !(55&/.g!  ^ .6.6b S$sDcN':";;< $==499==<#'===AYY $ DII T]] t  tyy  $  : $)) $! &<l#&< &< &< &<  &<  &<tCH~tT)*& >>>  >  >  $Z > >B *V* * * *F*y* * 6F6y6H6 6   8& 8Y 8  8)**r!