Вы можете встать на правильный путь, начав с функции, которая возвращает логическое значение в зависимости от того, входит ли предоставленный вход в некоторый набор допустимых входных данных:
def is_valid_option(option):
valid_options = ['A', 'B', 'C']
if option in valid_options:
return True
return False
Затем можно проверить, если is_valid_option(some_input)