U fw%Z@s$ddlZddlZGdddeZdS)Nc@seZdZdZd2ddZddZddZd d Zd d Zd dZ e ddZ ddZ ddZ d3ddZddZddZddZddZdd Zd!d"Zd4d$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1ZdS)5TerminalStatusaShow status and progress information on a terminal. All output is provided via widgets of various kinds. Many widgets format data that TerminalStatus stores. TerminalStatus provides a dict interface for setting and retrieving data items. Unlike a real dict, getting a value for a key that has not been set does not result in a KeyError exception, but in the empty string being returned. NcCs |ptj||d|_|dS)N)period _terminal) ttystatusZMessager_mclear)selfrZmessagerrr 2/usr/lib/python3/dist-packages/ttystatus/status.py__init__#s zTerminalStatus.__init__cCs |jS)zReturn terminal width, height.)rget_terminal_sizerr r r r (sz TerminalStatus.get_terminal_sizecCs4|jsgg|_|jd||||dS)z'Add a new widget to the status display.N) _widget_rowsappend_register_interestsflush)rwidgetr r r add,s  zTerminalStatus.addcCsPt|dddkr|j|n.|jD]&}|j|g}||||j|<q$dS)N interested_in)getattr_unknown_interestrr_interested_inget)rrkeywidgetsr r r r4s   z"TerminalStatus._register_interestscCs&|jsgg|_|jg|dS)zStart a new line of widgets.N)rrrr r r r start_new_line=s zTerminalStatus.start_new_linecCsNt|dD]2\}}|dkr&|t|D]}||q0q|dS)a)Add new widgets based on format string. The format string is taken literally, except that ``%%`` is a literal percent character, and ``%Foo(a,b,c)`` is a widget of type ``Foo`` with parameters a, b, and c. For example: ``format("hello, %String(name)")``.  rN) enumeratesplitrrparserr)r format_stringilinerr r r formatDs zTerminalStatus.formatcCsg}|jD] }||7}q |SN)r)rresultrowr r r rUs  zTerminalStatus.widgetscCs&g|_i|_i|_g|_|jdS)zRemove all widgets.N)r_valuesrrrrr r r r r\s zTerminalStatus.clearcCs|j|dS)z*Return value for key, or the empty string.r(r)rrr r r __getitem__dszTerminalStatus.__getitem__cCs|j||S)zLike dict.get.r*)rrdefaultr r r rhszTerminalStatus.getcCsZ||j|<|j|g|jg}|D]}|D]}||q*q"|jjrV|jrV|dS)zSet value for key.N) r(rrrupdaterenabledZ time_to_write_write)rrvalueZ widget_listsrwr r r __setitem__ls  zTerminalStatus.__setitem__cCs|jdS)aJHide current progress report. Use .flush() to make it visible again. Hiding is useful if you want to write things to stdout/stderr that might get mixed with progress output. The .notify() and .error() methods get disabled if progress reporting gets disabled, but .hide() doesn't. N)rrr r r r hideys zTerminalStatus.hidecCs|jdddS)zForce an update of current state to the screen. This happens even if it is not yet time to output the screen. TforceN)r/r r r r rszTerminalStatus.flushcsdfddjDS)z$Render current state of all widgets.rc3s|]}|VqdSr%) _render_row).0r'r r r sz)TerminalStatus._render..)joinrr r r r _renderszTerminalStatus._rendercCs|j}|}dgt|}t|D]2\}}|jr$||d||<|t||8}q$t|D]2\}}|js`|||||<|t||8}q`d|d|S)Nrr))rZget_max_line_lengthlenrZ static_width _make_safeZrenderr9)rZ widget_rowZ max_charsZ remainingZtextsr"r1r r r r6s zTerminalStatus._render_rowcs ddfdd|DS)z7Expand TABs, remove all other ASCII control characters. r)c3s"|]}t|kr|ndVqdS)r)N)ord)r7cZ ASCII_SPACEr r r8sz,TerminalStatus._make_safe..)r9 expandtabs)rr#r r@r r<szTerminalStatus._make_safeFcCs|jj||ddS)z/Render and output current state of all widgets.r4N)rwriter:)rr5r r r r/szTerminalStatus._writecCs||p d|||<dS)z+Increase value for a key by a given amount.rNr )rrZdeltar r r increaseszTerminalStatus.increasecCs|j|tjdS)zShow a message.N)rnotifysysstdoutrmsgr r r rDszTerminalStatus.notifycCs|jj|tjdddS)zWrite an error message.Tr4N)rrDrEstderrrGr r r errorszTerminalStatus.errorcCs||jdS)zFinish status display.N)r/rfinishr r r r rKszTerminalStatus.finishcCs|jdS)zDisable all output.N)rdisabler r r r rLszTerminalStatus.disablecCs|jdS)z&Enable output if it has been disabled.N)renabler r r r rMszTerminalStatus.enable)NNN)N)F)__name__ __module__ __qualname____doc__r r rrrr$propertyrrr+rr2r3rr:r6r<r/rCrDrJrKrLrMr r r r rs2        r)rErobjectrr r r r s