r!dZddlmZddlmZddlmZddlZddlZddlmZddl m Z dZ d d Z d Z Gd d eZy)z!Prompt completion support module.)absolute_import)division)unicode_literalsN) console_attr)rangec| |z S)zCreturns numerator/denominator rounded up if there is any remainder.) numerator denominators 3lib/googlecloudsdk/core/console/prompt_completer.py_IntegerCeilingDivider s : $ %%c Xfd}d}d}t|} || \} } } | |kDr5| dkDr0|D chc]} || | dz } } || \} } } | |kDr| dkDr0t| } |rAttjj | }| D cgc]} || | |||} } dg}d}|| krh|}t | D]<}|t| k\rn,|j|j| ||| z }>|jd|dz }|| krh|Scc} wcc} w) 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. ctd|D}tt|zz xsd}tt||}|||fS)z+Returns the transpose dimensions for items.c32K|]}t|yw)Nlen).0xs r z<_TransposeListToRows.._Dimensions..:s15a3q65s)maxintrr )itemslongest_item_len column_count row_countpadwidths r _Dimensionsz)_TransposeListToRows.._Dimensions8sO1511uC+; ;<=BL%c%j,?I \9 44rc2t||kr|S|d|dzS)z:Truncates and appends '*' if len(item) > longest_item_len.N*r)itemrs r _TrimAndAnnotatez._TransposeListToRows.._TrimAndAnnotate?s( 4y$$ k !! "S ((rclt|}||kDr|d||z||z|z||dzdz}|||z dzzS)zBHighlights the different part of the completion and left justfies.Nr r)r#rdifference_indexboldnormallengths r _Highlightz(_TransposeListToRows.._HighlightEsk YF   $$%,#$%'-.#A%&'(d #f,3 33r rr)setsortedrospath commonprefixrappend) all_itemsrheightrr(r)r r$r+rrrrrr'row_data row_index column_index_s ` ` r _TransposeListToRowsr;#sq*5) 4 i.%.9%.@+L)F/!3@I J 1 a!1A!5 6 E J0;E0B-lI F/!3 -% 277//67 +-=tV L  V()IL <  U # oocooeL)*il !  OOD NI I /1 K s D"D'cP|Dcgc]}|j|s|c}Scc}w)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. ) startswith)prefixpossible_matchesrs r _PrefixMatchesr@ms)& >%f)=!% >> >s##c0eZdZdZdZdZ ddZdZdZy) 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 Nc||_||_|xstj|_t j |_|jj\}}| d}||kDr|}||_ | d}||kDr|}||_ ||_ y)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. NP() _prompt_choicessysstderr_outr ConsoleAttr_attr GetTermSize_width_height_pad) selfpromptchoicesoutrr6r term_width term_heights r __init__zPromptCompleter.__init__sDLDM!szzDI))+DJ"jj446J  }e  DK ~f + DLDIrc^d}g}g}|jr%|jj|jd} |}|jj }|dddt j fvst|dk7r|jjdn|dt jfvr2|rm|j|jjd|}n>|dk(r dj|}||k(r.t||}t|dkDr|j||n|2t|jr|j}n |j}|}t||}dj|}tj j#|}t|t|kDrkt||}|jj|t|dt%|}n,|j'||jj|dj|S) z=Reads and returns one line of user input with TAB complation.Nr. rz  )rHrLwriterN GetRawKeyrB _CONTROL_Cr_DELETEpopjoinr@_DisplaycallablerIr1r2r3listr4)rS all_choicesmatchesresponsec previous_cresponse_prefix common_prefixs r InputzPromptCompleter.InputsKGH || iioodll# A j **   a tT4!;!;< <A!   ../ /  ,,. ))//' "' 9''(+ ?"?G<' \A  MM/7 3   & MMOk MMk'#?G<'GGH-/''..w7- #h- /%]Grir7s r rezPromptCompleter._Displays$t{{4<r{sH (&' 4& IM $GT ?{'f{'r