From 8baa4dcc599ea7a974644c8e504a67f07911cecc Mon Sep 17 00:00:00 2001
From: Gabriel RF <gabriel.ferreira@avidity.se>
Date: Wed, 15 Mar 2023 16:15:18 -0300
Subject: [PATCH] =?UTF-8?q?Adicionada=20a=20vari=C3=A1vel=20HIDE=5FBUTTON?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .github/workflows/cron.yml |  1 +
 README.md                  |  2 ++
 rss2telegram.py            | 27 ++++++++++++++++-----------
 3 files changed, 19 insertions(+), 11 deletions(-)

diff --git a/.github/workflows/cron.yml b/.github/workflows/cron.yml
index 1247cb0..c51727e 100644
--- a/.github/workflows/cron.yml
+++ b/.github/workflows/cron.yml
@@ -36,6 +36,7 @@ jobs:
           URL: ${{ vars.URL }}
           EMOJIS: ${{ vars.EMOJIS }}
           TOPIC: ${{ vars.TOPIC }}
+          HIDE_BUTTON: ${{ vars.HIDE_BUTTON }}
           MESSAGE_TEMPLATE: ${{ vars.MESSAGE_TEMPLATE }}
           BUTTON_TEXT: ${{ vars.BUTTON_TEXT }}
           PARAMETERS: ${{ vars.PARAMETERS }}
diff --git a/README.md b/README.md
index 4c34e93..08f7427 100644
--- a/README.md
+++ b/README.md
@@ -34,6 +34,8 @@ Defina as variáveis na aba `Variables` do repositório:
 
 `TOPIC`: (opcional) ID do tópico em que a mensagem será enviada. Necessário para grupos com a opção de tópicos ativada. [Como obter um ID de um tópico](#id-de-tópico)
 
+`HIDE_BUTTON`: (opcional) Caso definida, desabilita o botão no envio, permitindo assim a existência do `Leitura Rápida`.
+
 ### Opções de variáveis
 
 `{SITE_NAME}`: Nome do site;
diff --git a/rss2telegram.py b/rss2telegram.py
index 70645e4..3610dc7 100644
--- a/rss2telegram.py
+++ b/rss2telegram.py
@@ -21,6 +21,7 @@ def get_variable(variable):
 BOT_TOKEN = os.environ.get('BOT_TOKEN')
 EMOJIS = os.environ.get('EMOJIS', '🗞,📰,🗒,🗓,📋,🔗,📝,🗃')
 PARAMETERS = os.environ.get('PARAMETERS', False)
+HIDE_BUTTON = os.environ.get('HIDE_BUTTON', False)
 DRYRUN = os.environ.get('DRYRUN')
 TOPIC = os.environ.get('TOPIC', False)
 
@@ -81,19 +82,23 @@ def send_message(topic, button):
         btn = types.InlineKeyboardButton(f'{button}', url=topic['link'])
         btn_link.row(btn)
 
-    if topic['photo']:
-        response = requests.get(topic['photo'], headers = {'User-agent': 'Mozilla/5.1'})
-        open('img', 'wb').write(response.content)
+    if HIDE_BUTTON:
         for dest in DESTINATION.split(','):
-            photo = open('img', 'rb')
-            try:
-                bot.send_photo(dest, photo, caption=MESSAGE_TEMPLATE, parse_mode='HTML', reply_markup=btn_link, reply_to_message_id=TOPIC)
-            except telebot.apihelper.ApiTelegramException:
-                topic['photo'] = False
-                send_message(topic, button)
+            bot.send_message(dest, MESSAGE_TEMPLATE, parse_mode='HTML', reply_to_message_id=TOPIC)
     else:
-        for dest in DESTINATION.split(','):
-            bot.send_message(dest, MESSAGE_TEMPLATE, parse_mode='HTML', reply_markup=btn_link, disable_web_page_preview=True, reply_to_message_id=TOPIC)
+        if topic['photo']:
+            response = requests.get(topic['photo'], headers = {'User-agent': 'Mozilla/5.1'})
+            open('img', 'wb').write(response.content)
+            for dest in DESTINATION.split(','):
+                photo = open('img', 'rb')
+                try:
+                    bot.send_photo(dest, photo, caption=MESSAGE_TEMPLATE, parse_mode='HTML', reply_markup=btn_link, reply_to_message_id=TOPIC)
+                except telebot.apihelper.ApiTelegramException:
+                    topic['photo'] = False
+                    send_message(topic, button)
+        else:
+            for dest in DESTINATION.split(','):
+                bot.send_message(dest, MESSAGE_TEMPLATE, parse_mode='HTML', reply_markup=btn_link, disable_web_page_preview=True, reply_to_message_id=TOPIC)
     print(f'... {topic["title"]}')
     time.sleep(0.2)