blob: e5f4a22bd96e6ef3a327b962681bf6a9a76f5804 (
about) (
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
"""
Basic Flag and Flags data structures.
"""
from __future__ import annotations
from collections.abc import Iterable, Iterator, MutableSet
from typing import NamedTuple
class Flag(NamedTuple):
name: str
bit: int
class Flags(MutableSet): # type: ignore
"""
A simple MutableSet implementation that will only accept known flags as
elements.
Will behave like a regular set(), except that a ValueError will be thrown
when .add()ing unexpected flags.
"""
def __init__(self, defined_flags: Iterable[Flag]) -> None:
self._valid_flags = {flag.name for flag in defined_flags}
self._flags: set[str] = set()
def __repr__(self) -> str:
return repr(sorted(self._flags))
def __contains__(self, x: object) -> bool:
return self._flags.__contains__(x)
def __iter__(self) -> Iterator[str]:
return self._flags.__iter__()
def __len__(self) -> int:
return self._flags.__len__()
def discard(self, value: str) -> None:
return self._flags.discard(value)
def add(self, value: str) -> None:
if value not in self._valid_flags:
msg = f"Unexpected flag: {value}. Valid flags are: {self._valid_flags}"
raise ValueError(msg)
return self._flags.add(value)
|
