34 lines
856 B
Python
34 lines
856 B
Python
#!/usr/bin/env python3
|
|
"""Calculator CLI: python calc.py "<expression>"""
|
|
import sys
|
|
|
|
from calc.lexer import tokenize, LexError
|
|
from calc.parser import parse, ParseError
|
|
from calc.evaluator import evaluate, EvalError
|
|
|
|
|
|
def _fmt(value) -> str:
|
|
"""Return value as int string if whole, else as float string."""
|
|
if isinstance(value, float) and value.is_integer():
|
|
return str(int(value))
|
|
return str(value)
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) != 2:
|
|
print("usage: calc.py <expression>", file=sys.stderr)
|
|
sys.exit(1)
|
|
expr = sys.argv[1]
|
|
try:
|
|
tokens = tokenize(expr)
|
|
ast = parse(tokens)
|
|
result = evaluate(ast)
|
|
print(_fmt(result))
|
|
except (LexError, ParseError, EvalError) as exc:
|
|
print(f"error: {exc}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|