@@ -41,7 +41,7 @@ def format(self, record: logging.LogRecord) -> str:
4141 trunc : int = message_size + len (suffix ) - 32766
4242 msg .message = msg .message [:- trunc ] + suffix
4343
44- return LOGGING_PREFIX + msg .json (by_alias = True )
44+ return LOGGING_PREFIX + msg .model_dump_json (by_alias = True )
4545
4646 def to_level (self , levelname : str ) -> int :
4747 if levelname == "CRITICAL" :
@@ -67,7 +67,9 @@ def create_message(self, record: logging.LogRecord) -> EnoLogMessage:
6767 type = "infrastructure" ,
6868 severity = record .levelname ,
6969 severity_level = self .to_level (record .levelname ),
70- timestamp = datetime .datetime .utcnow ().strftime ("%Y-%m-%dT%H:%M:%S.%fZ" ),
70+ timestamp = datetime .datetime .now (datetime .timezone .utc ).strftime (
71+ "%Y-%m-%dT%H:%M:%S.%fZ"
72+ ),
7173 message = record .msg ,
7274 module = record .module ,
7375 function = record .funcName ,
@@ -81,4 +83,7 @@ def create_message(self, record: logging.LogRecord) -> EnoLogMessage:
8183 flag = getattr (checker_task , "flag" , None ),
8284 variant_id = getattr (checker_task , "variant_id" , None ),
8385 task_chain_id = getattr (checker_task , "task_chain_id" , None ),
86+ flag_regex = getattr (checker_task , "flag_regex" , None ),
87+ flag_hash = getattr (checker_task , "flag_hash" , None ),
88+ attack_info = getattr (checker_task , "attack_info" , None ),
8489 )
0 commit comments