blob: f5de7d98c56991d7fc159bf1a8cc84e6327325d7 [file] [log] [blame]
# Copyright 2021 The Pigweed Authors
# Licensed under the Apache License, Version 2.0 (the "License"); you may not
# use this file except in compliance with the License. You may obtain a copy of
# the License at
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations under
# the License.
"""Tools for working with tokenized logs."""
from dataclasses import dataclass, asdict
import re
from typing import Dict, Mapping, Iterator
def _mask(value: int, start: int, count: int) -> int:
mask = (1 << count) - 1
return (value & (mask << start)) >> start
class Metadata:
"""Parses the metadata payload used by pw_log_tokenized."""
value: int
log_bits: int = 3
line_bits: int = 11
flag_bits: int = 2
module_bits: int = 16
def __post_init__(self):
self.log_level = _mask(self.value, 0, self.log_bits)
self.line = _mask(self.value, self.log_bits, self.line_bits)
self.flags = _mask(self.value, self.log_bits + self.line_bits,
self.module_token = _mask(
self.value, self.log_bits + self.line_bits + self.flag_bits,
def __iter__(self):
return iter(asdict(self).items())
def __dict__(self):
return asdict(self)
class FormatStringWithMetadata:
"""Parses metadata from a log format string with metadata fields."""
_FIELD_KEY = re.compile(r'■([a-zA-Z]\w*)♦', flags=re.ASCII)
def __init__(self, string: str) -> None:
self.raw_string = string
self.fields: Dict[str, str] = {}
# Only look for fields if the raw string starts with one.
if self._FIELD_KEY.match(self.raw_string):
fields = self._FIELD_KEY.split(self.raw_string)[1:]
for name, value in zip(fields[::2], fields[1::2]):
self.fields[name] = value
def message(self) -> str:
"""Displays the msg field or the whole string if it is not present."""
return self.fields.get('msg', self.raw_string)
def module(self) -> str:
return self.fields.get('module', '')
def file(self) -> str:
return self.fields.get('file', '')
def __repr__(self) -> str:
return self.message