o d۷i @sddlZddlZddlmZddlmZddlmZddlZddlm Z ddl m Z ddl m Z ddlmZmZmZmZdd lmZmZdd lmZdd lmZmZeeZd d ddddddZedZ de!de!fddZ"dede!fddZ#de dede$de!de%ddf dd Z&dej'd!d"d#d$d%d&ej'd'd(d)d*d%e"d+ej'd%d,d-d.d/fd0eedBej(d1d/fde$de!de%ddf d2d3Z)dS)4N)datetime)Path) Annotated) HTTPError)escape) RichToolkit) APIClient AppLogEntryStreamLogErrorTooManyRetriesError) AppConfigget_app_config)Identity)get_rich_toolkithandle_http_errorbluecyanyellowredmagenta)debuginfowarningwarnerrorcriticalfatalz ^\d+[smhd]$valuereturncCst|s td|S)z&Validate the --since parameter format.zPInvalid format. Use a number followed by s, m, h, or d (e.g., '5m', '1h', '2d').) SINCE_PATTERNmatchtyper BadParameter)rr#U/home/ubuntu/vllm_env/lib/python3.10/site-packages/fastapi_cloud_cli/commands/logs.py_validate_since&s r%logcCstt|jdd}|dddd}t|j}t |j }|r2d|d|d|d |Sd |d |S) z7Format a log entry for display with a colored indicatorZz+00:00z%Y-%m-%dT%H:%M:%S.%fN[u]┃[/z] [dim]z[/dim] u[dim]┃[/dim] [dim]) r fromisoformat timestampreplacestrftimeLOG_LEVEL_COLORSgetlevellowerrmessage)r&r+ timestamp_strcolorr2r#r#r$_format_log_line0s r5toolkit app_configtailsincefollowc CsJd}zNt@}|j|j|||dD] }|t||d7}q|s4|dkr=|d WdWdSWdWdSWdWdS1sIwYWdSty^|YdSty}z(|jdkrmd} nt |j t ryt |j } n dt t|} || tddd}~wttfy|d tddw) z.Stream app logs and print them to the console.r)app_idr8r9r:z+No logs found for the specified time range.Niz4App not found. Make sure to use the correct account.z[red]Error:[/] z6Lost connection to log stream. Please try again later.)rstream_app_logsr;printr5KeyboardInterrupt print_liner status_code isinstance __cause__rrrstrr!Exitr TimeoutError) r6r7r8r9r: log_countclientr&er2r#r#r$_process_log_stream?sJ      &       rJdz--tailz-tz-Number of log lines to show before streaming.T)help show_default5mz--sincez-sz9Show logs since a specific time (e.g., '5m', '1h', '2d').)rLrMcallbackz--follow/--no-followz-fz=Stream logs in real-time (use --no-follow to fetch and exit).)rLpathzEPath to the folder containing the app (defaults to current directory)cCst}tdd^}|s|jdddtd|pt}t|}|s/|dtdt d|j |rE|jd |j d d dn |jd |j d d d| t |||||dWddS1siwYdS)a*Stream or fetch logs from your deployed app. Examples: fastapi cloud logs # Stream logs in real-time fastapi cloud logs --no-follow # Fetch recent logs and exit fastapi cloud logs --tail 50 --since 1h # Last 50 logs from the past hour T)minimalzr!rErcwdr loggerrr;r@rJ)rPr8r9r:identityr6app_pathr7r#r#r$rTosD$      "rT)*loggingrerpathlibrtypingrr!httpxr rich.markupr rich_toolkitrfastapi_cloud_cli.utils.apirr r r fastapi_cloud_cli.utils.appsr r fastapi_cloud_cli.utils.authrfastapi_cloud_cli.utils.clirr getLogger__name__rWr.compilerrDr%r5intboolrJOptionArgumentrTr#r#r#r$s           6