Состояние и методы генератора python

[Nicodim]

Здравствуйте, подскажите пожалуйста что нужно подправить в коде. В последней строчке теста выдает banana, а нужно cat.

На предыдущем уроке вы решали задачу «Словарь». В ней гарантировалось, что в сопрограмму alphabet будут передаваться только значения, которые являются ключами глобальной переменной DICTIONARY.

Теперь вам необходимо переписать сопрограмму alphabet так, чтобы она могла обрабатывать исключение KeyError. В случае, когда возникнет исключение KeyError, сопрограмма должна генерировать значение «default».



Переменная DICTIONARY вам в редакторе кода по-прежнему не видна, но вы можете обращаться к ней внутри сопрограммы alphabet.

Sample Input 1:

coro = alphabet()
next(coro)
print(coro.send('a'))
print(coro.send('b'))
print(coro.throw(KeyError))
print(coro.send('c'))
Sample Output 1:

apple
banana
default
cat
Sample Input 2:

coro = alphabet()
next(coro)
for letter in 'qwerty':
print(coro.send(letter))
print(coro.throw(KeyError))
Sample Output 2:

quail
default
walrus
default
elephant
default
rabbit
default
tiger
default
yak
default
Sample Input 3:

coro = alphabet()
next(coro)
print(coro.send('apple'))
print(coro.send('banana'))
print(coro.throw(KeyError))
print(coro.send('dog'))
print(coro.send('d'))
Sample Output 3:

default
default
default
default
dog

def alphabet():
    letter = yield
    while True:
        try:
           
           
             letter = yield DICTIONARY[letter]
        except KeyError:
 
            yield 'default'

[BDA]

def alphabet():
    letter = yield
    while True:
        try:
            letter = yield DICTIONARY.get(letter, 'default')
        except KeyError:
            letter = None

Так работает, но мне кажется, должен быть более элегантный способ. А в таком виде подавляется KeyError от словаря, а приходят KeyError только через throw.

[ViktorR]

BDA, всё ещё не понимаю до конца, но ведь ваш код работает и так:

def alphabet():
    letter = yield
    while True:
        try:
            letter = yield DICTIONARY[letter]
        except KeyError:
            letter = yield 'default' 

coro = alphabet()
next(coro)
print(coro.send('a'))
print(coro.send('b'))
print(coro.throw(KeyError))
print(coro.send('c'))
print(coro.send('e'))
print(coro.send('f'))
print(coro.send('z'))

В этой строке

letter = yield DICTIONARY[letter]

порождается исключение KeyError, если значение letter не является ключом словаря.

PS: Использовал этот словарь:

DICTIONARY = {
    'a': 'apple',
    'b': 'banana',
    'c': 'cat',
    'd': 'dog',
    'q': 'quail',
    'w': 'walrus',
    'e': 'elephant',
    'r': 'rabbit',
    't': 'tiger',
    'y': 'yak'
    }

[BDA]

ViktorR, я тоже сначала именно так исправил (добавил присваивание letter), но если будет два исключения подряд (сначала из-за словаря, а потом из-за throw), то не сработает, и придется второй yield тоже оборачивать в try блок и добавлять третий yield. Пример (на основе Sample Input 3):

coro = alphabet()
next(coro)
print(coro.send('apple'))
print(coro.throw(KeyError))

[ViktorR]

Да, так я не проверил.