[WIP]Kirill Obukhov

final_task/pycalc.py

@@ -0,0 +1,263 @@
+import math
+import sys
+import argparse
+import operator
+import re
+from collections import namedtuple
+
+
+
+class ErrorOperands(Exception):
+    pass
+
+
+class ErrorOperators(Exception):
+    pass
+
+
+class ErrorWhitespace(Exception):
+    pass
+
+
+class ErrorBracketsBalance(Exception):
+    pass
+
+
+class ErrorUnexpectedComma(Exception):
+    pass
+
+
+class ErrorEmptyExpression(Exception):
+    pass
+
+
+class ErrorParse(Exception):
+    pass
+
+
+class ErrorUnknownTokens(Exception):
+    pass
+
+
+sys.tracebacklimit = 0
+
+op = namedtuple('op', ['prec',  'func'])
+
+operators = {
+    '^': op(4, operator.pow),
+    '*': op(3, operator.mul),
+    '/': op(3, operator.truediv),
+    '//': op(3, operator.floordiv),
+    '%': op(3, operator.mod),
+    '+': op(2, operator.add),
+    '-': op(2, operator.sub)
+}
+
+prefix = namedtuple('prefix', ['func', 'args'])
+
+functions = {
+    'sin': prefix(math.sin, 1),
+    'cos': prefix(math.cos, 1),
+    'tan': prefix(math.tan, 1),
+    'asin': prefix(math.asin, 1),
+    'acos': prefix(math.acos, 1),
+    'atan': prefix(math.atan, 1),
+    'sqrt': prefix(math.sqrt, 1),
+    'exp': prefix(math.exp, 1),
+    'log': prefix(math.log, 2),
+    'log1p': prefix(math.log1p, 1),
+    'log10': prefix(math.log10, 1),
+    'factorial': prefix(math.factorial, 1),
+    'pow': prefix(math.pow, 2),
+    'abs': prefix(abs, 1),
+    'round': prefix(round, 1),
+    'p': prefix(operator.pos, 1),
+    'n': prefix(operator.neg, 1)
+}
+
+constants = {
+    'pi': math.pi,
+    'e': math.e,
+    'tau': math.tau
+}
+
+comparators = {
+    '<': operator.lt,
+    '>': operator.gt,
+    '<=': operator.le,
+    '>=': operator.ge,
+    '==': operator.eq,
+    '!=': operator.ne
+}
+
+
+def is_it_number(num):
+    try:
+        float(num)
+        return True
+    except ValueError:
+        return False
+
+
+def brackets_balance(expression):
+    count = 0
+    for ex in expression:
+        if ex == '(':
+            count += 1
+        elif ex == ')':
+            count -= 1
+        if count < 0:
+            raise ErrorBracketsBalance("ERROR: brackets are not balanced.")
+    if count > 0:
+        raise ErrorBracketsBalance("ERROR: brackets are not balanced.")
+
+
+def get_tokens(expression, input_queue=None):
+    if expression is '':
+        raise ErrorEmptyExpression("ERROR: no expression provided.")
+    if input_queue is None:
+        expression = expression.strip().lower().replace(' ', '')
+        input_queue = []
+    token = re.match(r'\)|\(|\d+\.?\d*|[-+*/,^]|\.\d+|\w+|\W+', expression)
+    try:
+        token.group()
+    except Exception:
+        raise ErrorParse("ERROR: couldn't parse this expression.")
+    input_queue.append(token.group())
+    if len(token.group()) < len(expression):
+        return get_tokens(expression[token.end():], input_queue)
+    else:
+        for index, token in enumerate(input_queue):
+            if (input_queue[index-1] in operators or
+                input_queue[index-1] is 'p' or
+                input_queue[index-1] is 'n' or
+                input_queue[index-1] is '(' or
+                    index is 0):
+                if token is '+':
+                    input_queue[index] = 'p'
+                if token is '-':
+                    input_queue[index] = 'n'
+        return input_queue
+
+
+def infix_to_postfix(input_queue):
+    if input_queue[-1] in operators or input_queue[-1] in functions:
+        raise ErrorOperators("ERROR: trailing operators.")
+    output_queue = []
+    operator_stack = []
+    while len(input_queue):
+        token = input_queue[0]
+        if is_it_number(token):
+            output_queue.append(float(input_queue.pop(0)))
+        elif token in constants:
+            output_queue.append(constants[token])
+            input_queue.pop(0)
+        elif token in functions:
+            operator_stack.append(input_queue.pop(0))
+        elif token is '(':
+            operator_stack.append(input_queue.pop(0))
+        elif token is ',':
+            if 'log' not in operator_stack and 'pow' not in operator_stack:
+                raise ErrorUnexpectedComma("ERROR: unexpected comma.")
+            try:
+                while operator_stack[-1] is not '(':
+                    output_queue.append(operator_stack.pop())
+                input_queue.pop(0)
+            except IndexError:
+                raise ErrorUnexpectedComma("ERROR: unexpected comma.")
+        elif token in operators:
+            while True:
+                if not operator_stack:
+                    operator_stack.append(input_queue.pop(0))
+                    break
+                if ((operator_stack[-1] is not '(')
+                    and (operator_stack[-1] in functions
+                         or operators[token].prec < operators[operator_stack[-1]].prec
+                         or (operators[operator_stack[-1]].prec == operators[token].prec
+                             and token is not '^'))):
+                    output_queue.append(operator_stack.pop())
+                else:
+                    operator_stack.append(input_queue.pop(0))
+                    break
+        elif token is ')':
+            while operator_stack[-1] is not '(':
+                output_queue.append(operator_stack.pop())
+            operator_stack.pop()
+            input_queue.pop(0)
+        else:
+            raise ErrorUnknownTokens("ERROR: unknown tokens.")
+    while len(operator_stack):
+        output_queue.append(operator_stack.pop())
+    return output_queue
+
+
+def calculate(expression):
+    stack = []
+    while expression:
+        token = expression.pop(0)
+        if is_it_number(token):
+            stack.append(token)
+        elif token in operators:
+            operand_2 = stack.pop()
+            operand_1 = stack.pop()
+            result = operators[token].func(operand_1, operand_2)
+            stack.append(result)
+        elif token in functions:
+            if functions[token].args is 2:
+                if len(stack) is 1:
+                    operand = stack.pop()
+                    result = functions[token].func(operand)
+                    stack.append(result)
+                else:
+                    operand_2 = stack.pop()
+                    operand_1 = stack.pop()
+                    result = functions[token].func(operand_1, operand_2)
+                    stack.append(result)
+            else:
+                operand = stack.pop()
+                result = functions[token].func(operand)
+                stack.append(result)
+    if len(stack) is 1:
+        return stack[0]
+    else:
+        raise ErrorOperands("ERROR: wrong number of operands.")
+
+
+def split_comparison(expression):
+    brackets_balance(expression)
+    expression = re.sub(r'\s', '', expression)
+    token = re.findall(r'==|>=|<=|>|<|!=', expression)
+    if len(token) > 1:
+        raise ErrorOperators("ERROR: more than one comparison operator.")
+    elif len(token) is 0:
+        return expression, None
+    else:
+        expressions = expression.split(token[0])
+        return expressions, token[0]
+
+
+def get_result(expression):
+    expression = get_tokens(expression)
+    expression = infix_to_postfix(expression)
+    try:
+        expression = calculate(expression)
+    except ZeroDivisionError as err:
+        print("ERROR: {err}.".format(err=err))
+    return expression
+
+
+def compare(expressions, comparator):
+    # calculated_expressions = [get_result(expr) for expr in expressions]
+    return comparators[comparator](expressions[0], expressions[1])
+
+
+def main():
+    # parsing args
+    arg_parser = argparse.ArgumentParser(description='Pure-python command-line calculator.')
+    arg_parser.add_argument('EXPRESSION', help='expression string to evaluate')
+    expression = arg_parser.parse_args().EXPRESSION
+    modules = arg_parser.parse_args().use_modules
+
+
+    if __name__ == '__main__':
+    main()

final_task/setup.py

@@ -0,0 +1,14 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='pycalc',
+    description='Calculator',
+    version='0.1',
+    author='UnicornTT',
+    author_email='[email protected]',
+    packages=find_packages(),
+    entry_points={
+        'console_scripts': [
+            'pycalc=pycalc.core:main',
+        ]
+    })