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 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89
|
from __future__ import annotations
import re
import typing as ty
from dataclasses import dataclass
import flexparser.flexparser as fp
from . import common, errors
@dataclass(frozen=True)
class Rule(fp.ParsedStatement):
new_unit_name: str
old_unit_name: ty.Optional[str] = None
@classmethod
def from_string(cls, s: str) -> fp.NullableParsedResult[Rule]:
if ":" not in s:
return cls(s.strip())
parts = [p.strip() for p in s.split(":")]
if len(parts) != 2:
return errors.DefinitionSyntaxError(
f"Exactly two terms expected for rule, not {len(parts)} (`{s}`)"
)
return cls(*parts)
@dataclass(frozen=True)
class BeginSystem(fp.ParsedStatement):
"""Being of a system directive.
@system <name> [using <group 1>, ..., <group N>]
"""
#: Regex to match the header parts of a context.
_header_re = re.compile(r"@system\s+(?P<name>\w+)\s*(using\s(?P<used_groups>.*))*")
name: str
using_group_names: ty.Tuple[str, ...]
@classmethod
def from_string(cls, s: str) -> fp.NullableParsedResult[BeginSystem]:
if not s.startswith("@system"):
return None
r = cls._header_re.search(s)
if r is None:
raise ValueError("Invalid System header syntax '%s'" % s)
name = r.groupdict()["name"].strip()
groups = r.groupdict()["used_groups"]
# If the systems has no group, it automatically uses the root group.
if groups:
group_names = tuple(a.strip() for a in groups.split(","))
else:
group_names = ("root",)
return cls(name, group_names)
@dataclass(frozen=True)
class SystemDefinition(common.DirectiveBlock):
"""Definition of a System:
@system <name> [using <group 1>, ..., <group N>]
<rule 1>
...
<rule N>
@end
See Rule and Comment for more parsing related information.
The syntax for the rule is:
new_unit_name : old_unit_name
where:
- old_unit_name: a root unit part which is going to be removed from the system.
- new_unit_name: a non root unit which is going to replace the old_unit.
If the new_unit_name and the old_unit_name, the later and the colon can be omitted.
"""
@property
def unit_replacements(self) -> ty.Tuple[ty.Tuple[str, str], ...]:
return tuple((el.new_unit_name, el.old_unit_name) for el in self.body)
|