o *i @sddlZddlZddlmZddlmZddlmZmZddlZddl m Z ddl m Z ddl mZddlmZmZmZmZdd lmZmZdd lmZdd lmZmZeeZd d ddddddZe dZ!de"de"fddZ#dede"fddZ$dedede%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/fd0eeeej)d1d/fde%de"de&ddf d2d3Z*dS)4N)datetime)Path) AnnotatedOptional) 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$\/home/ubuntu/veenaModal/venv/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_strcolorr3r$r$r%_format_log_line0s r6toolkit 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_idr9r:r;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.)r stream_app_logsr<printr6KeyboardInterrupt print_liner status_code isinstance __cause__rrrstrr"Exitr TimeoutError) r7r8r9r:r; log_countclientr'er3r$r$r%_process_log_stream?sJ      &       rKdz--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').)rMrNcallbackz--follow/--no-followz-fz=Stream logs in real-time (use --no-follow to fetch and exit).)rMpathzEPath 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)minimalzs          6