feat: add task 2 calculator draft

This commit is contained in:
2026-04-09 17:28:40 +06:00
parent 67e25b3f72
commit 3ac1c8b2a1
6 changed files with 222 additions and 1 deletions

39
task2/tokenizer.py Normal file
View File

@@ -0,0 +1,39 @@
from collections import deque
from typing import Iterable, Literal, Optional
from math_objects import Integer, Operator, Parenthesis, Token
def tokenize(string: str) -> Iterable[Token]:
buffer: deque[str] = deque()
prediction: Optional[Literal["number"]] = None
def _take_buffer() -> str:
string = "".join(buffer)
buffer.clear()
return string
for c in string:
if c.isdigit():
if prediction != "number" and buffer:
yield Integer.create_from_string(_take_buffer())
prediction = "number"
buffer.append(c)
elif c in ["+", "-", "*", "/", "(", ")"]:
if buffer:
yield Integer.create_from_string(_take_buffer())
prediction = None
match c:
case "+" | "-":
yield Operator(c, precedence=1)
case "*" | "/":
yield Operator(c, precedence=2)
case _:
yield Parenthesis(c) # type: ignore
prediction = None
elif c == " ":
pass
else:
raise SyntaxError()
if buffer:
yield Integer.create_from_string(_take_buffer())