From 0b19607095d4cead975705e905049cf48e797783 Mon Sep 17 00:00:00 2001
From: fmk17 <fmk17@inf.ufpr.br>
Date: Wed, 9 Feb 2022 21:31:53 -0300
Subject: [PATCH] Fix not accepting @

---
 main.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/main.py b/main.py
index 3204158..dea29ed 100644
--- a/main.py
+++ b/main.py
@@ -58,6 +58,9 @@ LOCATION_TO_HEADER = {
 
 def cardapio(update: Update, context: CallbackContext) -> None:
     command = next(iter(update.message.parse_entities(types=["bot_command"]).values()))[1:]
+    if '@' in command:
+        command = command[:command.index('@')]
+
     location = COMMAND_TO_LOCATION[command]
     days, update_time = get_menus_by_days(location)
 
-- 
GitLab