ruediger/advent-2022-python
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

simplified parser

Browse source
This commit is contained in:
Ruediger Ludwig 2022-12-12 05:49:41 +01:00
parent 4034c08cd9
commit 28fb30d493
3 changed files with 19 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

25
advent/parser/parser.py
View file

@ -112,24 +112,27 @@ class P(Generic[T]):
def __init__(self, func: ParserFunc[T]):
self.func = func
def parse(self, s: str) -> Result[T]:
all_results = self.func(SimpleParserInput(s, 0))
def parse(self, input: str | Iterator[str]) -> Result[T]:
if isinstance(input, str):
parser_input = SimpleParserInput(input, 0)
else:
parser_input = IteratorParserInput(StringDispenser(input), 0)
all_results = self.func(parser_input)
try:
_, result = next(all_results)
return Result.of(result)
except StopIteration:
return Result.fail("No result")
def parse_iterator(self, it: Iterator[str]) -> Result[T]:
all_results = self.func(IteratorParserInput(StringDispenser(it), 0))
try:
_, result = next(all_results)
return Result.of(result)
except StopIteration:
return Result.fail("No result")
def parse_multi(self, input: str | Iterator[str]) -> Iterator[T]:
if isinstance(input, str):
parser_input = SimpleParserInput(input, 0)
else:
parser_input = IteratorParserInput(StringDispenser(input), 0)
def parse_multi(self, s: str, i: int = 0) -> Iterator[T]:
return (v for _, v in self.func(SimpleParserInput(s, i)))
all_results = self.func(parser_input)
return (v for _, v in all_results)
@classmethod
def pure(cls, value: T) -> P[T]: