messages: Add logic to handle the SHOW_POLL_VOTES hotkey.

Author: rsashank

zulipterminal/ui_tools/messages.py

@@ -69,6 +69,7 @@ def __init__(self, message: Message, model: "Model", last_message: Any) -> None:
         self.topic_links: Dict[str, Tuple[str, int, bool]] = dict()
         self.time_mentions: List[Tuple[str, str]] = list()
         self.last_message = last_message
+        self.widget_type: str = ""
         # if this is the first message
         if self.last_message is None:
             self.last_message = defaultdict(dict)
@@ -733,9 +734,9 @@ def main_view(self) -> List[Any]:
             )
 
         if self.message.get("submessages"):
-            widget_type = find_widget_type(self.message.get("submessages", []))
+            self.widget_type = find_widget_type(self.message.get("submessages", []))
 
-            if widget_type == "todo":
+            if self.widget_type == "todo":
                 title, tasks = process_todo_widget(self.message.get("submessages", []))
 
                 todo_widget = "<strong>To-do</strong>\n" + f"<strong>{title}</strong>"
@@ -757,28 +758,28 @@ def main_view(self) -> List[Any]:
                 # though it's not very useful.
                 self.message["content"] = todo_widget
 
-            elif widget_type == "poll":
-                poll_question, poll_options = process_poll_widget(
+            elif self.widget_type == "poll":
+                self.poll_question, self.poll_options = process_poll_widget(
                     self.message.get("submessages", [])
                 )
 
                 # TODO: ZT doesn't yet support adding poll questions after the
                 # creation of the poll. So, if the poll question is not provided,
                 # we show a message to add one via the web app.
-                if not poll_question:
-                    poll_question = (
+                if not self.poll_question:
+                    self.poll_question = (
                         "No poll question is provided. Please add one via the web app."
                     )
 
-                poll_widget = f"<strong>Poll\n{poll_question}</strong>"
+                poll_widget = f"<strong>Poll\n{self.poll_question}</strong>"
 
-                if poll_options:
+                if self.poll_options:
                     max_votes_len = max(
                         len(str(len(option["votes"])))
-                        for option in poll_options.values()
+                        for option in self.poll_options.values()
                     )
 
-                    for option_info in poll_options.values():
+                    for option_info in self.poll_options.values():
                         padded_votes = f"{len(option_info['votes']):>{max_votes_len}}"
                         poll_widget += f"\n[ {padded_votes} ] {option_info['option']}"
                 else:
@@ -1188,4 +1189,6 @@ def keypress(self, size: urwid_Size, key: str) -> Optional[str]:
             self.model.controller.show_emoji_picker(self.message)
         elif is_command_key("MSG_SENDER_INFO", key):
             self.model.controller.show_msg_sender_info(self.message["sender_id"])
+        elif is_command_key("SHOW_POLL_VOTES", key) and self.widget_type == "poll":
+            self.model.controller.show_poll_vote(self.poll_question, self.poll_options)
         return key