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

added "times" to parser

Browse source
This commit is contained in:
Ruediger Ludwig 2022-12-09 07:32:30 +01:00
parent bb83caec1d
commit a60707185f
2 changed files with 36 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
advent/parser/parser.py
View file

@ -127,6 +127,18 @@ class P(Generic[T]):
def optional(self) -> P[T | None]:
return P.either(self, P.pure(None))
def times(self, *, max: int | None = None, min: int | None = None,
exact: int | None = None) -> P[list[T]]:
match (exact, min, max):
case (int(e), None, None):
return self.many().satisfies(lambda lst: len(lst) == e)
case (None, int(mn), None):
return self.many().satisfies(lambda lst: len(lst) >= mn)
case (None, None, int(mx)):
return self.many().satisfies(lambda lst: len(lst) <= mx)
case _:
raise Exception("Choose exactly one of exact, min or max")
def sep_by(self, sep: P[Any]) -> P[list[T]]:
return P.map2(self, P.snd(sep, self).many(), lambda f, r: [f] + r)