danvics/ytdl-sub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

top-level context manager

Browse source
This commit is contained in:
Jesse Bannon 2023-11-10 00:06:39 -08:00
parent 291c308cad
commit babda0a037
1 changed files with 40 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

76
src/ytdl_sub/script/parser.py
View file

@ -27,9 +27,11 @@ class _Parser:
def __init__(self, text: str): def __init__(self, text: str):
self._text = text self._text = text
self._pos = 0 self._pos = 0
self._error_highlight_pos = 0
self._ast: List[ArgumentType] = [] self._ast: List[ArgumentType] = []
self._syntax_tree = self._parse() with self._error_formatting():
self._syntax_tree = self._parse()
@property @property
def ast(self) -> SyntaxTree: def ast(self) -> SyntaxTree:
@ -42,14 +44,16 @@ class _Parser:
@contextmanager @contextmanager
def _error_formatting(self) -> None: def _error_formatting(self) -> None:
parked_pos = self._pos
try: try:
yield yield
except NonFormattedInvalidSyntaxException as exc: except NonFormattedInvalidSyntaxException as exc:
raise ParserExceptionFormatter( raise ParserExceptionFormatter(
self._text, parked_pos, self._pos, exc self._text, self._error_highlight_pos, self._pos, exc
).highlight() from exc ).highlight() from exc
def _set_highlight_position(self) -> None:
self._error_highlight_pos = self._pos
def _read(self, increment_pos: bool = True, length: int = 1) -> Optional[str]: def _read(self, increment_pos: bool = True, length: int = 1) -> Optional[str]:
try: try:
ch = self._text[self._pos : (self._pos + length)] ch = self._text[self._pos : (self._pos + length)]
@ -240,41 +244,41 @@ class _Parser:
key: Optional[ArgumentType] = None key: Optional[ArgumentType] = None
in_comma = False in_comma = False
with self._error_formatting(): self._set_highlight_position()
while ch := self._read(increment_pos=False): while ch := self._read(increment_pos=False):
if ch == "}": if ch == "}":
if key is not None: if key is not None:
raise NonFormattedInvalidSyntaxException("Map has a key with no value") raise NonFormattedInvalidSyntaxException("Map has a key with no value")
self._pos += 1 self._pos += 1
return UnresolvedMap(value=output) return UnresolvedMap(value=output)
elif ch == ",": elif ch == ",":
if in_comma: if in_comma:
raise StringFormattingException("Comma followed by comma") raise StringFormattingException("Comma followed by comma")
if key is not None: if key is not None:
raise InvalidSyntaxException("Map has a key with no value") raise InvalidSyntaxException("Map has a key with no value")
if output is None: if output is None:
raise StringFormattingException("Empty dict with comma") raise StringFormattingException("Empty dict with comma")
in_comma = True in_comma = True
self._pos += 1 self._pos += 1
elif key is None: elif key is None:
with self._error_formatting(): self._set_highlight_position()
in_comma = False in_comma = False
key_args = self._parse_args(breaking_chars=":") key_args = self._parse_args(breaking_chars=":")
if len(key_args) != 1: if len(key_args) != 1:
raise StringFormattingException("Lazy parsing but got mlutiple args") raise StringFormattingException("Lazy parsing but got mlutiple args")
key = key_args[0] key = key_args[0]
elif key is not None and ch == ":": elif key is not None and ch == ":":
with self._error_formatting(): self._set_highlight_position()
self._pos += 1 self._pos += 1
value_args = self._parse_args(breaking_chars=",}") value_args = self._parse_args(breaking_chars=",}")
if len(value_args) != 1: if len(value_args) != 1:
raise NonFormattedInvalidSyntaxException("Map has a key with no value") raise NonFormattedInvalidSyntaxException("Map has a key with no value")
output[key] = value_args[0] output[key] = value_args[0]
key = None key = None
else: else:
raise StringFormattingException("Invalid map") raise StringFormattingException("Invalid map")
def _parse(self) -> SyntaxTree: def _parse(self) -> SyntaxTree:
bracket_counter = 0 bracket_counter = 0