Арифметические действия. Задача по программированию!

учителя по информатике хотя меня до инфаркта довести ​

import operator

ACTIONS = [operator.add,

          operator.sub,

          operator.mul,

          operator.floordiv,

          operator.mod]

def main():

   action = int(input())

   a, b = map(int, input().split())

   print(ACTIONS[action - 1](a, b))

main()

ЗЫ: грех не землячке :)