mirror of
https://github.com/Garmelon/PFERD.git
synced 2023-12-21 10:23:01 +01:00
Overhaul transform logic
-re-> arrows now rename their parent directories (like -->) and don't require a full match (like -exact->). Their old behaviour is available as -exact-re->. Also, this change adds the ">>" arrow head, which modifies the current path and continues to the next rule when it matches.
This commit is contained in:
parent
8ab462fb87
commit
61d902d715
@ -24,8 +24,11 @@ ambiguous situations.
|
|||||||
|
|
||||||
### Added
|
### Added
|
||||||
- `skip` option for crawlers
|
- `skip` option for crawlers
|
||||||
|
- Rules with `>>` instead of `>` as arrow head
|
||||||
|
- `-exact-re->` arrow (behaves like `-re->` did previously)
|
||||||
|
|
||||||
### Changed
|
### Changed
|
||||||
|
- The `-re->` arrow can now rename directories (like `-->`)
|
||||||
- Use `/` instead of `\` as path separator for (regex) rules on Windows
|
- Use `/` instead of `\` as path separator for (regex) rules on Windows
|
||||||
- Use the label to the left for exercises instead of the button name to
|
- Use the label to the left for exercises instead of the button name to
|
||||||
determine the folder name
|
determine the folder name
|
||||||
|
@ -1,151 +1,159 @@
|
|||||||
# I'm sorry that this code has become a bit dense and unreadable. While
|
|
||||||
# reading, it is important to remember what True and False mean. I'd love to
|
|
||||||
# have some proper sum-types for the inputs and outputs, they'd make this code
|
|
||||||
# a lot easier to understand.
|
|
||||||
|
|
||||||
import ast
|
import ast
|
||||||
import re
|
import re
|
||||||
from abc import ABC, abstractmethod
|
from abc import ABC, abstractmethod
|
||||||
|
from dataclasses import dataclass
|
||||||
|
from enum import Enum
|
||||||
from pathlib import PurePath
|
from pathlib import PurePath
|
||||||
from typing import Dict, Optional, Sequence, Union
|
from typing import Callable, Dict, List, Optional, Sequence, TypeVar, Union
|
||||||
|
|
||||||
from .logging import log
|
from .logging import log
|
||||||
from .utils import fmt_path, str_path
|
from .utils import fmt_path, str_path
|
||||||
|
|
||||||
|
|
||||||
class Rule(ABC):
|
class ArrowHead(Enum):
|
||||||
@abstractmethod
|
NORMAL = 0
|
||||||
def transform(self, path: PurePath) -> Union[PurePath, bool]:
|
SEQUENCE = 1
|
||||||
"""
|
|
||||||
Try to apply this rule to the path. Returns another path if the rule
|
|
||||||
was successfully applied, True if the rule matched but resulted in an
|
|
||||||
exclamation mark, and False if the rule didn't match at all.
|
|
||||||
"""
|
|
||||||
|
|
||||||
|
|
||||||
|
class Ignore:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
# These rules all use a Union[T, bool] for their right side. They are passed a
|
class Empty:
|
||||||
# T if the arrow's right side was a normal string, True if it was an
|
pass
|
||||||
# exclamation mark and False if it was missing entirely.
|
|
||||||
|
|
||||||
class NormalRule(Rule):
|
|
||||||
def __init__(self, left: PurePath, right: Union[PurePath, bool]):
|
|
||||||
|
|
||||||
self._left = left
|
RightSide = Union[str, Ignore, Empty]
|
||||||
self._right = right
|
|
||||||
|
|
||||||
def _match_prefix(self, path: PurePath) -> Optional[PurePath]:
|
|
||||||
left_parts = list(reversed(self._left.parts))
|
|
||||||
path_parts = list(reversed(path.parts))
|
|
||||||
|
|
||||||
if len(left_parts) > len(path_parts):
|
@dataclass
|
||||||
|
class Transformed:
|
||||||
|
path: PurePath
|
||||||
|
|
||||||
|
|
||||||
|
class Ignored:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
TransformResult = Optional[Union[Transformed, Ignored]]
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class Rule:
|
||||||
|
left: str
|
||||||
|
name: str
|
||||||
|
head: ArrowHead
|
||||||
|
right: RightSide
|
||||||
|
|
||||||
|
def right_result(self, path: PurePath) -> Union[str, Transformed, Ignored]:
|
||||||
|
if isinstance(self.right, str):
|
||||||
|
return self.right
|
||||||
|
elif isinstance(self.right, Ignore):
|
||||||
|
return Ignored()
|
||||||
|
elif isinstance(self.right, Empty):
|
||||||
|
return Transformed(path)
|
||||||
|
else:
|
||||||
|
raise RuntimeError(f"Right side has invalid type {type(self.right)}")
|
||||||
|
|
||||||
|
|
||||||
|
class Transformation(ABC):
|
||||||
|
def __init__(self, rule: Rule):
|
||||||
|
self.rule = rule
|
||||||
|
|
||||||
|
@abstractmethod
|
||||||
|
def transform(self, path: PurePath) -> TransformResult:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class ExactTf(Transformation):
|
||||||
|
def transform(self, path: PurePath) -> TransformResult:
|
||||||
|
if path != PurePath(self.rule.left):
|
||||||
return None
|
return None
|
||||||
|
|
||||||
while left_parts and path_parts:
|
right = self.rule.right_result(path)
|
||||||
left_part = left_parts.pop()
|
if not isinstance(right, str):
|
||||||
path_part = path_parts.pop()
|
return right
|
||||||
|
|
||||||
if left_part != path_part:
|
return Transformed(PurePath(right))
|
||||||
|
|
||||||
|
|
||||||
|
class ExactReTf(Transformation):
|
||||||
|
def transform(self, path: PurePath) -> TransformResult:
|
||||||
|
match = re.fullmatch(self.rule.left, str_path(path))
|
||||||
|
if not match:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
if left_parts:
|
right = self.rule.right_result(path)
|
||||||
return None
|
if not isinstance(right, str):
|
||||||
|
return right
|
||||||
|
|
||||||
path_parts.reverse()
|
# For some reason, mypy thinks that "groups" has type List[str]. But
|
||||||
return PurePath(*path_parts)
|
# since elements of "match.groups()" can be None, mypy is wrong.
|
||||||
|
|
||||||
def transform(self, path: PurePath) -> Union[PurePath, bool]:
|
|
||||||
if rest := self._match_prefix(path):
|
|
||||||
if isinstance(self._right, bool):
|
|
||||||
return self._right or path
|
|
||||||
else:
|
|
||||||
return self._right / rest
|
|
||||||
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
class ExactRule(Rule):
|
|
||||||
def __init__(self, left: PurePath, right: Union[PurePath, bool]):
|
|
||||||
self._left = left
|
|
||||||
self._right = right
|
|
||||||
|
|
||||||
def transform(self, path: PurePath) -> Union[PurePath, bool]:
|
|
||||||
if path == self._left:
|
|
||||||
if isinstance(self._right, bool):
|
|
||||||
return self._right or path
|
|
||||||
else:
|
|
||||||
return self._right
|
|
||||||
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
class NameRule(Rule):
|
|
||||||
def __init__(self, subrule: Rule):
|
|
||||||
self._subrule = subrule
|
|
||||||
|
|
||||||
def transform(self, path: PurePath) -> Union[PurePath, bool]:
|
|
||||||
matched = False
|
|
||||||
result = PurePath()
|
|
||||||
|
|
||||||
for part in path.parts:
|
|
||||||
part_result = self._subrule.transform(PurePath(part))
|
|
||||||
if isinstance(part_result, PurePath):
|
|
||||||
matched = True
|
|
||||||
result /= part_result
|
|
||||||
elif part_result:
|
|
||||||
# If any subrule call ignores its path segment, the entire path
|
|
||||||
# should be ignored
|
|
||||||
return True
|
|
||||||
else:
|
|
||||||
# The subrule doesn't modify this segment, but maybe other
|
|
||||||
# segments
|
|
||||||
result /= part
|
|
||||||
|
|
||||||
if matched:
|
|
||||||
return result
|
|
||||||
else:
|
|
||||||
# The subrule has modified no segments, so this name version of it
|
|
||||||
# doesn't match
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
class ReRule(Rule):
|
|
||||||
def __init__(self, left: str, right: Union[str, bool]):
|
|
||||||
self._left = left
|
|
||||||
self._right = right
|
|
||||||
|
|
||||||
def transform(self, path: PurePath) -> Union[PurePath, bool]:
|
|
||||||
if match := re.fullmatch(self._left, str_path(path)):
|
|
||||||
if isinstance(self._right, bool):
|
|
||||||
return self._right or path
|
|
||||||
|
|
||||||
vars: Dict[str, Union[str, int, float]] = {}
|
|
||||||
|
|
||||||
# For some reason, mypy thinks that "groups" has type List[str].
|
|
||||||
# But since elements of "match.groups()" can be None, mypy is
|
|
||||||
# wrong.
|
|
||||||
groups: Sequence[Optional[str]] = [match[0]] + list(match.groups())
|
groups: Sequence[Optional[str]] = [match[0]] + list(match.groups())
|
||||||
|
|
||||||
|
locals_dir: Dict[str, Union[str, int, float]] = {}
|
||||||
for i, group in enumerate(groups):
|
for i, group in enumerate(groups):
|
||||||
if group is None:
|
if group is None:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
vars[f"g{i}"] = group
|
locals_dir[f"g{i}"] = group
|
||||||
|
|
||||||
try:
|
try:
|
||||||
vars[f"i{i}"] = int(group)
|
locals_dir[f"i{i}"] = int(group)
|
||||||
except ValueError:
|
except ValueError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
try:
|
try:
|
||||||
vars[f"f{i}"] = float(group)
|
locals_dir[f"f{i}"] = float(group)
|
||||||
except ValueError:
|
except ValueError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
result = eval(f"f{self._right!r}", vars)
|
result = eval(f"f{right!r}", {}, locals_dir)
|
||||||
return PurePath(result)
|
return Transformed(PurePath(result))
|
||||||
|
|
||||||
return False
|
|
||||||
|
class RenamingParentsTf(Transformation):
|
||||||
|
def __init__(self, sub_tf: Transformation):
|
||||||
|
super().__init__(sub_tf.rule)
|
||||||
|
self.sub_tf = sub_tf
|
||||||
|
|
||||||
|
def transform(self, path: PurePath) -> TransformResult:
|
||||||
|
for i in range(len(path.parts), -1, -1):
|
||||||
|
parent = PurePath(*path.parts[:i])
|
||||||
|
child = PurePath(*path.parts[i:])
|
||||||
|
|
||||||
|
transformed = self.sub_tf.transform(parent)
|
||||||
|
if not transformed:
|
||||||
|
continue
|
||||||
|
elif isinstance(transformed, Transformed):
|
||||||
|
return Transformed(transformed.path / child)
|
||||||
|
elif isinstance(transformed, Ignored):
|
||||||
|
return transformed
|
||||||
|
else:
|
||||||
|
raise RuntimeError(f"Invalid transform result of type {type(transformed)}: {transformed}")
|
||||||
|
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
class RenamingPartsTf(Transformation):
|
||||||
|
def __init__(self, sub_tf: Transformation):
|
||||||
|
super().__init__(sub_tf.rule)
|
||||||
|
self.sub_tf = sub_tf
|
||||||
|
|
||||||
|
def transform(self, path: PurePath) -> TransformResult:
|
||||||
|
result = PurePath()
|
||||||
|
for part in path.parts:
|
||||||
|
transformed = self.sub_tf.transform(PurePath(part))
|
||||||
|
if not transformed:
|
||||||
|
result /= part
|
||||||
|
elif isinstance(transformed, Transformed):
|
||||||
|
result /= transformed.path
|
||||||
|
elif isinstance(transformed, Ignored):
|
||||||
|
return transformed
|
||||||
|
else:
|
||||||
|
raise RuntimeError(f"Invalid transform result of type {type(transformed)}: {transformed}")
|
||||||
|
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
class RuleParseError(Exception):
|
class RuleParseError(Exception):
|
||||||
@ -162,18 +170,15 @@ class RuleParseError(Exception):
|
|||||||
log.error_contd(f"{spaces}^--- {self.reason}")
|
log.error_contd(f"{spaces}^--- {self.reason}")
|
||||||
|
|
||||||
|
|
||||||
|
T = TypeVar("T")
|
||||||
|
|
||||||
|
|
||||||
class Line:
|
class Line:
|
||||||
def __init__(self, line: str, line_nr: int):
|
def __init__(self, line: str, line_nr: int):
|
||||||
self._line = line
|
self._line = line
|
||||||
self._line_nr = line_nr
|
self._line_nr = line_nr
|
||||||
self._index = 0
|
self._index = 0
|
||||||
|
|
||||||
def get(self) -> Optional[str]:
|
|
||||||
if self._index < len(self._line):
|
|
||||||
return self._line[self._index]
|
|
||||||
|
|
||||||
return None
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def line(self) -> str:
|
def line(self) -> str:
|
||||||
return self._line
|
return self._line
|
||||||
@ -190,155 +195,192 @@ class Line:
|
|||||||
def index(self, index: int) -> None:
|
def index(self, index: int) -> None:
|
||||||
self._index = index
|
self._index = index
|
||||||
|
|
||||||
def advance(self) -> None:
|
@property
|
||||||
self._index += 1
|
def rest(self) -> str:
|
||||||
|
return self.line[self.index:]
|
||||||
|
|
||||||
def expect(self, string: str) -> None:
|
def peek(self, amount: int = 1) -> str:
|
||||||
for char in string:
|
return self.rest[:amount]
|
||||||
if self.get() == char:
|
|
||||||
self.advance()
|
def take(self, amount: int = 1) -> str:
|
||||||
|
string = self.peek(amount)
|
||||||
|
self.index += len(string)
|
||||||
|
return string
|
||||||
|
|
||||||
|
def expect(self, string: str) -> str:
|
||||||
|
if self.peek(len(string)) == string:
|
||||||
|
return self.take(len(string))
|
||||||
else:
|
else:
|
||||||
raise RuleParseError(self, f"Expected {char!r}")
|
raise RuleParseError(self, f"Expected {string!r}")
|
||||||
|
|
||||||
|
def expect_with(self, string: str, value: T) -> T:
|
||||||
|
self.expect(string)
|
||||||
|
return value
|
||||||
|
|
||||||
|
def one_of(self, parsers: List[Callable[[], T]], description: str) -> T:
|
||||||
|
for parser in parsers:
|
||||||
|
index = self.index
|
||||||
|
try:
|
||||||
|
return parser()
|
||||||
|
except RuleParseError:
|
||||||
|
self.index = index
|
||||||
|
|
||||||
|
raise RuleParseError(self, description)
|
||||||
|
|
||||||
|
|
||||||
|
# RULE = LEFT SPACE '-' NAME '-' HEAD (SPACE RIGHT)?
|
||||||
|
# SPACE = ' '+
|
||||||
|
# NAME = '' | 'exact' | 'name' | 're' | 'exact-re' | 'name-re'
|
||||||
|
# HEAD = '>' | '>>'
|
||||||
|
# LEFT = STR | QUOTED_STR
|
||||||
|
# RIGHT = STR | QUOTED_STR | '!'
|
||||||
|
|
||||||
|
|
||||||
|
def parse_zero_or_more_spaces(line: Line) -> None:
|
||||||
|
while line.peek() == " ":
|
||||||
|
line.take()
|
||||||
|
|
||||||
|
|
||||||
|
def parse_one_or_more_spaces(line: Line) -> None:
|
||||||
|
line.expect(" ")
|
||||||
|
parse_zero_or_more_spaces(line)
|
||||||
|
|
||||||
|
|
||||||
|
def parse_str(line: Line) -> str:
|
||||||
|
result = []
|
||||||
|
while c := line.peek():
|
||||||
|
if c == " ":
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
line.take()
|
||||||
|
result.append(c)
|
||||||
|
|
||||||
|
if result:
|
||||||
|
return "".join(result)
|
||||||
|
else:
|
||||||
|
raise RuleParseError(line, "Expected non-space character")
|
||||||
|
|
||||||
|
|
||||||
QUOTATION_MARKS = {'"', "'"}
|
QUOTATION_MARKS = {'"', "'"}
|
||||||
|
|
||||||
|
|
||||||
def parse_string_literal(line: Line) -> str:
|
def parse_quoted_str(line: Line) -> str:
|
||||||
escaped = False
|
escaped = False
|
||||||
|
|
||||||
# Points to first character of string literal
|
# Points to first character of string literal
|
||||||
start_index = line.index
|
start_index = line.index
|
||||||
|
|
||||||
quotation_mark = line.get()
|
quotation_mark = line.peek()
|
||||||
if quotation_mark not in QUOTATION_MARKS:
|
if quotation_mark not in QUOTATION_MARKS:
|
||||||
# This should never happen as long as this function is only called from
|
raise RuleParseError(line, "Expected quotation mark")
|
||||||
# parse_string.
|
line.take()
|
||||||
raise RuleParseError(line, "Invalid quotation mark")
|
|
||||||
line.advance()
|
|
||||||
|
|
||||||
while c := line.get():
|
while c := line.peek():
|
||||||
if escaped:
|
if escaped:
|
||||||
escaped = False
|
escaped = False
|
||||||
line.advance()
|
line.take()
|
||||||
elif c == quotation_mark:
|
elif c == quotation_mark:
|
||||||
line.advance()
|
line.take()
|
||||||
stop_index = line.index
|
stop_index = line.index
|
||||||
literal = line.line[start_index:stop_index]
|
literal = line.line[start_index:stop_index]
|
||||||
|
try:
|
||||||
return ast.literal_eval(literal)
|
return ast.literal_eval(literal)
|
||||||
|
except SyntaxError as e:
|
||||||
|
line.index = start_index
|
||||||
|
raise RuleParseError(line, str(e)) from e
|
||||||
elif c == "\\":
|
elif c == "\\":
|
||||||
escaped = True
|
escaped = True
|
||||||
line.advance()
|
line.take()
|
||||||
else:
|
else:
|
||||||
line.advance()
|
line.take()
|
||||||
|
|
||||||
raise RuleParseError(line, "Expected end of string literal")
|
raise RuleParseError(line, "Expected end of string literal")
|
||||||
|
|
||||||
|
|
||||||
def parse_until_space_or_eol(line: Line) -> str:
|
def parse_left(line: Line) -> str:
|
||||||
result = []
|
if line.peek() in QUOTATION_MARKS:
|
||||||
while c := line.get():
|
return parse_quoted_str(line)
|
||||||
if c == " ":
|
|
||||||
break
|
|
||||||
result.append(c)
|
|
||||||
line.advance()
|
|
||||||
|
|
||||||
return "".join(result)
|
|
||||||
|
|
||||||
|
|
||||||
def parse_string(line: Line) -> Union[str, bool]:
|
|
||||||
if line.get() in QUOTATION_MARKS:
|
|
||||||
return parse_string_literal(line)
|
|
||||||
else:
|
else:
|
||||||
string = parse_until_space_or_eol(line)
|
return parse_str(line)
|
||||||
|
|
||||||
|
|
||||||
|
def parse_right(line: Line) -> Union[str, Ignore]:
|
||||||
|
c = line.peek()
|
||||||
|
if c in QUOTATION_MARKS:
|
||||||
|
return parse_quoted_str(line)
|
||||||
|
else:
|
||||||
|
string = parse_str(line)
|
||||||
if string == "!":
|
if string == "!":
|
||||||
return True
|
return Ignore()
|
||||||
return string
|
return string
|
||||||
|
|
||||||
|
|
||||||
def parse_arrow(line: Line) -> str:
|
def parse_arrow_name(line: Line) -> str:
|
||||||
line.expect("-")
|
return line.one_of([
|
||||||
|
lambda: line.expect("exact-re"),
|
||||||
name = []
|
lambda: line.expect("exact"),
|
||||||
while True:
|
lambda: line.expect("name-re"),
|
||||||
c = line.get()
|
lambda: line.expect("name"),
|
||||||
if not c:
|
lambda: line.expect("re"),
|
||||||
raise RuleParseError(line, "Expected rest of arrow")
|
lambda: line.expect(""),
|
||||||
elif c == "-":
|
], "Expected arrow name")
|
||||||
line.advance()
|
|
||||||
c = line.get()
|
|
||||||
if not c:
|
|
||||||
raise RuleParseError(line, "Expected rest of arrow")
|
|
||||||
elif c == ">":
|
|
||||||
line.advance()
|
|
||||||
break # End of arrow
|
|
||||||
else:
|
|
||||||
name.append("-")
|
|
||||||
continue
|
|
||||||
else:
|
|
||||||
name.append(c)
|
|
||||||
|
|
||||||
line.advance()
|
|
||||||
|
|
||||||
return "".join(name)
|
|
||||||
|
|
||||||
|
|
||||||
def parse_whitespace(line: Line) -> None:
|
def parse_arrow_head(line: Line) -> ArrowHead:
|
||||||
line.expect(" ")
|
return line.one_of([
|
||||||
while line.get() == " ":
|
lambda: line.expect_with(">>", ArrowHead.SEQUENCE),
|
||||||
line.advance()
|
lambda: line.expect_with(">", ArrowHead.NORMAL),
|
||||||
|
], "Expected arrow head")
|
||||||
|
|
||||||
|
|
||||||
def parse_eol(line: Line) -> None:
|
def parse_eol(line: Line) -> None:
|
||||||
if line.get() is not None:
|
if line.peek():
|
||||||
raise RuleParseError(line, "Expected end of line")
|
raise RuleParseError(line, "Expected end of line")
|
||||||
|
|
||||||
|
|
||||||
def parse_rule(line: Line) -> Rule:
|
def parse_rule(line: Line) -> Rule:
|
||||||
# Parse left side
|
parse_zero_or_more_spaces(line)
|
||||||
leftindex = line.index
|
left = parse_left(line)
|
||||||
left = parse_string(line)
|
|
||||||
if isinstance(left, bool):
|
|
||||||
line.index = leftindex
|
|
||||||
raise RuleParseError(line, "Left side can't be '!'")
|
|
||||||
leftpath = PurePath(left)
|
|
||||||
|
|
||||||
# Parse arrow
|
parse_one_or_more_spaces(line)
|
||||||
parse_whitespace(line)
|
|
||||||
arrowindex = line.index
|
|
||||||
arrowname = parse_arrow(line)
|
|
||||||
|
|
||||||
# Parse right side
|
line.expect("-")
|
||||||
if line.get():
|
name = parse_arrow_name(line)
|
||||||
parse_whitespace(line)
|
line.expect("-")
|
||||||
right = parse_string(line)
|
head = parse_arrow_head(line)
|
||||||
else:
|
|
||||||
right = False
|
|
||||||
rightpath: Union[PurePath, bool]
|
|
||||||
if isinstance(right, bool):
|
|
||||||
rightpath = right
|
|
||||||
else:
|
|
||||||
rightpath = PurePath(right)
|
|
||||||
|
|
||||||
|
index = line.index
|
||||||
|
right: RightSide
|
||||||
|
try:
|
||||||
|
parse_zero_or_more_spaces(line)
|
||||||
|
parse_eol(line)
|
||||||
|
right = Empty()
|
||||||
|
except RuleParseError:
|
||||||
|
line.index = index
|
||||||
|
parse_one_or_more_spaces(line)
|
||||||
|
right = parse_right(line)
|
||||||
parse_eol(line)
|
parse_eol(line)
|
||||||
|
|
||||||
# Dispatch
|
return Rule(left, name, head, right)
|
||||||
if arrowname == "":
|
|
||||||
return NormalRule(leftpath, rightpath)
|
|
||||||
elif arrowname == "name":
|
def parse_transformation(line: Line) -> Transformation:
|
||||||
if len(leftpath.parts) > 1:
|
rule = parse_rule(line)
|
||||||
line.index = leftindex
|
|
||||||
raise RuleParseError(line, "SOURCE must be a single name, not multiple segments")
|
if rule.name == "":
|
||||||
return NameRule(ExactRule(leftpath, rightpath))
|
return RenamingParentsTf(ExactTf(rule))
|
||||||
elif arrowname == "exact":
|
elif rule.name == "exact":
|
||||||
return ExactRule(leftpath, rightpath)
|
return ExactTf(rule)
|
||||||
elif arrowname == "re":
|
elif rule.name == "name":
|
||||||
return ReRule(left, right)
|
return RenamingPartsTf(ExactTf(rule))
|
||||||
elif arrowname == "name-re":
|
elif rule.name == "re":
|
||||||
return NameRule(ReRule(left, right))
|
return RenamingParentsTf(ExactReTf(rule))
|
||||||
|
elif rule.name == "exact-re":
|
||||||
|
return ExactReTf(rule)
|
||||||
|
elif rule.name == "name-re":
|
||||||
|
return RenamingPartsTf(ExactReTf(rule))
|
||||||
else:
|
else:
|
||||||
line.index = arrowindex + 1 # For nicer error message
|
raise RuntimeError(f"Invalid arrow name {rule.name!r}")
|
||||||
raise RuleParseError(line, f"Invalid arrow name {arrowname!r}")
|
|
||||||
|
|
||||||
|
|
||||||
class Transformer:
|
class Transformer:
|
||||||
@ -347,32 +389,40 @@ class Transformer:
|
|||||||
May throw a RuleParseException.
|
May throw a RuleParseException.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
self._rules = []
|
self._tfs = []
|
||||||
for i, line in enumerate(rules.split("\n")):
|
for i, line in enumerate(rules.split("\n")):
|
||||||
line = line.strip()
|
line = line.strip()
|
||||||
if line:
|
if line:
|
||||||
rule = parse_rule(Line(line, i))
|
tf = parse_transformation(Line(line, i))
|
||||||
self._rules.append((line, rule))
|
self._tfs.append((line, tf))
|
||||||
|
|
||||||
def transform(self, path: PurePath) -> Optional[PurePath]:
|
def transform(self, path: PurePath) -> Optional[PurePath]:
|
||||||
for i, (line, rule) in enumerate(self._rules):
|
for i, (line, tf) in enumerate(self._tfs):
|
||||||
log.explain(f"Testing rule {i+1}: {line}")
|
log.explain(f"Testing rule {i+1}: {line}")
|
||||||
|
|
||||||
try:
|
try:
|
||||||
result = rule.transform(path)
|
result = tf.transform(path)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
log.warn(f"Error while testing rule {i+1}: {line}")
|
log.warn(f"Error while testing rule {i+1}: {line}")
|
||||||
log.warn_contd(str(e))
|
log.warn_contd(str(e))
|
||||||
continue
|
continue
|
||||||
|
|
||||||
if isinstance(result, PurePath):
|
if not result:
|
||||||
log.explain(f"Match found, transformed path to {fmt_path(result)}")
|
|
||||||
return result
|
|
||||||
elif result: # Exclamation mark
|
|
||||||
log.explain("Match found, path ignored")
|
|
||||||
return None
|
|
||||||
else:
|
|
||||||
continue
|
continue
|
||||||
|
|
||||||
log.explain("No rule matched, path is unchanged")
|
if isinstance(result, Ignored):
|
||||||
|
log.explain("Match found, path ignored")
|
||||||
|
return None
|
||||||
|
|
||||||
|
if tf.rule.head == ArrowHead.NORMAL:
|
||||||
|
log.explain(f"Match found, transformed path to {fmt_path(result.path)}")
|
||||||
|
path = result.path
|
||||||
|
break
|
||||||
|
elif tf.rule.head == ArrowHead.SEQUENCE:
|
||||||
|
log.explain(f"Match found, updated path to {fmt_path(result.path)}")
|
||||||
|
path = result.path
|
||||||
|
else:
|
||||||
|
raise RuntimeError(f"Invalid transform result of type {type(result)}: {result}")
|
||||||
|
|
||||||
|
log.explain(f"Final result: {fmt_path(path)}")
|
||||||
return path
|
return path
|
||||||
|
Loading…
Reference in New Issue
Block a user