o *@sdZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlZd d d d d ddddddd d dddZdZdZdZdZddZddZddZd d!Zd"d#Zd$d%ZdS)&z4Methods for suggesting corrections to command typos.)absolute_import)division)unicode_literalsN)lookup)log)filesupdatescpaddremovegetzauth-configure-dockerdescribeimages instancesinstancecreatedelete)changez copy-filesrrr dockerr imagerrmakemodifypatchr showgffffff? dc stt}|s |Sdtt|dddd}|D]=}tj||}t |'}|D]}t t j |}|rG|d}d7||d7<q+Wdn1sRwYqfddt|DS) 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)reverseNrcsi|] \}}||qSr).0surfacecounttotalrE/tmp/google-cloud-sdk/lib/googlecloudsdk/calliope/suggest_commands.py Ssz1_GetSurfaceHistoryFrequencies..) collections defaultdictintsortedoslistdirpathjoinr FileReaderresearchrUSED_SURFACE_PATTERNgroupsix iteritems) logs_dirsurfaces_countlast_100_invocationsfilename file_pathlog_filelinematchr!rr#r%_GetSurfaceHistoryFrequencies;s*    r>cCs^|tjs||dddSt|tjD]\}}||t||||qdS)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_COMMANDSappendr4r5_GetCanonicalCommandsHelperpop)treeresultsprefixcommand command_treerrr%rAXs    rAcCsg}t||gd|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. )rE)rA)rCrDrrr%_GetCanonicalCommandsns rHc Cs0d}||vr |}|}n ||vr|}|}n|S||}|dkr)||ddkr)|S|d7}|dkr6|d7}n|dkr?|d7}n|dkrG|d7}|dkrP|d 7}n |dkrY|d 7}n|d 7}t|t|}|dkrzd|}||rv|d9}||7}|dkr|dkr|d7}|S|dkr||dkr|d 7}|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_wordhitextrarrr% _WordScore}sF      r]cCst}tt}dd|D}g}t|}ttt|}|D]}t |}t} d} t |D]/\} } |D](} t | | | |}t | }|rRt | || |}||krR|}|r]| | | |7} q5q/t | t |krk| d7} | dkrd|dd}||vr| t||t7} d}d |dkr| d 8} t|d d|vrd }td g|d d|vrd }d |dkr| d 8} t|d d|vrd }|s||| fq |jddd|S)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. cSsg|] }|ddqS)_rI)lowerreplace)r command_wordrrr% sz0_GetScoredCommandsContaining..rr.NFalpharPrTbetacSs|d |dfS)Nrrr)tuplerrr%sz._GetScoredCommandsContaining..)key)rLoadCompletionCliTreer>r GetLogDirrHsetmaprgrR enumerater]SYNONYMSr r r.r)FREQUENCY_FACTORr@sort) command_wordsrootsurface_historynormalized_command_wordsscored_commandsall_canonical_commandscanonical_command_setcanonical_command_wordsrWmatchedrXrTrVrU incrementalternate_command_wordalternate_incrementr!better_track_existsrrr%_GetScoredCommandsContainingsv     rc Csdg}zt|}Wn tjyd}Ynw|s|St|dd}d}t}|D]+\}}||tkrS|ddg||d|ddt |t krSd}nq(|rt |t krt |dd}|D]\}}||tkrrn |t |kr|t |}qf|d} | rt}|D]#\}}||tkrn|ddg|d| t |t krnqt |}|S) 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. NrrF gcloudrdT) rrCannotHandleCompletionErrorfloatrl MIN_RATIOr@r.r rRMAX_SUGGESTIONSMIN_SUGGESTED_GROUPSr*) rrsuggested_commandsrv top_scoretoo_manysuggested_groupsrFrX min_length common_lengthrrr%GetCommandSuggestionssN           r)__doc__ __future__rrrr'r+r0,googlecloudsdk.command_lib.static_completionrgooglecloudsdk.corergooglecloudsdk.core.utilrr4rorrrrpr>rArHr]rrrrrr%sJ      H P