blob: 9b25a8182de27ef713f3009f7369107189bd2111 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Generic registry class."""
from __future__ import annotations
from typing import Generic
from typing import TypeVar
T = TypeVar("T")
class Registry(Generic[T]):
"""Generic registry for name-config pairs."""
def __init__(self) -> None:
"""Create an empty registry."""
self.items: dict[str, T] = {}
def __str__(self) -> str:
"""List all registered items."""
return "\n".join(
f"- {name}: {item}"
for name, item in sorted(self.items.items(), key=lambda v: v[0])
)
def register(self, name: str, item: T) -> bool:
"""Register an item: returns `False` if already registered."""
if name in self.items:
return False # already registered
self.items[name] = item
return True
|
