blob: e7faa2be7ba57c89b0698d10b77d8088b757d771 [file] [log] [blame]
# Copyright 2025 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
class MetadataDictionary(dict):
"""
This is a very simple class that prints out a textproto using a dictionary.
Realistically, we should not be re-inventing the wheel as we are doing here
and we should be using protobuf instead.
TODO(b/360322121): Use protobuf generated classes instead of this.
"""
def __init__(self, field_name):
super().__init__()
self.field_name = field_name
def _as_string(self, dict_items, width=2, depth=1):
string = self.field_name + " {\n"
for (key, value) in dict_items:
if not isinstance(value, MetadataDictionary):
string += (" " * width * depth) + f"{key}: {value}\n"
else:
string += (" " * width * depth) + value._as_string(
value.items(), width, depth + 1)
string += (" " * width * (depth - 1)) + "}\n"
return string
def __repr__(self):
return self._as_string(self.items())