\beer PoC

Author: fservida

aesc_bot/bot.py

@@ -42,6 +42,8 @@ def create_bot(**kwargs):
     dispatcher.add_handler(CommandHandler('version', version))
     dispatcher.add_handler(CommandHandler('menu', menu))
     dispatcher.add_handler(CallbackQueryHandler(menu_handler, pattern=r"menu_.*"))
+    dispatcher.add_handler(CommandHandler('beers', beers))
+    dispatcher.add_handler(CallbackQueryHandler(beers_handler, pattern=r"beers_.*"))
 
     dispatcher.add_handler(MessageHandler(Filters.text, echo))
 

aesc_bot/commands.py

@@ -18,6 +18,8 @@
 import requests
 from requests import RequestException
 from datetime import datetime, timedelta
+from aesc_bot.parsers import parse_beer_types
+from aesc_bot.formatters import format_beer_types
 
 CACHE_MENU_PATH = "cache/menu/"
 ACTIVITIES = os.environ.get("ACTIVITY_YAML",
@@ -31,6 +33,16 @@
     "Unithèque": "unitheque",
 }
 
+beers_buttons = {
+    "Types": "types",
+}
+
+beers_functions = {
+    'types': {
+        'parser': parse_beer_types,
+        'formatter': format_beer_types,
+    },
+}
 
 # start
 def start(bot, update):
@@ -136,6 +148,27 @@ def menu_handler(bot, update):
                           parse_mode='Markdown')
 
 
+def beers(bot, update):
+    button_list = [InlineKeyboardButton(beers_button, callback_data="beers_%s" % beers_button_ref) for beers_button, beers_button_ref in
+                   beers_buttons.items()]
+
+    reply_markup = InlineKeyboardMarkup(build_menu(button_list, n_cols=2))
+
+    update.message.reply_text("Fait ton choix mon jeune padawan!", reply_markup=reply_markup)
+
+
+def beers_handler(bot, update):
+    query = update.callback_query
+    button = query.data.replace("beers_", "")
+
+    message = beers_functions[button]['formatter'](beers_functions[button]['parser']())[:4095]
+
+    bot.edit_message_text(chat_id=query.message.chat_id,
+                          message_id=query.message.message_id,
+                          text=message,
+                          parse_mode='Markdown')
+
+
 def version(bot, update):
     conf = Configuration.get_instance()
     bot.send_message(chat_id=update.message.chat_id,

aesc_bot/formatters.py

@@ -0,0 +1,9 @@
+# formatters.py
+# Project: aesc_bot
+# 
+# Created by "Francesco Servida"
+# Created on 13.06.18
+
+
+def format_beer_types(beer_types):
+    return "\n".join(["*{}* - {}".format(beer_type, description) for beer_type, description in beer_types.items()])

aesc_bot/parsers.py

@@ -0,0 +1,48 @@
+# parsers.py
+# Project: aesc_bot
+# 
+# Created by "Francesco Servida"
+# Created on 13.06.18
+
+import requests
+from requests import RequestException
+import json
+import bs4
+
+from pprint import pprint
+
+BEER_TYPES = "https://satellite.bar/bar/biere.php?type_biere=sortes"
+
+
+def parse_beer_types():
+    try:
+        response = requests.get(BEER_TYPES)
+        html = response.text
+    except (RequestException):
+        return []
+
+    soup = bs4.BeautifulSoup(html, "html.parser")
+    beer_types_raw = soup.select("div#main div.text_block")[2:]
+
+    beer_types = {}
+    for beer_type in beer_types_raw:
+        try:
+            beer = beer_type.select_one("h2").text
+            type_desc = " ".join(
+                [line.strip() for line in beer_type.select_one("div.text_block_text").text.split("\n")]).strip()
+
+            beer_types[beer] = type_desc
+
+        except AttributeError:
+            pass
+
+    return beer_types
+
+
+def beers_to_json():
+    beer_types = parse_beer_types()
+    pprint(beer_types)
+
+
+if __name__ == '__main__':
+    beers_to_json()