Fix "key not found" exception if the very first message handler in a ConversationHandler returns the state ConversationHandler.END.

This commit is contained in:
Li-aung 'Lewis' Yip 2016-08-22 05:49:37 +08:00
parent 00bba73673
commit ab2d6eb494

View file

@ -213,7 +213,8 @@ class ConversationHandler(Handler):
def update_state(self, new_state, key):
if new_state == self.END:
del self.conversations[key]
if key in self.conversations:
del self.conversations[key]
elif isinstance(new_state, Promise):
self.conversations[key] = (self.conversations[key], new_state)