*dZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlZd d d d d ddddddd d dddZdZdZdZdZdZdZdZdZdZdZy) z4Methods for suggesting corrections to command typos.)absolute_import)division)unicode_literalsN)lookup)log)filesupdatescpaddremovegetzauth-configure-dockerdescribeimages instancesinstancecreatedelete)changez copy-filesrrrdockerr imagerrmakemodifypatchr showgffffff? dc0tjt}|s|Sd}tt j |ddd}|D]}tj j||}tj|5}|D]L}tjtj|}|s*|jd} |dz }|| xxdz cc<N dddtj |D cic] \} } | | |z  c} } S#1swYxYwcc} } w)zLoad the last 100 surfaces user used today from local command history. Args: logs_dir: str, the path to today's logs directory Returns: dict mapping surfaces to normalized frequencies. rT)reverseNr) collections defaultdictintsortedoslistdirpathjoinr FileReaderresearchrUSED_SURFACE_PATTERNgroupsix iteritems) logs_dirsurfaces_counttotallast_100_invocationsfilename file_pathlog_filelinematchsurfacecounts /lib/googlecloudsdk/calliope/suggest_commands.py_GetSurfaceHistoryFrequenciesr<;s**3/.   % 8 4dCDSI&h Xx0I   ) $$ #22D9 KKN' 1*%  !Q & !  % $'47==3P 3P%guu}3P  % $ s=,D*%D2DD c$|jtjs|j|ddyt j |tjD]3\}}|j|t ||||j5y)afHelper method to _GetCanonicalCommands. Args: tree: The root of the tree that will be traversed to find commands. results: The results list to append to. prefix: [str], the canonical command line words so far. Once we reach a leaf node, prefix contains a canonical command and a copy is appended to results. Returns: None N)r rLOOKUP_COMMANDSappendr.r/_GetCanonicalCommandsHelperpop)treeresultsprefixcommand command_trees r;r@r@Xsn &(( ) NN6!9 "}}T&2H2H-IJg| MM' gv> JJL Kc&g}t||g|S)aReturn list of all canonical commands in CLI tree in arbitrary order. Args: tree: The root of the tree that will be traversed to find commands. Returns: [[canonical_command_words]]: List of lists, all possible sequences of canonical command words in the tree. )rD)r@)rBrCs r;_GetCanonicalCommandsrIns 'dGB7 .rGcd}||vr|}|}n ||vr|}|}n|S|j|}|dkDr ||dz dk7r|S|dz }|dk(r|dz }n|dk(r|dz }n |dk(r|dz }|dk(r|d z }n|dk(r|d z }n|d z }t|t|z }|dkr d|z }|j|r|dz}||z }|dk(r |dkDr|dz }|S|dkDr ||dzkDr|d z }|S) arReturns the integer word match score for a command word. Args: index: The position of the word in the command. normalized_command_word: The normalized command word. canonical_command_word: The actual command word to compare with. canonical_command_length: The length of the actual command. Returns: The integer word match score, always >= 0. rr -r)findlen startswith) indexnormalized_command_wordcanonical_command_wordcanonical_command_lengthscore shorter_word longer_wordhitextras r; _WordScorer_}sY" % 66*L(K!88)L)K L &#1WS1W%, L2+%" RKE1$ RKE1$ RKE aZ RKE z RKE QJE k S. .% aZ IEl+ qje UNE aZ,q0 RKE , qy- 9 RKE ,rGc tj}ttj}|Dcgc]"}|j j dd$}}g}t|}ttt|}|D]F}t|} t} d} t|D]_\} } |D]U}t| || | }tj|}|rt| || | }||kr|}|s@| j!|| |z } Wat| t|k(r| dz } | dkDsdj#|dd}||vr| t%||t&zz } d}d |dk(r-| d z} t|d d|vrd }td g|d dz|vrd }d |dk(r| d z} t|d d|vrd }|r4|j)|| fI|j+d|Scc}w)aReturn scored canonical commands containing input command words. Args: command_words: List of input command words. Returns: [(canonical_command_words, score)]: List of tuples, where canonical_command_words is a list of strings and score is an integer > 0. The tuples are sorted from highest score to lowest, and commands with the same score appear in lexicographic order. _rKrr.NFalpharRr Tbetac|d |dfS)Nr r)tuples r;z._GetScoredCommandsContaining..s58)U1X)>rG)key)rLoadCompletionCliTreer<r GetLogDirlowerreplacerIsetmaprhrT enumerater_SYNONYMSr r r(r#FREQUENCY_FACTORr?sort) command_wordsrootsurface_history command_wordnormalized_command_wordsscored_commandsall_canonical_commandscanonical_command_setcanonical_command_wordsrYmatchedrZrVrXrW incrementalternate_command_wordalternate_incrementr9better_track_existss r;_GetScoredCommandsContainingrsT  % % '$1#--/B/ANANl""3,/06c%)?@A!7"#:;eG E)23J)K%%%= !  # " $  "*.E!F ! *$$& !  , ,+I  ++- . 9 %)&>*L0 7|s344 rke qy0"56g O # _W-0@@AA" +A. .   (, -1F F $  6(4QR88 9$ %!%  *1- -   (, -1F F $   7?@k"8p>? s'G&c g} t|}|s|St|dd}d}t }|D]o\}}||z t k\s|j djdg|z|jdj|ddt|tk\smd}n|rt|tkrt|dd}|D]-\}}||z t krn|t|kDs#t|}/|dz } | rht }|D]N\}}||z t krn=|jdjdg|d| zt|tk\sNnt|}|S#tj$rd}YuwxYw) aReturn suggested commands containing input command words. Args: command_words: List of input command words. Returns: [command]: A list of canonical command strings with 'gcloud' prepended. Only commands whose scores have a ratio of at least MIN_RATIO against the top score are returned. At most MAX_SUGGESTIONS command strings are returned. If many commands from the same group are being suggested, then the common groups are suggested instead. Nrr F gcloudrcT) rrCannotHandleCompletionErrorfloatro MIN_RATIOr?r(r rTMAX_SUGGESTIONSMIN_SUGGESTED_GROUPSr$) rusuggested_commandsrz top_scoretoo_manysuggested_groupsrErZ min_length common_lengths r;GetCommandSuggestionsrs2=AO  OA&q)*) (U'ngu yI%(g)= >?388GCRL12  O 3 (#&'*>>_Q'*+J)  Y & c'l "\ * NM+.'5 9 y ( SXXxj7>M3J&JKL  O 3  , ""23 K  + +Os E,,FF)__doc__ __future__rrrr!r%r*,googlecloudsdk.command_lib.static_completionrgooglecloudsdk.corergooglecloudsdk.core.utilrr.rrrrrrsr<r@rIr_rrrgrGr;rs ;&' ?#* %      $  :, EPM`5rG