o $i(@sddlZddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZddlmZeeZddZGd d d Zd ejd efd dZeZiad ejd efddZdede efddZdededefddZ dS)N)datetime)DictOptional)Parse)message_to_dict)EventcCsdddtdDS)NcSsg|]}ttjqS)randomchoicestring hexdigits).0_r r \/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/_private/event/event_logger.py sz get_event_id..$)joinranger r r r get_event_idsrc@seZdZdejdejfddZddee e ffddZ d e fd d Z d e fd d Z d e fddZ d e fddZd e fddZd e fddZdejd e fddZdS)EventLoggerAdaptersourceloggercCs4||_||_t|_t|_t |_ i|_ dS)zAdapter for the Python logger that's used to emit events. When events are emitted, they are aggregated and available via state API and dashboard. This class is thread-safe. N) rrsocket gethostnamesource_hostnameosgetpid source_pid threadingLocklockglobal_context)selfrrr r r__init__s     zEventLoggerAdapter.__init__Nr"cCs:|j|sin||_WddS1swYdS)zuSet the global metadata. This method overwrites the global metadata if it is called more than once. N)r!r")r#r"r r rset_global_context,s"z%EventLoggerAdapter.set_global_contextmessagecK|jtjj|fi|dSN)_emitrSeverityTRACEr#r&kwargsr r rtrace4zEventLoggerAdapter.tracecKr'r()r)rr*DEBUGr,r r rdebug7r/zEventLoggerAdapter.debugcKr'r()r)rr*INFOr,r r rinfo:r/zEventLoggerAdapter.infocKr'r()r)rr*WARNINGr,r r rwarning=r/zEventLoggerAdapter.warningcKr'r()r)rr*ERRORr,r r rerror@r/zEventLoggerAdapter.errorcKr'r()r)rr*FATALr,r r rfatalCr/zEventLoggerAdapter.fatalseverityc Kst}t|_tt|_||_||_d|_ |j |_ |j |_ |j |_ |j}|j|jD]\}}|durA|durA|||<q1Wdn1sLwY|D]\}}|dure|dure|||<qU|jtt|dddd|jjddS)NrTF)$always_print_fields_with_no_presencepreserving_proto_field_nameuse_integers_for_enumsr)rrevent_idintrnow timestampr&r:labelr source_typerr custom_fieldsr!r"itemsrr3jsondumpsrhandlersflush)r#r:r&r-eventrDkvr r rr)Fs@ zEventLoggerAdapter._emitr()__name__ __module__ __qualname__r SourceTypeloggingLoggerr$rstrr%r.r1r3r5r7r9r*r)r r r rrsrrsink_dircCs~td}|tjt|d}|d|d}|jdd|jddt|}t d}| || |d|_ |S) N_ray_event_loggereventsevent_z.logT)exist_okz %(message)sF) rQ getLoggersetLevelr2pathlibPathmkdirtouch FileHandler Formatter setFormatter addHandler propagate)rrTrdir_pathfilepathhandler formatterr r r_build_event_file_loggerks        rhcCs\t"tj|}|tvrt||}t||t|<t|WdS1s'wYdS)apGet the event logger of the current process. There's only 1 event logger per (process, source). TODO(sang): Support more impl than file-based logging. Currently, the interface also ties to the file-based logging impl. Args: source: The source of the event. sink_dir: The directory to sink event logs. N)_event_logger_lockrrPName _event_loggerrhr)rrT source_namerr r rget_event_loggers  $rm event_strreturncCs4zt|tWStytd|YdSw)zParse an event from a string. Args: event_str: The string to parse. Expect to be a JSON serialized Event protobuf. Returns: The parsed event if parsable, else None zFailed to parse event: N)rr Exception global_logger exception)rnr r r parse_events   rsrJfilter_event_levelc Csbtjjdtjjdtjjdtjjdtjjdtjjdi}|}tj |}||j ||kr/dSdS) a*Filter an event based on event level. Args: event: The event to filter. filter_event_level: The event level string to filter by. Any events that are lower than this level will be filtered. Returns: True if the event should be filtered, else False. rTF) rr*r+r0r2r4r6r8upperValuer:)rJrt event_levelsr r rfilter_event_by_levels  r})!rFrQrr[r rr rrtypingrrgoogle.protobuf.json_formatrray._private.protobuf_compatrray.core.generated.event_pb2rrYrMrqrrrPrSrhr rirkrmrsboolr}r r r rs,     S