NdZddlmZddlmZddlmZddlZddlZddlZddl Zddl Zddl Zddl Zddl ZddlZddlZddlZddlmZddlmZddlmZddlZd Zd Zd Zd ZGd deZdZGddeZy)zA library for logs tailing.)absolute_import)division)unicode_literalsN)apis) gapic_util)logzMhttps://cloud.google.com/logging/docs/reference/tools/gcloud-logging#tailing.c tjjdtjjdtjjdtjj dtjj di}|jtjjk(ry|||dj|j|jd|jy) z(Handles _MultiThreadedRendezvous errors.zInvalid argument.z&There are too many tail sessions open.zInternal error.z-Access is denied or has changed for resource.zTThe maximum duration for tail has been met. The command may be repeated to continue.Nz{} ({})zUnknown error encountered.) grpc StatusCodeINVALID_ARGUMENTRESOURCE_EXHAUSTEDINTERNALPERMISSION_DENIED OUT_OF_RANGEcode CANCELLEDformatgetdetails) rendezvous output_debugoutput_warningerror_messages_by_codes -lib/googlecloudsdk/api_lib/logging/tailing.py_HandleGrpcRendezvousr0s oo&&  oo(( 2 oo  oo'' 9 oo"" 6 __$//333 z!!  !2!=?cPtjdd}|jjj}|j j d|j jdi}|jD]/\}}|j|dj|}|||1y)zHandles supression counts.loggingv2zLogging API backend rate limitz,client not consuming messages quickly enoughzUNKNOWN REASON: {}N) rGetGapicClientClasstypesTailLogEntriesResponseSuppressionInfoReason RATE_LIMIT NOT_CONSUMEDitemsrr)counts_by_reasonhandler client_classsuppression_infosuppression_reason_stringsreasoncount reason_strings r_HandleSuppressionCountsr1Ks)))T:,"((??OO(( *** 8   (--/mfe.22$++F35M M5!0rc@eZdZdZdZdZdZdZdZdZ dZ d Z y ) _SuppressionInfoAccumulatorzKAccumulates and outputs information about suppression for the tail session.c||_||_||_tj|_tj|_||_yN)_get_now_warning_error collectionsCounter_count_by_reason_delta_count_by_reason_cumulative _last_flush)selfget_nowr output_errors r__init__z$_SuppressionInfoAccumulator.__init___sCDM"DMDK"-"5"5"7D'2':':'s r_OutputSuppressionHelpMessagez9_SuppressionInfoAccumulator._OutputSuppressionHelpMessagegsMM.55oFHrcf|j|jz jtkDSr5)r6r= total_seconds&_SUPPRESSION_INFO_FLUSH_PERIOD_SECONDSrDs r _ShouldFlushz(_SuppressionInfoAccumulator._ShouldFlushks, MMOd.. . ]_E FFrcF|jdj||y)Nz Suppressed {} entries due to {}.)r8rr>r0r/s r_OutputSuppressionDeltaMessagez:_SuppressionInfoAccumulator._OutputSuppressionDeltaMessageosKK299%OPrcF|jdj||y)Nz+In total, suppressed {} messages due to {}.)r7rrKs r#_OutputSuppressionCumulativeMessagez?_SuppressionInfoAccumulator._OutputSuppressionCumulativeMessagers!MM?FF }rc|j|_t|j|j|xj |jz c_|jj yr5)r6r=r1r;rLr<clearrDs r_Flushz"_SuppressionInfoAccumulator._FlushvsS}}DT88!@@B$$(C(CC$%%'rc|jt|j|j|jr|j yyr5)rQr1r<rNrErDs rFinishz"_SuppressionInfoAccumulator.Finish}s?KKMT==!EEG '' ((*(rc |xjtj|Dcic]}|j|jc}z c_|j r|j yycc}wr5)r;r9r:r.suppressed_countrIrQ)r>r,infos rAddz_SuppressionInfoAccumulator.Addse;#6#68HI8Hd++ +8HI$KK  kkm JsA. N) __name__ __module__ __qualname____doc__rArErIrLrNrQrSrWrrr3r3\s0S!HFQ(+rr3c#K|jt|||}d}|jrN |j}|j |j|jD]} | |jrN|r t||||j|jy#tj $r }|}Yd}~Jd}~wwxYww)aStreams entries back from the Logging API. Args: get_now: A callable that returns the current time. output_warning: A callable that outputs the argument as a warning. output_error: A callable that outputs the argument as an error. output_debug: A callable that outputs the argument as debug info. tail_stub: The `BidiRpc` stub to use. Yields: Entries included in the tail session. N) openr3 is_activerecvr RpcErrorrWr,entriesrrSclose) r?rr@r tail_stubsuppression_info_accumulatorerrorresponseeentrys r_StreamEntriesrjs ..!< ~|"- % !h!$$X%>%>?!! k"  %~>%%' // == e  s3,CB,DKDNrNc#K|jjj}|jj |||_t j|j|jjjj||_ |rtj||_t|||||jD]} | yw)aTails log entries from the Cloud Logging API. Args: resource_names: The resource names to tail. logs_filter: The Cloud Logging filter identifying entries to include in the session. buffer_window_seconds: The amount of time that Cloud Logging should buffer entries to get correct ordering, or None if the backend should use its default. output_warning: A callable that outputs the argument as a warning. output_error: A callable that outputs the argument as an error. output_debug: A callable that outputs the argument as debug. get_now: A callable that returns the current time. Yields: Entries for the tail session. )initial_request)secondsN)ror"TailLogEntriesRequestresource_namesextendfilterr MakeBidiRpcr transporttail_log_entriesrddatetime timedelta buffer_windowrj) r>rt logs_filterbuffer_window_secondsrr@rr?requestris rTailLogszLogTailer.TailLogss2kk557G !!.1 GN++ T[[((22CC!DN&009NOg ,dnn> k>sCCcR|jr|jjyyr5)rdrcrDs rStopzLogTailer.Stops ~~ nnr)rXrYrZr[rArerrPrintrfdebugrznowrrr\rrrlrlsB &*!ggmmIIII((,,$Lrrl)r[ __future__rrrr9rz+google.appengine.logging.v1.request_log_pb2google.google.cloud.appengine_v1.proto.audit_data_pb23google.cloud.appengine_v1alpha.proto.audit_data_pb22google.cloud.appengine_v1beta.proto.audit_data_pb25google.cloud.bigquery_logging_v1.proto.audit_data_pb2,google.cloud.cloud_audit.proto.audit_log_pb2.google.cloud.iam_admin_v1.proto.audit_data_pb2$google.iam.v1.logging.audit_data_pb2google.type.money_pb2googlecloudsdk.api_lib.utilrgooglecloudsdk.corerrr rHrCrr1objectr3rjrlr\rrrsz"&'35:9<35+-*# *+&a6"",&,^"J//r