Таблицы значений функций

from math import sqrt
 
 
class Indentity:
    def call(self, x):
        return x
 
 
class SqrtFun:
    def call(self, x):
        return sqrt(x)
 
 
class Const:
    def __init__(self, value):
        self.value = value
 
    def call(self, x):
        return self.value
 
 
class Expression:
    def __init__(self, f1, op, f2):
        self.f1 = f1
        self.f2 = f2
        self.op = op
 
    def call(self, x):
        a = self.f1.call(x)
        b = self.f2.call(x)
        if self.op == '+':
            return a + b
        if self.op == '-':
            return a - b
        if self.op == '/':
            return a / b
        if self.op == '*':
            return a * b
 
 
def main():
    functions = {'x': Indentity(), 'sqrt_fun': SqrtFun()}
    n = int(input())
    for _ in range(n):
        command, *args = input().split()
        if command == 'define':
            name, f1, op, f2 = args
            if f1.isdigit() or (f1[0] == '-' and f1[1:].isdigit()):
                f1 = Const(int(f1))
            else:
                f1 = functions[f1]
            if f2.isdigit() or (f2[0] == '-' and f2[1:].isdigit()):
                f2 = Const(int(f2))
            else:
                f2 = functions[f2]
            functions[name] = Expression(f1, op, f2)
        elif command == 'calculate':
            name, *x_values = args
            print(' '.join(str(functions[name].call(int(x))) for x in x_values))
 
 
if __name__ == '__main__':
    main()

class X:
    def __call__(self, x):
        return x
 
 
class SQRT:
    def __call__(self, x):
        return x ** 0.5
 
 
d = {'x': X(), 'sqrt_fun': SQRT()}
 
 
class A:
    def __init__(self, znak, arg1, arg2):
        self.znak = znak
        self.arg1 = arg1
        self.arg2 = arg2
 
    def __call__(self, arg):
        if self.arg1 in d:
            a = d[self.arg1](arg)
        else:
            a = int(self.arg1)
        b = d[self.arg2](arg)
        if self.znak == '+':
            return a + b
        if self.znak == '-':
            return a - b
        if self.znak == '*':
            return a * b
        if self.znak == '/':
            return a / b
        if self.znak == '//':
            return a // b
 
 
n = int(input())
for i in range(n):
    s = input().split()
    if s[0] == 'define':
        d[s[1]] = A(s[3], s[2], s[4])
    if s[0] == 'calculate':
        g = s[2:]
        g = [int(j) for j in g]
        ans = d[s[1]]
        for u in g:
            print(ans(u), end=' ')