Classes

import Mylib def scalc(p1): splitted = p1.split(',') if splitted[2] == '+': return Mylib.addition(float(splitted[0]), float(splitted[1])) if splitted[2] == '-': return Mylib.subtraction(float(splitted[0]), float(splitted[1])) if splitted[2] == '*': return Mylib.multiplication(float(splitted[0]), float(splitted[1])) if splitted[2] == '/': return Mylib.division(float(splitted[0]), float(splitted[1])) def allInOne(n1, n2): return "\"add\":"+str(Mylib.addition(n1, n2))+","+"\"sub\":"+str(Mylib.subtraction(n1, n2))+",""\"mult\":"\ + str(Mylib.multiplication(n1, n2))+","+"\"div\":"+str(Mylib.division(n1, n2)) def main(): menuList = ["Add two numbers", "Subtract two number", "Multiply two number", "Divide two numbers", "Scalc", "All in one"] print('\n'.join('{}) {}'.format(*k) for k in enumerate(menuList))) input1 = int(input()) n1 = int(input("Input number 1")) n2 = int(input("input number 2")) if input1 == 0: print(Mylib.addition(n1, n2)) elif input1 == 1: print(Mylib.subtraction(n1, n2)) elif input1 == 2: print(Mylib.multiplication(n1, n2)) elif input1 == 3: print(Mylib.division(n1, n2)) elif input1 == 4: print(scalc(str(n1)+","+str(n2)+",+")) print(scalc(str(n1)+","+str(n2)+",-")) print(scalc(str(n1)+","+str(n2)+",*")) print(scalc(str(n1)+","+str(n2)+",/")) elif input1 == 5: res = (allInOne(n1, n2)) splitted = res.split(',') for i in range(4): splitted1 = splitted[i].split(':') if splitted1[0] == "\"add\"": print(str(n1) + "+" + str(n2) + "=" + splitted1[1]) elif splitted1[0] == "\"sub\"": print(str(n1) + "-" + str(n2) + "=" + splitted1[1]) elif splitted1[0] == "\"mult\"": print(str(n1) + "*" + str(n2) + "=" + splitted1[1]) elif splitted1[0] == "\"div\"": print(str(n1) + "/" + str(n2) + "=" + splitted1[1]) if __name__=="__main__": main()