while not ((data := input("data: ")).isnumeric() and (0 < int(data) < 5)):
    print("wrong! try again")

print("ok")