Zamiana typu string na function

Question

Witam,

Mam kod:

function calculate(action, a, b):
    return action(a, b)

function add(a,b):
    return a+b

action = input = ("Podaj działanie: ")
a = input("Liczba 1: ")
b = input("Liczba 2: ")

result = calculate(action, a, b)

print("result")

Po wykonaniu tego kodu mam TypeError, ponieważ String is not callable. Czyli action jest stringiem zamiast funkcją add.

Moje pytanie to jak skonwertować zmienną globalną action ("add") na funkcję add tak bym mógł ją wywoływać poprzez funkcję calculate?

Dzięki za pomoc.

Cześć :-).

Comments

BTW,

a = input("Liczba 1: ")

To Ci da stringa, nie liczbę.

---

Racja.

Answer 1

Osobiście najprędzej zrobiłbym mapę ze stringa na obiekt funkcji:

def add(a,b):
    return a+b

def multiply(a,b):
    return a*b

name_to_func = {
    "add": add,
    "multiply": multiply,
}

def calculate(action, a, b):
    func = name_to_func[action]
    return func(a, b)

 

Comments

Fakt :) Przekopiowałem z oryginalnego kodu i nawet nie zauważyłem.

---

Myślę że to jest dobre rozwiązanie. Dzięki.