import math
lol = input('Коофиценты (1), числа (2)')
def aboba():
if lol = 1
def calc(a,b, operand):
if operand == '+':
res = a + b
elif operand == '-':
res = a - b
elif operand == '*':
res = a * b
elif operand == '/':
res = a / b
else:
res = 'Извините, но такой операции нет в калькуляторе.'
return res
num1 = int(input('Введите число:'))
num2 = int(input('Введите число:'))
opel = input('введите операцию:')
toga= calc(num1, num2, opel)
print(toga)
file = open('calculations.txt', 'a')
file.write(f'{toga}' + '\n')
print("Введите коэффициенты для уравнения")
print("ax^2 + bx + c = 0:")
a = float(input("a = "))
b = float(input("b = "))
c = float(input("c = "))
discr = b ** 2 - 4 * a * c
print("Дискриминант D = %.2f" % discr)
if discr > 0:
x1 = (-b + math.sqrt(discr)) / (2 * a)
x2 = (-b - math.sqrt(discr)) / (2 * a)
print("x1 = %.2f \nx2 = %.2f" % (x1, x2))
elif discr == 0:
x = -b / (2 * a)
print("x = %.2f" % x)
else:
print("Корней нет")
Нужно сделать так чтобы в зависимости от введенного числа менялась нужная мне функция