modules/WSAlliterationAssociation.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
   5 import json
   6 import re
   7
   8 from AbstractWSHandler import *
   9 class WSAlliterationAssociation(AbstractWSHandler):
  10     NAME = "alliterationassociation"
  11
  12     def __init__(self, channel_ids):
  13         super().__init__()
  14
  15         self.channel_ids = channel_ids
  16
  17
  18     def on_WS_EVENT(self, data):
  19         post = json.loads(data["data"]["post"])
  20
  21         if post["channel_id"] not in self.channel_ids:
  22             return False
  23
  24         if post["user_id"] == self.bot.api._my_user_id:
  25             return False
  26
  27         message = re.sub(r"\s+", " ", re.sub(r"[^\w\.\:;\!\?~@\n ]+", " ", post["message"]))
  28
  29         sentences = [s.strip() for s in re.split(r"\.|\:|;|\!|\?|~|@|\n", message)]
  30         for sentence in sentences:
  31             words = sentence.lower().split()
  32             for word in words:
  33                 if word[0] != words[0][0] and post["type"] in ["", "me"]:
  34                     if "root_id" not in post or post["root_id"] == "":
  35                         self.bot.api.create_post(post["channel_id"], "``"+sentence+"`` is not an alliteration! Kicking...", root_id=post["id"])
  36                     else:
  37                         self.bot.api.create_post(post["channel_id"], "``"+sentence+"`` is not an alliteration! Kicking...", root_id=post["root_id"])
  38                     self.bot.api.remove_user_from_channel(post["channel_id"], post["user_id"])
  39                     return True
  40
  41
  42         # make user to channel admin, once they manage to say one alliteration.
  43         self.bot.api.update_channel_members_scheme_roles(post["channel_id"], post["user_id"], {"scheme_user": True, "scheme_admin": True})
  44
  45         return True