ну идея здравая, называется оператор switch-case в питоне реализуется пока вот так как я нарисовал - через словарь с функциями в виде значений словаря через строковые переменные можно колхозить через eval но это небезопасно и не рекомендуется
Есть по-моему пеп с идеей добавить match в новые версии python