modules/CommandRoom.py

   1 # Mattermost Bot module.
   2 #  Copyright (c) 2016-2022 by Someone <someone@somenet.org> (aka. Jan Vales <jan@jvales.net>)
   3 #  published under MIT-License
   4 #  previous implementation written 2019 by Ju <daju@fsinf.at> was completely replaced by a modified previous CommandToss implementation by Someone <someone@somenet.org> in 2020.
   5
   6 import requests
   7
   8
   9 # pylint: disable=wrong-import-position
  10 from AbstractCommand import AbstractCommand
  11 class CommandRoom(AbstractCommand):
  12     TRIGGER = "room"
  13     CONFIG = {"display_name": "somebot-command", "auto_complete": True,
  14               "auto_complete_hint": "['--chan'] <room name>",
  15              }
  16     USEINFO = CONFIG["auto_complete_desc"] = CONFIG["description"] = "Use TOSS to get room information."
  17
  18
  19     def __init__(self, team_id, toss_url_api_search):
  20         super().__init__(team_id)
  21         self.toss_url_api_search = toss_url_api_search
  22
  23
  24     def on_POST(self, request, data):
  25         search = data["text"]
  26         send_to_chan = False
  27
  28         if search.startswith("--chan "):
  29             search = search.replace("--chan ", "", 1)
  30             send_to_chan = True
  31
  32         if not search or search == "--chan":
  33             request.cmd_respond_text_temp("You did not provide a room name.\n``/"+self.TRIGGER+" "+data["text"]+"``")
  34             return
  35
  36         r = requests.get(self.toss_url_api_search+"?q="+requests.utils.quote(search))
  37         if r.status_code != 200:
  38             request.cmd_respond_text_temp(":stop_sign: TOSS-query failed. :(\n"+str(repr(r))+"\n``/"+self.TRIGGER+" "+data["text"]+"``")
  39             return
  40
  41         msg = "#### ** [:toss: ``Results``](https://toss.fsinf.at/?q="+requests.utils.quote(search)+") ``(max 5)``**"
  42         cnt = 0
  43         for entry in r.json():
  44             if entry["type"] == "room":
  45                 cnt += 1
  46                 if cnt > 5:
  47                     break
  48
  49                 msg += "\n#### "+entry["name"]+" ("+entry["code"]+")\n"
  50                 if entry["address"]:
  51                     msg += ":world_map:[``"+entry["address"]+"``](https://www.google.com/maps/search/"+requests.utils.quote(entry["address"].split(";")[0])+")"
  52                 if entry["description"]:
  53                     msg += "\n"+entry["description"]
  54                 msg += "\n\n----\n\n"
  55
  56         if cnt == 0:
  57             request.cmd_respond_text_temp("#### **``No`` :toss: ``Results``** :(\n``/"+self.TRIGGER+" "+data["text"]+"``")
  58         elif not send_to_chan:
  59             request.cmd_respond_text_temp(msg+"\n``/"+self.TRIGGER+" "+data["text"]+"``")
  60         else:
  61             request.cmd_respond_text_chan(msg+"\n``/"+self.TRIGGER+" "+data["text"]+"``")