o r!@sdZddlmZddlmZddlmZddlZddlZddlmZddl m Z dd Z  dd dZ ddZ GdddeZdS)z!Prompt completion support module.)absolute_import)division)unicode_literalsN) console_attr)rangecCs | | S)zCreturns numerator/denominator rounded up if there is any remainder.) numerator denominatorrrI/tmp/google-cloud-sdk/lib/googlecloudsdk/core/console/prompt_completer.py_IntegerCeilingDivides r P( csfdd}ddddt|}||\}} | |kr;dkr;fdd |D}||\}} | |kr;dks"t|}rVttj|fd d |D}d g} d } | | kr| } t|D]} | t|kron| | || | | 7} qe| d | d7} | | ks_| S)a2Returns padded newline terminated column-wise list for items. Used by PromptCompleter to pretty print the possible completions for TAB-TAB. Args: all_items: [str], The ordered list of all items to transpose. width: int, The total display width in characters. height: int, The total display height in lines. pad: str, String inserted before each column. bold: str, The bold font highlight control sequence. normal: str, The normal font highlight control sequence. Returns: [str], A padded newline terminated list of colum-wise rows for the ordered items list. The return value is a single list, not a list of row lists. Convert the return value to a printable string by ''.join(return_value). The first "row" is preceded by a newline and all rows start with the pad. csBtdd|D}tt|pd}tt||}|||fS)z+Returns the transpose dimensions for items.css|]}t|VqdS)Nlen.0xrrr :sz<_TransposeListToRows.._Dimensions..)maxintrr )itemslongest_item_len column_count row_count)padwidthrr _Dimensions8s z)_TransposeListToRows.._DimensionscSs t||kr|S|d|dS)z:Truncates and appends '*' if len(item) > longest_item_len.N*r)itemrrrr _TrimAndAnnotate?s z._TransposeListToRows.._TrimAndAnnotatecSsLt|}||kr|d|||||||dd}|||dS)zBHighlights the different part of the completion and left justfies.Nr r)r rdifference_indexboldnormallengthrrr _HighlightEsz(_TransposeListToRows.._Highlightcsh|] }|dqS)rr)r!rrr Rz'_TransposeListToRows..csg|] }|qSrrr)r'r$r#rr%rr Ysz(_TransposeListToRows.. rr)setsortedrospath commonprefixrappend) all_itemsrheightrr$r%rrrrrow_data row_index column_index_r)r'r!r$r#rr%rrr _TransposeListToRows#s:       r:csfdd|DS)aReturns the subset of possible_matches that start with prefix. Args: prefix: str, The prefix to match. possible_matches: [str], The list of possible matching strings. Returns: [str], The subset of possible_matches that start with prefix. csg|] }|r|qSr) startswithrprefixrr r,wr+z"_PrefixMatches..r)r=possible_matchesrr<r _PrefixMatchesms r?c@s6eZdZdZdZdZ  d ddZdd Zd d ZdS) PromptCompletera Prompt + input + completion. Yes, this is a roll-your own implementation. Yes, readline is that bad: linux: is unaware of the prompt even though it overrise raw_input() macos: different implementation than linux, and more brokener windows: didn't even try to implement Nrc Csv||_||_|p tj|_t|_|j\}}|dur$d}||kr$|}||_ |dur3d}||kr3|}||_ ||_ dS)aConstructor. Args: prompt: str or None, The prompt string. choices: callable or list, A callable with no arguments that returns the list of all choices, or the list of choices. out: stream, The output stream, sys.stderr by default. width: int, The total display width in characters. height: int, The total display height in lines. pad: str, String inserted before each column. Nr r ) _prompt_choicessysstderr_outr ConsoleAttr_attr GetTermSize_width_height_pad) selfpromptchoicesoutrr5r term_width term_heightrrr __init__s     zPromptCompleter.__init__cCsnd}g}g}|jr|j|jd} |}|j}|dddtjfvs)t|dkr0|jdn|dtjfvrF|rE| |jd|}nk|dkrd |}||kret ||}t|dkrd| ||nL|durvt |jrs|}n|j}|}t ||}d |}tj|}t|t|krt ||}|j|t|dt|}n |||j|qd |S) z=Reads and returns one line of user input with TAB complation.NTr- rz  )rCrGwriterI GetRawKeyr@ _CONTROL_Cr_DELETEpopjoinr?_DisplaycallablerDr0r1r2listr3)rN all_choicesmatchesresponsec previous_cresponse_prefix common_prefixrrr InputsT                7zPromptCompleter.InputcCs\t||j|j|j|jjdd|jd}|jr||j|||j d |dS)zDisplays the possible completions and redraws the prompt and response. Args: prefix: str, The current response. matches: [str], The list of strings that start with prefix. T)r$)rr5rr$r%rXN) r:rKrLrMrI GetFontCoderCr3rGrYr^)rNr=rcr6rrr r_s  zPromptCompleter._Display)NNNNr) __name__ __module__ __qualname____doc__r[r\rTrir_rrrr r@zs   Cr@)r r rNN)rn __future__rrrr0rEgooglecloudsdk.core.consoler six.movesrr r:r?objectr@rrrr s      J