Initial
This commit is contained in:
commit
a355de5d8b
24 changed files with 1133 additions and 0 deletions
37
advent/common/char_provider.py
Normal file
37
advent/common/char_provider.py
Normal file
|
|
@ -0,0 +1,37 @@
|
|||
from typing import Iterable, Iterator
|
||||
|
||||
from .provider import EofException, Provider
|
||||
|
||||
|
||||
class CharProvider(Provider[str]):
|
||||
data: Iterator[str]
|
||||
|
||||
def __init__(self, data: Iterator[str] | Iterable[str]) -> None:
|
||||
if isinstance(data, Iterator):
|
||||
self.data = data
|
||||
else:
|
||||
self.data = iter(data)
|
||||
self.peeked: list[str] = []
|
||||
|
||||
def _ensure_next(self) -> str:
|
||||
if not self.peeked:
|
||||
try:
|
||||
self.peeked = [next(self.data)]
|
||||
except StopIteration:
|
||||
raise EofException() from None
|
||||
return self.peeked[0]
|
||||
|
||||
def peek(self) -> str:
|
||||
return self._ensure_next()
|
||||
|
||||
def get(self) -> str:
|
||||
result = self._ensure_next()
|
||||
self.peeked = self.peeked[1:]
|
||||
return result
|
||||
|
||||
def finished(self) -> bool:
|
||||
try:
|
||||
self._ensure_next()
|
||||
return False
|
||||
except EofException:
|
||||
return True
|
||||
Loading…
Add table
Add a link
Reference in a new issue