tni: ddlZddlZddlmZddlmZddlmZmZddlZddl m Z ddl m Z ddl mZmZmZmZddlmZmZdd lmZdd lmZejeZd d d d ddddZejdZdedefdZ dedefdZ!de dede"dede#ddf dZ$dej%dddd d!"ej%d#d$d%d&d!e 'ej%d!d(d)d*+fd,eeeej&d-+fde"dede#ddf d.Z'dS)/N)datetime)Path) AnnotatedOptional)escape) RichToolkit) APIClient AppLogEntryStreamLogErrorTooManyRetriesError) AppConfigget_app_config)Identity)get_rich_toolkitbluecyanyellowredmagenta)debuginfowarningwarnerrorcriticalfatalz ^\d+[smhd]$valuereturncbt|stjd|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)rs \/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/fastapi_cloud_cli/commands/logs.py_validate_sincer%%s7   u % %  ^    LlogcVtj|jdd}|ddddz}t |j}t|j }|rd|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_strcolorr4s r$_format_log_liner7/s&s}'<'>ssCcIM  !2!2 3 3ES[!!G NM5MMMMmMMGMMM A- A A A AAr&toolkit app_configtailsincefollowcd} t5}||j|||D])}|t ||dz }*|s)|dkr#|d ddddSddddS#1swxYwYdS#t $r|YdSt$r}t|} d| vsd| vr|dn?d | vr|d n%|d t| tj ddd}~wttf$r+|d tj ddwxYw) z.Stream app logs and print them to the console.r)app_idr:r;r<z+No logs found for the specified time range.NzHTTP 401zHTTP 403zWThe specified token is not valid. Use [blue]`fastapi login`[/] to generate a new token.zHTTP 404z4App 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>printr7KeyboardInterrupt print_liner strrr"Exitr TimeoutError) r8r9r:r;r< log_countclientr'e error_msgs r$_process_log_streamrK>sVI$& [[ F--!( .    .s33444Q  i1nn KLLL                            &&&FF  " "jI&=&= MMi    9 $ $ MMF     MM5&"3"355   jmm%  .&&& D   jmm% &sHB A'B9 B  B BB BB F  F  BE  ?F dz--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)ct}td5}|s+|ddt jd|pt j}t|}|s)|dt jdt d|j |r!|d |j d d n |d |j d d | t|||||ddddS#1swxYwYdS)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)minimalzrCrK)rQr:r;r<identityr8app_pathr9s r$rVrVosHzzH $ ' ' '% 7$$&& MMN    *Q-- %48::#H--  MMW   *Q--  3Z5FGGG   MMZJ,=ZZZ      MMH:+<HHH     !     ?% % % % % % % % % % % % % % % % % % sDEE  E )(loggingrerpathlibrtypingrrr" rich.markupr rich_toolkitrfastapi_cloud_cli.utils.apir r r r fastapi_cloud_cli.utils.appsr rfastapi_cloud_cli.utils.authrfastapi_cloud_cli.utils.clir getLogger__name__rYr0compiler rDr%r7intboolrKOptionArgumentrVr&r$rns &&&&&&&& $$$$$$ CBBBBBBB111111888888  8 $ $     >** 33 B+ B# B B B B.& .&.& .&  .&  .&  .&.&.&.&n    <   H  5<  L -J J X     J  J  J , -J 8 9J J J J J J r&