o 3g5@sdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z gdZ d$d d Z d%d&ddZd'd(ddZd'ddZeeddfddZd)d d!Zd'd"d#ZdS)*z7 Terminal-related utilities -------------------------- ) annotationsN)local)Progress)get_terminal_size)readlineaskchoosepromptrrrreturn list[str]cCstSN)__all__rrL/home/garg/my-data/venv/lib/python3.10/site-packages/plumbum/cli/terminal.py__dir__srmessagestrcCs tj|tjtjS)z*Gets a line of input from the user (stdin))sysstdoutwriteflushstdinr)rrrrr s   rquestiondefault bool | NoneboolcCs|dd}|dur|d7}n |r|d7}n|d7} z t|}Wn ty5d}Ynw|dvryyes>nnoFz#Invalid response, please try again )rstriprstriplowerEOFErrorrrr)rranswerrrrr's(      rc CsPt|dr |}tj|di}d}t|dD]9\}}t|tt fr6t |dkr6|d}|d}n|}|}|||<|durH||krH|}tjd|d|dq|durk|durdd |d nd |d } nd } zt |  } Wn t yd } Ynw| s|r|Szt| } | |vrtW|| StytjdYqmw)aPrompts the user with a question and a set of options, from which the user needs to choose. :param question: The question to ask :param options: A set of options. It can be a list (of strings or two-tuples, mapping text to returned-object) or a dict (mapping text to returned-object).`` :param default: If ``None``, the user must answer. Otherwise, lack of response is interpreted as this answer :returns: The user's choice Example:: ans = choose("What is your favorite color?", ["blue", "yellow", "green"], default = "yellow") # `ans` will be one of "blue", "yellow" or "green" ans = choose("What is your favorite color?", {"blue" : 0x0000ff, "yellow" : 0xffff00 , "green" : 0x00ff00}, default = 0x00ff00) # this will display "blue", "yellow" and "green" but return a numerical value items Nrr(z) zChoice [z]: zChoice: Trz!Invalid choice, please try again )hasattrr(rrrr# enumerate isinstancetuplelistlenrr$r&int ValueError) roptionsrchoicesdefindexiitemtextvalmsgchoicerrrr GsH  "  r cCsdS)NTr)_rrrsr>c Cs |d}|tur|d|d7}|d7} zt|}Wn ty)d}Ynw|s3|tur2|Sqz||}WnttfyW}ztj d|dWYd }~qd }~wwz||}Wntyy}ztj |d WYd }~qd }~ww|stj d q|S) a Presents the user with a validated question, keeps asking if validation does not pass. :param question: The question to ask :param type: The type of the answer, defaults to str :param default: The default choice :param validator: An extra validator called after type conversion, can raise ValueError or return False to trigger a retry. :returns: the user's choice z :z []z: TrzInvalid value (z), please try again Nz, please try again z/Value not in specified range, please try again ) r#NotImplementedrr$r& TypeErrorr3rrr)rtyper validatoransexvalidrrrr s@       r Tc #stdrfdd}nfdd}d}d}t|D]@\}}ddd |D}d d d |D} |r?||kr?d }q|}|rFd V|dd} |dd| d| Vd}qdS)zConvert the given bytes (or a stream with a buffering ``read()`` method) to hexdump-formatted lines, with possible aggregation of identical lines. Returns a generator of formatted lines. readc3s }|s dS|Vqr )rH)bufbytes_per_linedata_or_streamrr read_chunks zhexdump..read_chunkc3s.tdtD] }||Vq dS)Nr)ranger1)r7rJrrrMsNF css|] }t|dVqdS)02xNord.0chrrr szhexdump..rcss4|]}dt|krdkrnn|ndVqdS) .NrQrSrrrrVs2T*06xz | z| )r,r-joinljust) rLrK aggregaterMprevskippedr7chunkhexdr9 hexd_ljustrrJrhexdumps&   recCs|stdd}t|dr|}|jddd}zp|D]}|d}z |j||jWqty9Ynw|j| Wt t  |Wdn1sXwY|r| durt t  |Wdn1sywYtddSdSdSt t  |Wdn1swY|r| durt t  |Wdn1swYtdwww)zOpens a pager (e.g., ``less``) to display the given text. Requires a terminal. :param rows: a ``bytes`` or a list/iterator of "rows" (``bytes``) :param pagercmd: the pager program to run. Defaults to ``less -RSin`` lessz-RSin splitlinesN)rstderrr)reset)rr,rgpopenrrrOSErrorclosewait contextlibsuppress Exceptionpoll terminateossystem)rowspagercmdpgrowlinerrrpagersD               rz)r r )r)rrr rr )rrrrr r)rGT)__doc__ __future__rrnrsrplumbumrprogressrtermsizerrrrrr rr@r rerzrrrrs&       ; 1#