o a<@sddlmZmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z e jdkr:dZndZeejddd eZd d Zd d ZdZGdddeZdS))print_functionabsolute_importN)SqliteDatabase3gettextZugettextzcommand-not-foundT)fallbackcsvdfddttdD}dd|D}dd|D}fdd|D}fdd|D}t||||S) zw return a set with spelling1 distance alternative spellings based on http://norvig.com/spell-correct.html z&abcdefghijklmnopqrstuvwxyz-_0123456789cs$g|]}d||dfqSN).0i)wordr A/usr/lib/python3/dist-packages/CommandNotFound/CommandNotFound.py #$z!similar_words..cSs$g|]\}}|r||ddqSrNr r abr r r r$rcSs<g|]\}}t|dkr||d|d|ddqS)rrNlenrr r r r%s<cs2g|]\}}D]}|r|||ddqqSrr r rrcalphabetr r r&s2cs&g|]\}}D]}|||qqSr r rrr r r's&)rangerset)r sZdeletesZ transposesZreplacesinsertsr )rr r similar_wordssr cCsDzt}tdd|vptdd|vWSty!YdSw)NsudorZadminF)posix getgroupsgrpgetgrnamKeyError)groupsr r r user_can_sudo+s r(z&/var/lib/command-not-found/commands.dbc@seZdZdZdZdZdZejZ d*ddZ d+d d Z d d Z d dZ ddZddZddZd,ddZddZddZddZddZd d!Zd"d#Zd$d%Zd-d'd(Zd)S).CommandNotFoundz programs.d)z /snap/binz/binz/usr/binz/usr/local/binz/sbinz /usr/sbinz/usr/local/sbinz /usr/gamesz /usr/bin/snap/usr/share/command-not-foundcCsN||_tjtrtttjrtt|_ nt dt |_ t |_dS)NzCannot find database)_getSourcesList sources_listospathexistsdbpathaccessR_OKrdbFileNotFoundErrorr(r"geteuideuid)selfZdata_dirr r r __init__Ks  zCommandNotFound.__init__c Csdg}|t|kr|jks|S|St|D]}||}|D]\}}}|||||fq q|S)z try to correct the spelling )rmax_lenr get_packagesappend) r8r Zmin_lenZpossible_alternativeswpackagespackagevercompr r r spelling_suggestionsUs  z$CommandNotFound.spelling_suggestionscCs |j|Sr)r4lookupr8commandr r r r<`s zCommandNotFound.get_packagesc Cslg}g}tj|jstd|jggfSz%ttj}tj |jddd|g|dd}Wdn1s6wYWntj yZ}ztd|ggfWYd}~Sd}~wwtd|zt |}Wnt j y}ztd |ggfWYd}~Sd}~ww|D]'}|d |kr||d |d |d fq||d |d |d fq||fS) Nz %s not existsz advise-snapz --format=jsonz --commandT)stderrZuniversal_newlinesz.calling snap advice-snap returned an error: %szgot %s from snap advise-snapzcannot decoding json: %sZCommandZSnapZVersion)r.r/r0snap_cmdloggingdebugopendevnull subprocessZ check_outputZCalledProcessErrorjsonloadsZJSONDecodeErrorr=get) r8rFZ exact_resultZmispell_resultrLoutputesnapssnapr r r get_snapscsD    zCommandNotFound.get_snapscCsjz)ttjtdddf}dd|DWdWS1s"wYWdSty4gYSw)NHOMEz/rootz.command-not-found.blacklistcSs g|] }|dkr|qS))strip)r liner r r rs z0CommandNotFound.getBlacklist..)rKr.sepjoingetenvIOError)r8Z blacklistr r r getBlacklists ( zCommandNotFound.getBlacklistc Cszzddl}ddlm}|Wn ttfygYSwtg}|ddD]}|js:|js:|j D]}| |q2q'|S)Nr) SourcesListF)Z withMatcher) apt_pkgZaptsources.sourceslistr_Zinit SystemError ImportErrorrZdisabledZinvalidcompsadd)r8r`r_r-sourceZ componentr r r r,s     zCommandNotFound._getSourcesListcCsdtjvrdS|r[td}tjdkrt|}dd}n||}tjjr/t|t r/| tjj}| tdkr]|j dkr?d}nd }d ||f}t d |tjd tj|d ddSdSdS)NZ COMMAND_NOT_FOUND_INSTALL_PROMPTz Do you want to install it? (N/y)rcSs|Srr )xr r r sz0CommandNotFound.install_prompt..yrrWsudo z%sapt install %sz%sfileF)shell)r.environ_sysversioninputstdinencoding isinstancestrdecodelowerr7printstdoutrMcallsplit)r8Z package_namepromptZanswerZ raw_inputZcommand_prefixZinstall_commandr r r install_prompts&      zCommandNotFound.install_promptc CsVt|t||kr ttd|t|f|jd|jdSt|t|dkrtttd||jd|D]\}}}|rCd|}nd}ttd|||f|jdq7|D]\}}} }|rcd|}nd}ttd |||f|jdqVt|dkrttd |jdn t|dkr|jrttd d |jdn ttd d |jd|jdS)z3 print spelling suggestions for packages and snaps z6Command '%s' not found, but there are %s similar ones.rjNr%Command '%s' not found, did you mean:z (%s)rWz command '%s' from snap %s%s command '%s' from deb %s%s3See 'snap info ' for additional versions.zTry: %s zsudo apt installz apt install)rrxrn output_fdflushr() r8r mispell_packages mispell_snapsZmax_altrFrTrAr@rBr r r print_spelling_suggestionss.     z*CommandNotFound.print_spelling_suggestionscCsttdd|i|jddS)Nz;Command '%(command)s' not found, but can be installed with:rFrj)rxrnrrEr r r _print_exact_headers  z#CommandNotFound._print_exact_headercCs|||d}|jdkrtd|d|jdn |jr)td|d|jdntd|d|jdttd|jdS)Nrzsnap install %srjzsudo snap install %sPlease ask your administrator.)rr7rxrr(rnr)r8rFr?rSrTr r r advice_single_snap_packages   z*CommandNotFound.advice_single_snap_packagecCs|||jdkr!td|dd|jd||ddnE|jr;td|dd|jd||ddn+td|dd|jdttd|dd|jvrfttd|dd|jd|jdS)Nrzapt install %srjzsudo apt install %srrz1You will have to enable the component called '%s') rr7rxrr}r(rnr-r)r8rFr?rSr r r advice_single_deb_packages   z)CommandNotFound.advice_single_deb_packagecCs|jdkr |jr dSdS)NrrirW)r7r()r8r r r r!szCommandNotFound.sudocCs||tdd||D}t|D]W\}}d}|dr5|dkr/t|dkr/d|d}nd|d}|d|jvrNtd |||d|f|jd qtd |||d|fd td |dd |jd q|j dkr}|j s}ttd|jd |j dS)NcSg|]}t|dqSrrr rr r r rz.rWrr # version %s, or # version %srz%sapt install %-*s%srjz (z-You will have to enable component called '%s')z.Ask your administrator to install one of them.) rmax enumeraterr-rxr!rrnr7r(r)r8rFr?rSpadr r@rAr r r advice_multi_deb_packages  $<z(CommandNotFound.advice_multi_deb_packagecCs||tdd||D}t|D]2\}}d}|dr5|dkr/t|dkr/d|d}nd|d}td|||d|f|jd qttd |jd |jdS) NcSrrrrr r r rrz>CommandNotFound.advice_multi_snap_packages..rWrrrr%ssnap install %-*s%srjr) rrrrrxr!rrnr)r8rFr?rSrr rTrAr r r advice_multi_snap_packagess  $z*CommandNotFound.advice_multi_snap_packagesc Cs||tdd||D}t|D],\}}d}|dr/|dkr)d|d}nd|d}td|||d|f|jd q|D]}d}|d rQd|d }td |||d|f|jd qCt|d kryttd |dd|jd n ttd |jd |jdS)NcSrrrrr r r rrz?CommandNotFound.advice_multi_mixed_packages..rWrrrrrrjrz%sapt install %-*s%sz+See 'snap info %s' for additional versions.r) rrrrxr!rrrnr) r8rFr?rSrr rTrAr@r r r advice_multi_mixed_packagess$  $ $  z+CommandNotFound.advice_multi_mixed_packagesFc sPddt|jkr dSdr#tjr tjg}ng}n fdd|jD}|r|st|dkrNtt dtj |d d |j d n tt d d i|j d |D]}tdtj ||j d q]t t |t tddd}t|d krtt dd ||j d dd |vrtt d|j d dStjdstjdsdS|vrdSdkrtt d|j d tt dd|j d tt dd|j d dS|}|\}}td||ft|d kr t|d kr |} t| d kst|d kr|| |nmt|d kr6t|dkr6|||nWt|d krLt|d krL|||nAt|dkrbt|d krb|||n+t|dkrxt|d krx|||nt|d krt|d kr|||t|d kpt|d kpt|d kpt| d kS)z7 give advice where to find the given command to stderr cSs.tjtj||otjtj|| S)z? helper that returns if a command is found in the given prefix )r.r/r0r[isdir)prefixrFr r r _in_prefix'sz*CommandNotFound.advise.._in_prefixF/csg|] }|r|qSr r )r rrrFr r r5sz*CommandNotFound.advise..rz1Command '%(command)s' is available in '%(place)s'r)rFZplacerjz:Command '%(command)s' is available in the following placesrFz * %sPATHrW:z_The command could not be located because '%s' is not included in the PATH environment variable.ZsbinzfThis is most likely caused by the lack of administrative privileges associated with your user account.z /usr/bin/aptz/usr/bin/aptitudepythonrr)python3rrW)rzpython-is-python3rWTzgot debs: %s snaps: %s)rr; startswithr.r/r0dirnameprefixesrxrnr[rlistrr\r{r^r<rUrIrJrCrrrrrr) r8rFZignore_installedrrmissingr?rSrrr rr advise%sl   ,"        zCommandNotFound.adviseN)r+)r:)r~)F)__name__ __module__ __qualname__Z programs_dirr;rrHrorGrr9rCr<rUr^r,r}rrrrr!rrrrr r r r r)8s,      r))Z __future__rrrr$rNrIr.os.pathr"rorMZCommandNotFound.db.dbrrpZ_gettext_methodgetattr translationrnr r(r1objectr)r r r r s&