Fix PollBot._on_poll_button_press

This commit is contained in:
AlberLC
2022-12-03 06:25:31 +01:00
parent 9925084ef4
commit 96d117a9fe

View File

@@ -198,13 +198,13 @@ class PollBot(MultiBot, ABC):
option_name = results[0] if (results := re.findall('(.*?) ➜.+', message.buttons_info.pressed_text)) else message.buttons_info.pressed_text option_name = results[0] if (results := re.findall('(.*?) ➜.+', message.buttons_info.pressed_text)) else message.buttons_info.pressed_text
selected_option_votes = message.buttons_info.data['votes'][option_name] selected_option_votes = message.buttons_info.data['votes'][option_name]
if [presser_id, presser_name] in selected_option_votes: if (presser_id, presser_name) in selected_option_votes:
selected_option_votes.remove([presser_id, presser_name]) selected_option_votes.remove((presser_id, presser_name))
else: else:
if not message.buttons_info.data['is_multiple_answer']: if not message.buttons_info.data['is_multiple_answer']:
for option_votes in message.buttons_info.data['votes'].values(): for option_votes in message.buttons_info.data['votes'].values():
try: try:
option_votes.remove([presser_id, presser_name]) option_votes.remove((presser_id, presser_name))
except ValueError: except ValueError:
pass pass
else: else: