Use the new in-memory entity cache

This should avoid a disk access every time an input entity
is needed, which is very often. Another step for #1141.

telethon/events/callbackquery.py

@@ -156,10 +156,8 @@ class CallbackQuery(EventBuilder):
             if not getattr(self._input_sender, 'access_hash', True):
                 # getattr with True to handle the InputPeerSelf() case
                 try:
-                    self._input_sender = self._client.session.get_input_entity(
-                        self._sender_id
-                    )
-                except ValueError:
+                    self._input_sender = self._client._entity_cache[self._sender_id]
+                except KeyError:
                     m = await self.get_message()
                     if m:
                         self._sender = m._sender

telethon/events/chataction.py

@@ -362,10 +362,8 @@ class ChatAction(EventBuilder):
                 self._input_users = []
                 for peer in self._user_peers:
                     try:
-                        self._input_users.append(
-                            self._client.session.get_input_entity(peer)
-                        )
-                    except ValueError:
+                        self._input_users.append(self._client._entity_cache[peer])
+                    except KeyError:
                         pass
             return self._input_users or []
 

telethon/events/common.py

@@ -154,10 +154,8 @@ class EventCommon(ChatGetter, abc.ABC):
             self._input_chat = utils.get_input_peer(self._chat)
         except TypeError:
             try:
-                self._input_chat = self._client.session.get_input_entity(
-                    self._chat_peer
-                )
-            except ValueError:
+                self._input_chat = self._client._entity_cache[self._chat_peer]
+            except KeyError:
                 self._input_chat = None
 
     @property

telethon/events/userupdate.py

@@ -170,7 +170,7 @@ class UserUpdate(EventBuilder):
         def _set_client(self, client):
             if isinstance(self._chat_peer, int):
                 try:
-                    chat = client.session.get_input_entity(self._chat_peer)
+                    chat = client._entity_cache[self._chat_peer]
                     if isinstance(chat, types.InputPeerChat):
                         self._chat_peer = types.PeerChat(self._chat_peer)
                     elif isinstance(chat, types.InputPeerChannel):
@@ -178,7 +178,7 @@ class UserUpdate(EventBuilder):
                     else:
                         # Should not happen
                         self._chat_peer = types.PeerUser(self._chat_peer)
-                except ValueError:
+                except KeyError:
                     # Hope for the best. We don't know where this event
                     # occurred but it was most likely in a channel.
                     self._chat_peer = types.PeerChannel(self._chat_peer)