U 祡c @s ddlmZmZmZmZmZddlmZddlm Z ddl m Z ddl m Z ertddlmZmZmZmZddlmZGd d d Zed krdd lmZdd lmZddlmZddlmZeddee DdddZ!eee!dddddZedqW5QRXdS))castListOptional TYPE_CHECKINGUnion)SPINNERS) Measurement)Table)Text)ConsoleConsoleOptions RenderResultRenderableType) StyleTypec@seZdZddddedededddd Zd d d d ddZd d ed ddZ eddddZ dddddedeeddddZ dS)SpinnerNg?)stylespeedrr)nametextrrreturncCsz t|}Wn"tk r.td|YnXt|trDt|n||_ttt|ddd|_ tt |d|_ d|_ ||_ ||_d|_d|_dS)aA spinner animation. Args: name (str): Name of spinner (run python -m rich.spinner). text (RenderableType, optional): A renderable to display at the right of the spinner (str or Text typically). Defaults to "". style (StyleType, optional): Style for spinner animation. Defaults to None. speed (float, optional): Speed factor for animation. Defaults to 1.0. Raises: KeyError: If name isn't one of the supported spinner animations. zno spinner called framesNinterval)rKeyError isinstancestrr from_markuprrrrfloatr start_timerrframe_no_offset _update_speed)selfrrrrspinnerr%0/tmp/pip-unpacked-wheel-qkhzhrww/rich/spinner.py__init__s zSpinner.__init__r r r)consoleoptionsrccs||VdS)N)renderZget_time)r#r(r)r%r%r&__rich_console__0szSpinner.__rich_console__cCs|d}t|||S)Nr)r*r get)r#r(r)rr%r%r&__rich_measure__5s zSpinner.__rich_measure__)timercCs|jdkr||_||j|j|jd|j}t|jt|t|j|jpPdd}|j rv||_||_|j |_d|_ |j s|St |j t tfrt |d|j Stjdd}|||j |SdS) zRender the spinner for a given time. Args: time (float): Time in seconds. Returns: RenderableType: A renderable containing animation frame. Ng@@rrr r)padding)r rrr!r rintlenrr"rrrZassembler ZgridZadd_row)r#r.Zframe_noframetabler%r%r&r*;s.  zSpinner.render)rrr)rrrrcCs6|rt|trt|n||_|r(||_|r2||_dS)aUpdates attributes of a spinner after it has been started. Args: text (RenderableType, optional): A renderable to display at the right of the spinner (str or Text typically). Defaults to "". style (StyleType, optional): Style for spinner animation. Defaults to None. speed (float, optional): Speed factor for animation. Defaults to None. N)rrr rrrr")r#rrrr%r%r&update]s zSpinner.update)r) __name__ __module__ __qualname__rrrr'r+r r-r*r6r%r%r%r&r s: #  %r__main__)sleep)Columns)Panel)LivecCs$g|]}t|tt|dddqS)Zgreenr/)r)rr repr).0Z spinner_namer%r%r& {srAT)Z column_firstexpandZSpinnersZblue)titleZ border_style)Zrefresh_per_secondg?N)"typingrrrrrZ _spinnersrZmeasurer r5r rr r(r r rrrrrr7r.r;columnsr<Zpanelr=Zliver>sortedkeysZ all_spinnersr%r%r%r&s4     f