Помогите с задачей на программирование пожалуйста

Question

Следить за чем-то бывает не так просто, а когда следить нужно сразу за многими вещами – здесь справится не каждый. Но можно доверить эту задачу роботам. 
На Белоптичкинской птицефабрике проводится работа по технологизации слежки за подрастающими цыплятами. Руководство фабрики обратилось за помощью к местному кружку робототехников, и они собрали «робо-маму». Она круглосуточно отслеживает и различает цыплят с помощью ИИ и машинного зрения. Не хватало лишь одной немаловажной части – «подружить» робота с системой управления птицефабрикой, написанной на языке COBOL задолго до основания кружка. 
Напишите программу-драйвер для «робо–мамы», реализующую логику приёма и ответа согласно заданного протокола. Система управления отправляет в драйвер одну из следующих команд: 
REMEMBER name – добавить в память цыплёнка с заданным именем. Если такой цыплёнок уже добавлен, ответить ERROR EXIST, иначе OK. 
FORGET name – удалить цыплёнка с заданным именем из памяти. Если такой цыплёнок уже удалён, ответить ERROR MISSING, иначе OK. 
WATCH name – начать следить за цыплёнком с заданным именем. Если такой цыплёнок не существует, ответить ERROR MISSING, иначе OK, даже если слежение за цыплёнком уже ведётся. 
UNWATCH name – перестать следить за цыплёнком с заданным именем. Если такой цыплёнок не существует, ответить ERROR MISSING, иначе OK, даже если слежение за цыплёнком не ведётся. 
LIST type – перечислить цыплят в алфавитном порядке через запятую с пробелом. Например: 123, CHEEPY3, TAPI. Параметр type задаёт необходимый список: ALL – вывести всех цыплят в памяти, WATCH – всех, за которыми идёт слежка. 
Имя цыплёнка (name) всегда представляет собой последовательность заглавных латинских букв и цифр. При попытке назвать цыплёнка иначе нужно отвечать ERROR NAME в любой команде. Если переданная команда не соответствует протоколу, следует отвечать ERROR SYNTAX. 
Формат входных данных: последовательность строк (до  
Unexpected text node: ' 
                             
                        ' 
200), в каждой из которых приведена отдельная команда. 
Формат выходных данных: такое же число строк, в каждой из которых приводится ответ на соответствующую команду.

Рустам Абдрашитов · Answer

На import sys 
import re 
 
def main(): 
    chicks, watching = set(), set() 
    valid_name = re.compile(r'^[A-Z0-9]+$') 
 
    for line in sys.stdin: 
        line = line.strip() 
        if not line or ' ' not in line: 
            print(&#34;ERROR SYNTAX&#34;) 
            continue 
 
        command, name = line.split(maxsplit=1) 
        if not valid_name.match(name): 
            print(&#34;ERROR NAME&#34;) 
            continue 
 
        if command == &#34;REMEMBER&#34;: 
            print(&#34;OK&#34; if name not in chicks and not chicks.add(name) else &#34;ERROR EXIST&#34;) 
        elif command == &#34;FORGET&#34;: 
            print(&#34;OK&#34; if name in chicks and not (chicks.remove(name), watching.discard(name)) else &#34;ERROR MISSING&#34;) 
        elif command == &#34;WATCH&#34;: 
            print(&#34;OK&#34; if name in chicks and not watching.add(name) else &#34;ERROR MISSING&#34;) 
        elif command == &#34;UNWATCH&#34;: 
            print(&#34;OK&#34; if name in chicks and not watching.discard(name) else &#34;ERROR MISSING&#34;) 
        elif command == &#34;LIST&#34;: 
            if name == &#34;ALL&#34;: 
                print(&#34;, &#34;.join(sorted(chicks))) 
            elif name == &#34;WATCH&#34;: 
                print(&#34;, &#34;.join(sorted(watching))) 
            else: 
                print(&#34;ERROR SYNTAX&#34;) 
        else: 
            print(&#34;ERROR SYNTAX&#34;) 
 
if __name__ == &#34;__main__&#34;: 
    main()

Sergio 2.1 · Answer

def is_valid_name(name): 
    return name.isalnum() and name.isupper() 
 
def process_command(command): 
    parts = command.split() 
     
    if not parts: 
        return &#34;ERROR SYNTAX&#34; 
         
    cmd = parts[0] 
     
    if cmd in [&#34;REMEMBER&#34;, &#34;FORGET&#34;, &#34;WATCH&#34;, &#34;UNWATCH&#34;]: 
        if len(parts) != 2: 
            return &#34;ERROR SYNTAX&#34; 
        name = parts[1] 
        if not is_valid_name(name): 
            return &#34;ERROR NAME&#34; 
             
    if cmd == &#34;REMEMBER&#34;: 
        if name in all_chicks: 
            return &#34;ERROR EXIST&#34; 
        all_chicks.add(name) 
        return &#34;OK&#34; 
         
    elif cmd == &#34;FORGET&#34;: 
        if name not in all_chicks: 
            return &#34;ERROR MISSING&#34; 
        all_chicks.remove(name) 
        watched_chicks.discard(name) 
        return &#34;OK&#34; 
         
    elif cmd == &#34;WATCH&#34;: 
        if name not in all_chicks: 
            return &#34;ERROR MISSING&#34; 
        watched_chicks.add(name) 
        return &#34;OK&#34; 
         
    elif cmd == &#34;UNWATCH&#34;: 
        if name not in all_chicks: 
            return &#34;ERROR MISSING&#34; 
        watched_chicks.discard(name) 
        return &#34;OK&#34; 
         
    elif cmd == &#34;LIST&#34;: 
        if len(parts) != 2: 
            return &#34;ERROR SYNTAX&#34; 
        type_ = parts[1] 
        if type_ == &#34;ALL&#34;: 
            result = sorted(all_chicks) 
        elif type_ == &#34;WATCH&#34;: 
            result = sorted(watched_chicks) 
        else: 
            return &#34;ERROR SYNTAX&#34; 
        return &#34;, &#34;.join(result) if result else &#34;&#34; 
         
    return &#34;ERROR SYNTAX&#34; 
 
all_chicks = set() 
watched_chicks = set() 
  
while True: 
    try: 
        command = input() 
        print(process_command(command)) 
    except EOFError: 
        break