professor.py

import random


def main():
    lvl = get_level("Level: ")
    bodovi = 0
    for broj in range(10):
        x = generate_integer(lvl)
        y = generate_integer(lvl)
        izraz = f"{x} + {y} = "
        zbir, bod = get_int(x + y, izraz)
        if bod == 0:
            print(izraz, zbir)
        bodovi += bod
    print(f"{bodovi}/10")



def get_int(suma, expression):
    n = 0
    while n < 3:
        try:
            if suma == int(input(expression)):
                return suma, 1
            else:
                print("EEE")
                n += 1
        except(ValueError, TypeError, OverflowError):
            print("EEE")
            n += 1
    return suma, 0

def get_level(prompt):
    while True:
        try:
            number = input(prompt)
            number = int(number)
            if number in [1, 2, 3]:
                return number
            else:
                pass
        except(ValueError, TypeError, OverflowError):
            pass

def generate_integer(level):
    if level == 1:
        return random.randint(0, 9)
    elif level == 2:
        return random.randint(10, 99)
    elif level == 3:
        return random.randint(100, 999)



if __name__ == "__main__":
    main()