tPj("dZddlmZddlZddlmZmZddlmZddl m Z ddl m Z ddl mZdd lmZmZmZdd lmZdd lmZdd lmZGd dZGddeZGddZGddeZy)z Interfaces for for propagating feedback from the API to provide responsive progress indicators as well as a progress spinner implementation for use with CLI applications. ) annotationsN)ABCabstractmethod)Sequence) MemoryHandler)Any) StyleType)ConsoleGroupRenderableType)Live)Panel)SpinnercXeZdZdZgdd dZd d dZd dZd dZddZ dd Z y) AuditStatea An object that handles abstract "updates" to `pip-audit`'s state. Non-UI consumers of `pip-audit` (via `pip_audit`) should have no need for this class, and can leave it as a default construction in whatever signatures it appears in. Its primary use is internal and UI-specific: it exists solely to give the CLI enough state for a responsive progress indicator during user requests. )membersc||_y)zG Create a new `AuditState` with the given member list. N)_members)selfrs U/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/pip_audit/_state.py__init__zAuditState.__init__ s   NcJ|jD]}|j||y)z Called whenever `pip_audit`'s internal state changes in a way that's meaningful to expose to a user. `message` is the message to present to the user. N)r update_state)rmessagelogsmembers rrzAuditState.update_state's'mm /F    . /rcF|jD]}|jy)zB Called when `pip-audit`'s state is initializing. N)r initializerrs rrzAuditState.initialize2s# mm F     rcF|jD]}|jy)zE Called when `pip_audit`'s state is "done" changing. N)rfinalizer s rr"zAuditState.finalize:s!mm F OO  rc&|j|S)zb Create an instance of the `pip-audit` state for usage within a `with` statement. )rrs r __enter__zAuditState.__enter__As  rc$|jy)z Helper to ensure `finalize` gets called when the `pip-audit` state falls out of scope of a `with` statement. N)r")r _exc_type _exc_value_exc_tracebacks r__exit__zAuditState.__exit__Is  r)rzSequence[_StateActor]Nrstrr str | NonereturnNoner/r0)r/r)r'rr(rr)rr/r0) __name__ __module__ __qualname____doc__rrrr"r%r*rrrrsJ<>  / *-?B rrcDeZdZedddZeddZeddZy) _StateActorNctr+NotImplementedErrorrrrs rrz_StateActor.update_stateTs!!rct)z Called when `pip-audit`'s state is initializing. Implementors should override this to do nothing if their state management requires no initialization step. r:r$s rrz_StateActor.initializeX "!rct)z Called when the overlaying `AuditState` is "done," i.e. `pip-audit`'s state is done changing. Implementors should override this to do nothing if their state management requires no finalization step. r:r$s rr"z_StateActor.finalizear>rr+r,r1)r2r3r4rrrr"r6rrr8r8Ss<""""""rr8c~eZdZdZdZdddddd dd Zedd Z dd Zdd Z dd Z ddZ y) StatusLogz Displays a status indicator with an optional log panel to display logs for external processes. This code is based off of Rich's `Status` component: https://github.com/Textualize/rich/blob/master/rich/status.py Ndotszstatus.spinnerg?g)@)consolespinner spinner_stylespeedrefresh_per_secondct|||||_td|j|_t |j ||d|_y)a Construct a new `StatusLog`. `status` is the status message to display next to the spinner. `console` is the Rich console to display the log status in. `spinner` is the name of the spinner animation (see python -m rich.spinner). Defaults to `dots`. `spinner_style` is the style of the spinner. Defaults to `status.spinner`. `speed` is the speed factor for the spinner animation. Defaults to 1.0. `refresh_per_second` is the number of refreshes per second. Defaults to 12.5. )textstylerG)heightT)rDrH transientN)r_spinnerrLOG_PANEL_HEIGHT _log_panelr renderable_live)rstatusrDrErFrGrHs rrzStatusLog.__init__ysG* fMQVW 4+@+@A OO1   rc|jjr t|j|jS|jS)z Create a Rich renderable type for the log panel. If the log panel contains text, we should create a group and place the log panel underneath the spinner. )rQrRr rOr$s rrRzStatusLog.renderables0 ?? % %8 8}}rc|d}n-dj|j|j d}|jj |||j _|jj |j dy)z) Update status and logs. NrL )rJT)refresh)join splitlinesrPrOupdaterQrRrS)rrTrs rr[zStatusLog.updatesr <D99T__.0E0E/E/GHID &)%)" $//48rc8|jjy)z- Start the status animation. N)rSstartr$s rr]zStatusLog.starts rc8|jjy)z- Stop the spinner animation. N)rSstopr$s rr_zStatusLog.stops rc|jS)z4 Convert to a Rich renderable type. )rRr$s r__rich__zStatusLog.__rich__s r) rTr-rDzConsole | NonerEr-rFr rGfloatrHrb)r/r )rTr-rr.r/r0r1) r2r3r4r5rPrpropertyrRr[r]r_rar6rrrArAks #'#3$(      !   " <  999  9$rrAc4eZdZdZdddZd d dZd dZd dZy) AuditSpinnerzg A progress spinner for `pip-audit`, using `rich.status`'s spinner support under the hood. ct|_t||jdd|_t dt j dd|_g|_y)z0 Initialize the `AuditSpinner`. line)rDrErHrNF) flushLeveltarget flushOnClose) r _consolerArOrloggingERROR log_handler prev_handlers)rrs rrzAuditSpinner.__init__sL   ! T]]Fr  ) '--5 57rNc<|jj||y)z- Update the spinner's state. N)rOr[r<s rrzAuditSpinner.update_states Wd+rc*tj}|jD]}|jj ||jD]}|j ||j |j|jjy)z Redirect logging to an in-memory log handler so that it doesn't get mixed in with the spinner output. N) rmroothandlersrpappend removeHandler addHandlerrorOr])r root_loggerhandlers rrzAuditSpinner.initializesll "++ /G    % %g . /)) /G  % %g . / t//0 rc|jjtj}tj}|j tj tj|jj||jj|j|j|jD]}|j|y)z Cleanup the spinner output so it doesn't get combined with subsequent `stderr` output and flush any logs that were recorded while the spinner was active. N)rOr_rmrs StreamHandler setFormatter Formatter BASIC_FORMATro setTargetflushrvrprw)rrxstream_handlerrys rr"zAuditSpinner.finalizes ll  ..0##G$5$5g6J6J$KL "">2   !!$"2"23)) ,G  " "7 + ,r)rL)rr-r/r0r+r,r1)r2r3r4r5rrrr"r6rrreres 7$,*,rre)r5 __future__rrmabcrrcollections.abcrlogging.handlersrtypingr rich.alignr rich.consoler r r rich.liver rich.panelr rich.statusrrr8rArer6rrrsb ##$* 77;;|"#"0^^BF,;F,r