a Fb/@s<dZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddl mZmZdd lmZd d lmZdd lmZdd lmZe Zd ZdZeddddeDZedZddZddZ ddZ!ddZ"d,e#ee#dddZ$d d!Z%d"d#Z&d$d%Z'd&d'Z(d(d)Z)d*d+Z*dS)-z\Implementations for various useful completers. These are all loaded by default by IPython. N) import_module) all_suffixes)time) zipimporter) expand_user compress_user)TryNext) arg_split) get_ipython)Listz?(?P[^\W\d]\w*?)(?P[/\\]__init__)?(?P%s)$|ccs|]}t|VqdSN)reescape.0sr8lib/python3.9/site-packages/IPython/core/completerlib.py :rz.*(\.ipy|\.ipynb|\.py[w]?)$cs|dkr d}tjjtj|rg}tj|ddD]T\}}}|t|ddr||fdd|Dg|dd<q2||q2n(ztt|j }Wng}Yn0g}|D]"}t |}|r| |d qtt|S) z` Return the list containing the names of the modules available in the given folder. .T) followlinksrNc3s|]}|VqdSrrrfZpjoinZsubdirrrrWrzmodule_list..name)ospathjoinisdirwalklenextendlistr_fileskeys import_rematchappendgroupset)r"filesrootdirsnondirsmodulesrmrrr module_listCs*   r6c Cst}|durttjS|jdi}ttj}t}d}tjD]}z ||}Wntyt |}z| dWnt yYn0|dvr|||<t|t kr|sd}t dt dtjt|tkrt d gYSYn0||q@|r ||jd<tt|}|S) z Returns a list containing the names of all the modules available in the folders of the pythonpath. ip.db['rootmodules_cache'] maps sys.path entries to list of modules. Nrootmodules_cacheF__init__)rrTz/ Caching the list of root modules, please wait!z@(This will only be done once - type '%rehashx' to reset cache!) z%This is taking too long, we give up. )r r(sysbuiltin_module_namesdbgetrr"KeyErrorr6remove ValueErrorTIMEOUT_STORAGEprintstdoutflushTIMEOUT_GIVEUPr'r/)ipr7Z rootmodulesZ start_timeZstorer"r4rrrget_root_modulesks>          rFcCs:|rtt||S|dddko2|dddk SdS)Nr __)inspectismodulegetattr)moduleattr only_modulesrrr is_importablesrOcCs@zt||}Wn&ty"YdSty4YdS0t|S)NTF)rKAttributeError TypeErrorrIrJ)rLrMobjrrris_possible_submodules  rSF)modreturncs|d}z t|WngYS0dtddp8dv}g}tdrRrR|rp|fddtDtdg}r|fdd |Dn |||r|ttj j d d |D}| dt |S) zN Try to import given module and return list of potential completions. rr8__file__rcsg|]}t|r|qSr)rOrrMr5rNrr s ztry_import..__all__c3s|]}t|r|VqdSr)rSrW)r5rrrrztry_import..cSsh|]}t|tr|qSr) isinstancestrrcrrr rztry_import..) rstriprrKhasattrr'dirr6r!r"dirnamerVdiscardr()rTrNZ m_is_init completionsZm_allZcompletions_setrrXr try_imports$      rfcs4ttrfdd}tjd||ddS)a Easily create a trivial completer for a command. Takes either a list of completions, or all completions in string (that will be split on whitespace). Example:: [d:\ipython]|1> import ipy_completers [d:\ipython]|2> ipy_completers.quick_completer('foo', ['bar','baz']) [d:\ipython]|3> foo b bar baz [d:\ipython]|3> foo ba csSrrselfeventrerr do_completesz$quick_completer..do_completeZcomplete_command)Zstr_keyN)r[r\splitr Zset_hook)cmdrerkrrjrquick_completers  rncs|d}t|}|dkr,|ddkr,dgS|dkr|ddvr|dkrNtS|ddtd krntStdd d d }fd d|DS|dkr|ddkr|dtSd S)z Returns a list containing the completion possibilities for an import line. The line looks like this : 'import xml.d' 'from xml.dom import'  rfromzimport >rqz%aimportimportrrr NTcs$g|]}ddd|gqS)rNrs)r#)rZelrTrrrYrz%module_completion..)rlr&rFrfr#)lineZwordsZnwordsZcompletion_listrrtrmodule_completions  rvcCs t|jS)z@Give completions after user has typed 'import ...' or 'from ...')rvrurgrrrmodule_completers rwcst|jdd}t|dkr6|jds6|dd}nd}tj}tjjt |\}t dd |Drfd d ||d D}n\fd d ||d D}dd ||d||d||d||dD}||}fdd |DS)zKComplete files that end in .py or .ipy or .ipynb for the %run command. F)strictrrorsz'"rcss|]}t|VqdSr) magic_run_rer,r]rrrr+rz&magic_run_completer..cs(g|] }|dd|rdndqS)\/rreplacerr$rrrY,sz'magic_run_completer..*cs$g|]}|r|dddqSrzr{r|rr~rrrY/rcSsg|]}|ddqSrr|rrrrrY0sz*.pyz*.ipyz*.ipynbz*.pywcsg|]}t|qSrrrp tilde_expand tilde_valrrrY7r) r rur&endswithstripglobr!r"r$rany)rhricompsrelpathZlglobZmatchesr2Zpysr)r$rrrmagic_run_completers*    rc s|t}j}jds"djvr@|jdd}|r<|SgSjdkrtttt|j dd}d|d fd d t |j dD}t|dkr|SgSj d rd d |j dDSt |\}| dd}g}dd t|dDD]}d|vr t||q|shtj|r:t|gS|jdi} fdd | D} | rd| Stfdd |DS)z:Completer function for cd, which only returns directories.z-bz -b Z bookmarksN-Z_dhrz-%0zd [%s]csg|]\}}||fqSrr)rir)fmtrrrYLrz cd_completer..--cSsg|]}dtj|qS)r)r!r"basename)rdrrrrYRrrzr{cSs(g|] }tj|r|dddqSr)r!r"r$r}rrrrrYYs rrocsg|]}|jr|qSr) startswithsymbolr)rirrrYhrcsg|]}t|qSrrrrrrrYnr)r rrurr;r<r*r\r&Zuser_ns enumeraterrr}rr r-r!r"r$r) rhrirErZbkmsZwidth_dhZentsfoundrZbksZ bkmatchesr)rirrrr cd_completer:s@       rcCsdS)zA completer for %reset magicz-f -s in out array dhist)rlrgrrrreset_completerpsr)F)+__doc__rrIr!rr9 importlibrimportlib.machineryrr zipimportrZ completerrrerrorr Zutils._process_commonr ZIPythonr typingr Z _suffixesr@rDcompiler#r+ryr6rFrOrSr\rfrnrvrwrrrrrrrs@         (, "%%6