checkers.apology

  1from . import Checker
  2import src.utils.db
  3
  4
  5class ApologyChecker(Checker):
  6    def __init__(self):
  7        self.conn = src.utils.db.DbConnector()
  8        self.conn.connect()
  9
 10    def check_message(self, message):
 11        """
 12        Just checks for Sorry message
 13        :param message: message string
 14        :return:True or False
 15        """
 16        return message.find("sorry") != -1
 17
 18    def check_user_for_warning(self, user_id, server_name):
 19        """
 20        Function to check whether a user has any outstanding warnings
 21        """
 22        try:
 23            self.add_user(user_id, server_name)
 24            out = 0
 25            cursor = self.conn.connector.cursor()
 26            sql_query = "SELECT offense_count - apology_count FROM discorddb.user_activity WHERE user_id = %s AND server_name = %s"
 27            cursor.execute(sql_query, (user_id, server_name))
 28            result = cursor.fetchone()
 29            out = result[0]
 30        except Exception as error:
 31            print("Failed to get record from MySQL table: {}".format(error))
 32        finally:
 33            cursor.close()
 34            if out > 0:
 35                print("Offenses more than expected. Banning User")
 36                self.ban_user(user_id, server_name, 1)
 37                return False
 38
 39            self.ban_user(user_id, server_name, 0)
 40            return True
 41
 42    def add_warning(self, user_id, server_name):
 43        try:
 44            self.add_user(user_id, server_name)
 45            cursor = self.conn.connector.cursor()
 46            sql_query = "UPDATE discorddb.user_activity SET offense_count = offense_count+1 WHERE user_id = %s AND server_name = %s"
 47
 48            val = (user_id, server_name)
 49            cursor.execute(sql_query, val)
 50            self.conn.connector.commit()
 51        except Exception as error:
 52            print("Failed to get record from MySQL table: {}".format(error))
 53        finally:
 54            cursor.close()
 55            return True
 56
 57    def add_user(self, user_id, server_name):
 58        """
 59        Function to add user onto the database. If already banned, can't add user.
 60        :param user_id:
 61        :param server_name:
 62        :return:
 63        """
 64        try:
 65            cursor = self.conn.connector.cursor()
 66            sql_query = "SELECT is_banned FROM discorddb.user_activity WHERE user_id = %s AND server_name = %s"
 67            cursor.execute(sql_query, (user_id, server_name))
 68            result = cursor.fetchone()
 69            if not result:
 70                print("User not present in the channel.")
 71                sql_query = "INSERT INTO discorddb.user_activity (user_id, server_name) VALUES (%s,%s)"
 72                val = (user_id, server_name)
 73                cursor.execute(sql_query, val)
 74                self.conn.connector.commit()
 75                print("User is added into the channel")
 76            else:
 77                if result[0] == 1:
 78                    print("User is banned from this server. Can't insert into the channel")
 79
 80        except Exception as error:
 81            print("Failed to get record from MySQL table: {}".format(error))
 82        finally:
 83            cursor.close()
 84            return True
 85
 86    def add_apology(self, user_id, server_name):
 87        try:
 88            self.add_user(user_id, server_name)
 89            cursor = self.conn.connector.cursor()
 90            sql_query = "UPDATE discorddb.user_activity SET apology_count = apology_count+1 WHERE user_id = %s AND server_name = %s"
 91            val = (user_id, server_name)
 92            cursor.execute(sql_query, val)
 93            sql_query = "UPDATE discorddb.user_activity SET offense_count = CASE WHEN offense_count < apology_count THEN apology_count ELSE offense_count END WHERE user_id = %s AND server_name = %s"
 94            cursor.execute(sql_query, val)
 95            self.conn.connector.commit()
 96        except Exception as error:
 97            print("Failed to get record from MySQL table: {}".format(error))
 98        finally:
 99            cursor.close()
100            return True
101
102    def ban_user(self, user_id, server_name, is_banned):
103        try:
104            self.add_user(user_id, server_name)
105            cursor = self.conn.connector.cursor()
106            sql_query = "UPDATE discorddb.user_activity SET is_banned = %s WHERE user_id = %s AND server_name = %s"
107            val = (is_banned, user_id, server_name)
108            cursor.execute(sql_query, val)
109            self.conn.connector.commit()
110        except Exception as error:
111            print("Failed to get record from MySQL table: {}".format(error))
112        finally:
113            cursor.close()
114            return True