o ;i4C@sUdZddlZddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z m Z mZddlmZe r=ddlmZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeZGdddeZeZeaeed<dS)aOutput management interface for Modal CLI. This module defines the interface for output management and provides a no-op implementation for when output is disabled. The rich-based implementation lives in rich.py to avoid importing rich when it's not needed. N)ABCabstractmethod) Generator)AbstractContextManager) TYPE_CHECKINGAnyCallableOptional) write_to_fd)api_pb2c@sfeZdZdZedddZedddZededdfd d Zedd d Z ed e ddfddZ dS) StatusContextzWAbstract base class for status context managers that support manual start/stop control.returnNcCdS)z!Start showing the status spinner.NselfrrI/home/ubuntu/.local/lib/python3.10/site-packages/modal/_output/manager.pystartzStatusContext.startcCr)z Stop showing the status spinner.NrrrrrstoprzStatusContext.stopstatuscCr)zUpdate the status message.Nrrrrrrupdate$rzStatusContext.updatecCdSNrrrrr __enter__)zStatusContext.__enter__argscGrrrrrrrr__exit__,rzStatusContext.__exit__r N)r r ) __name__ __module__ __qualname____doc__rrrstrrrrrrrrrr s   r c@sJeZdZdZededdfddZed d d Zededdfd d ZdS) StatusRowzHAbstract base class describing a row in the object creation status tree.messager NcCr)z.Update the primary message shown for this row.Nrrr'rrrr'3rzStatusRow.messagewarningapi_pb2.WarningcCr)z2Append a warning message associated with this row.Nrrr)rrrwarn8rzStatusRow.warncCr)z0Mark the row as finished with the given message.Nrr(rrrfinish=rzStatusRow.finishr)r*r N) r!r"r#r$rr%r'r,r-rrrrr&0s r&c@s>eZdZdZdeddfddZd d d Zdeddfd d ZdS)DisabledStatusRowz-No-op StatusRow used when output is disabled.r'r NcCrrrr(rrrr'FrzDisabledStatusRow.messager)r*cCrrrr+rrrr,IrzDisabledStatusRow.warncCrrrr(rrrr-LrzDisabledStatusRow.finishr.)r!r"r#r$r%r'r,r-rrrrr/Cs  r/c@s\eZdZdZe      d dedeedeedeedee d ee d efd d Z dS)TransferProgressContextz;Abstract base class for transfer progress tracking context.NFtask_idadvancenamesizeresetcompleter cCr)z:Update progress. Returns task_id when creating a new task.Nrrr1r2r3r4r5r6rrrprogressS z TransferProgressContext.progressNNNNFF) r!r"r#r$rrr floatr%boolr8rrrrr0Ps0r0c@sXeZdZdZ      d dedeedeedeedeed eed dfd d Z dS)_DisabledTransferProgresszd?Z&e d@edAeedeeegd ffdBdCZ'e dDedEedd fdFdGZ(e dHedIdJdd fdKdLZ)e dMedNedAeedOeedd f dPdQZ*e dldTdUZ+dldVdWZ,e djdXdYZ-e dZede"d[fd\d]Z.e djd^d_Z/d`edd fdadbZ0dcedd fdddeZ1e d1edd fdfdgZ2d S)m OutputManageraAbstract base class defining the interface for output management. This class allows for different implementations: - RichOutputManager: Full rich-based terminal output with progress spinners, trees, etc. - DisabledOutputManager: No-op implementation for when output is disabled. Subclasses must implement all abstract methods. Use OutputManager.get() to get the current output manager instance. F _quiet_mode_show_timestampsr cCstS)aIGet the current output manager. Returns a RichOutputManager when output is enabled, otherwise returns a DisabledOutputManager that provides no-op implementations of all methods. This allows code to call output methods without checking if output is enabled, simplifying the calling code. _current_output_manager)clsrrrget zOutputManager.getmanagerNcCs|adS)zCSet the current output manager. Used internally by enable_output().NrA)rCrFrrr_setszOutputManager._setcCr)zWhether output is currently being displayed somewhere. Returns False if using DisabledOutputManager OR if quiet mode is active. This reflects whether print(), status(), etc. will actually produce visible output. Nrrrrr is_enabledszOutputManager.is_enabledcCr)z4Whether the output is connected to a terminal (TTY).Nrrrrr is_terminalzOutputManager.is_terminalcCr)zWhether to show image logs.Nrrrrr_show_image_logsrJzOutputManager._show_image_logsNNNcCr)z>Context manager that displays a tree of objects being created.Nrrrrrdisplay_object_treerJz!OutputManager.display_object_treecCr)z,Add a status row to the current object tree.Nrrrrradd_status_rowrzOutputManager.add_status_rowT  stderr highlightstylesepend renderablerRrSrTrUrVcCr)aPrint a renderable to the console. Args: renderable: The content to print. stderr: If True, print to stderr instead of stdout. highlight: If True, apply syntax highlighting. style: Optional Rich style string (e.g., "green", "bold cyan"). sep: The separator to use between items. end: The string to use at the end of the output. NrrrWrRrSrTrUrVrrrprintszOutputManager.print error_textcCr)afPrint an error message to stderr, ignoring quiet/disabled mode. This method always prints the error message regardless of quiet mode or whether output is disabled. It should be used for critical error messages that must always be visible to the user. Args: error_text: The error message text to display. NrrrZrrr print_errorr9zOutputManager.print_errorr)categoryfilenamelinenobase_showwarning.NcCr)aDisplay a warning message. This method is called by the patched warnings.showwarning for all warnings. Args: warning: The warning instance. category: The warning category (class). filename: The source file where the warning originated. lineno: The line number where the warning originated. base_showwarning: The original warnings.showwarning function to fall back to. Nrrr)r]r^r_r`rrr show_warningszOutputManager.show_warningdatacCr)z Print JSON data with formatting.Nrrrdrrr print_jsonrzOutputManager.print_jsonr'r cCr)zContext manager that displays a status spinner with a message. Returns a context manager that shows a spinner while active. The returned object has start(), stop(), and update() methods for manual control. Nrr(rrrrzOutputManager.statuscCr)zContext manager that shows a live spinner with a message. This combines spinner creation and live display into a single interface, hiding the implementation details of how spinners are rendered. Nrr(rrrmake_live_spinnerrgzOutputManager.make_live_spinnercCr)zEnable showing image logs.Nrrrrrenable_image_logsrzOutputManager.enable_image_logsRunning app... status_textcCr)zContext manager that shows a status spinner. Args: status_text: The text to display next to the spinner. Defaults to "Running app...". Nrrrkrrrshow_status_spinnerrgz!OutputManager.show_status_spinner app_page_urlcCr)z$Update the app page URL for display.Nrrrnrrrupdate_app_page_urlrz!OutputManager.update_app_page_urltagtotalcCr)z.Get a callback for updating function progress.Nr)rrqrrrrrfunction_progress_callbackrz(OutputManager.function_progress_callbackr1statecCr)zUpdate the state of a task.Nrrr1rtrrrupdate_task_staterzOutputManager.update_task_stateimage_id task_progressapi_pb2.TaskProgresscCr)z Update snapshot upload progress.Nrrrwrxrrrupdate_snapshot_progress rz&OutputManager.update_snapshot_progress function_id completed descriptioncCr)z"Update function queueing progress.Nrrr|r}rrr~rrrupdate_queueing_progress%sz&OutputManager.update_queueing_progresslogapi_pb2.TaskLogscsdS)a}Process and display log content. Note: In RichOutputManager, log output is always displayed even when quiet mode is enabled. This is intentional - quiet mode suppresses progress indicators and status messages, but not actual log output from running functions/images. In contrast, DisabledOutputManager suppresses all output including logs. Nrrrrrrput_log_content,s zOutputManager.put_log_contentcstj}t|drw|jd}d}z t|dr|}Wn ty'd}Ynw|dur6t||IdHdSd}d}|t|kruz||j ||d7}| Wnt yl|dkr^|d7}t dIdHYnw|t|ks@dSdS| |j| dS) a#Write PTY content to stdout for interactive terminal sessions. This handles raw PTY output from interactive shells/debuggers, writing directly to stdout with proper handling of non-blocking writes. Note: This method is implemented in the base class (not abstract) because interactive/raw terminal content is always forwarded to stdout regardless of whether output is enabled or quiet mode is active. This ensures that interactive debugging sessions (e.g., pdb breakpoints) work correctly. bufferzutf-8Nfilenorg?)sysstdouthasattrrdencoder Exceptionr lenrwriteflushBlockingIOErrorasynciosleep)rrrrdfdwritten n_retriesrrrput_pty_content7s:        zOutputManager.put_pty_contentcCr)zFlush any buffered output.Nrrrrr flush_lines`rzOutputManager.flush_linestyper0cCr)zContext manager for tracking file transfer progress. Args: type: Either "download" or "upload". Returns: A context manager that yields a TransferProgressContext with a progress() method. Nrrrrrrtransfer_progresserEzOutputManager.transfer_progresscCr)zStop the status spinner if it's running. This is used to cleanly stop the spinner before entering PTY mode or other situations where the spinner would interfere with output. Nrrrrrstop_status_spinnerqrgz!OutputManager.stop_status_spinnerquietcC ||_dS)aEnable or disable quiet mode. When quiet mode is enabled: - All print() output is suppressed, including stderr - Progress indicators (spinners, progress bars) are suppressed - Warnings (via show_warning) are still displayed - Log output (via put_log_content) is still displayed - Interactive/raw PTY content (via put_pty_content) is still forwarded to stdout - Error output (via print_error) is still displayed Note: This differs from DisabledOutputManager, which suppresses all output including logs. Quiet mode is meant for reducing noise while still showing relevant content. N)r?)rrrrrset_quiet_modezs zOutputManager.set_quiet_modeshowcCr)z2Enable or disable timestamp display in log output.N)r@)rrrrrset_timestamps zOutputManager.set_timestampscCr)z;Print a step completed message with appropriate formatting.Nrr(rrrstep_completedrzOutputManager.step_completed)r r>)rFr>r Nr rjrrr N)3r!r"r#r$r?r<__annotations__r@ classmethodrDrGpropertyrrHrIrK contextlibcontextmanagerrrMr&rNrr r%rYr\Warningrintrrcrfrrrhrirmrprsrvr{rrrrrrrrrrrrrr>ps        (   )  r>c@sReZdZdZdddZdddZdeddfd d Zdd d Zd e ddfddZ dS)_DisabledStatusz9No-op status context manager for when output is disabled.r NcCrrrrrrrrrz_DisabledStatus.startcCrrrrrrrrrz_DisabledStatus.stoprcCrrrrrrrrrz_DisabledStatus.updatecCs|Srrrrrrrrz_DisabledStatus.__enter__rcGrrrrrrrrrz_DisabledStatus.__exit__r )r r) r!r"r#r$rrr%rrrrrrrrrs   rc@s$eZdZdZedefddZedefddZedefddZe j de d fd d Z de fd d ZdddddddedededeedededdfddZdeddfddZded eed!ed"ed#ed$ddf d%d&Zd'eddfd(d)Zd*edefd+d,Ze j d*ede d fd-d.ZdWd/d0Ze j dXd2ede d fd3d4Zd5eddfd6d7Zd8ed9eedeeegdffd:d;Z dd?Z!d@edAdBddfdCdDZ"dEedFed9eedGeeddf dHdIZ#dYdLdMZ$dWdNdOZ%e j dPede e&ddffdQdRZ'dWdSdTZ(d*eddfdUdVZ)dS)ZDisabledOutputManageraNo-op implementation of OutputManager for when output is disabled. Most methods are no-ops that do nothing, allowing code to call output methods without checking if the output manager exists. Exceptions: - Interactive/raw PTY content (via put_pty_content) is always forwarded to stdout to ensure interactive debugging sessions work correctly. - Warnings (via show_warning) fall back to Python's default warning display. r cCr)NFrrrrrrHz DisabledOutputManager.is_enabledcCs tjSr)rrisattyrrrrrIrz!DisabledOutputManager.is_terminalcCr)NTrrrrrrKrz&DisabledOutputManager._show_image_logsrLcc dVdSrrrrrrrM z)DisabledOutputManager.display_object_treecCtSr)r/rrrrrNz$DisabledOutputManager.add_status_rowFTNrOrPrQrWrRrSrTrUrVcCrrrrXrrrrYrEzDisabledOutputManager.printrZcCstj|ddS)NrP)rrRrr[rrrr\sz!DisabledOutputManager.print_errorr)r]r^r_r`racCs|||||ddddS)N)filelinerrbrrrrcs z"DisabledOutputManager.show_warningrdcCrrrrerrrrfrz DisabledOutputManager.print_jsonr'cCrr)rr(rrrrrzDisabledOutputManager.statusccrrrr(rrrrhrz'DisabledOutputManager.make_live_spinnercCrrrrrrrrirz'DisabledOutputManager.enable_image_logsrjrkccrrrrlrrrrmrz)DisabledOutputManager.show_status_spinnerrncCrrrrorrrrprz)DisabledOutputManager.update_app_page_urlrqrrcCsdtdtddfdd}|S)Nr}rrr cSrrr)r}rrrrrnooprz>DisabledOutputManager.function_progress_callback..noop)r)rrqrrrrrrrssz0DisabledOutputManager.function_progress_callbackr1rtcCrrrrurrrrvrz'DisabledOutputManager.update_task_staterwrxrycCrrrrzrrrr{rz.DisabledOutputManager.update_snapshot_progressr|r}r~cCrrrrrrrrrz.DisabledOutputManager.update_queueing_progressrrcsdSrrrrrrrsz%DisabledOutputManager.put_log_contentcCrrrrrrrr rz!DisabledOutputManager.flush_linesrccstVdS)z(No-op transfer progress context manager.N)r=rrrrr s z'DisabledOutputManager.transfer_progresscCrrrrrrrrrz)DisabledOutputManager.stop_status_spinnercCrrrr(rrrrrz$DisabledOutputManager.step_completedr rr)*r!r"r#r$rr<rHrIrKrrrrMr&rNrr r%rYr\rrrrrcrfr rrhrirmrprsrvr{rrrr0rrrrrrrrs      &   rrB)r$rrrabcrrcollections.abcrrtypingrrrr modal._utils.shell_utilsr modal_protor r r&r/r0r=r>rr_DISABLED_OUTPUT_MANAGERrBrrrrrs.     'r