Example #1
0
def check_channel_whitelist(client, message):
	deny_all = []
	allow_all = ["politbiuro", "linki", "retro", "luzna_jazda"]
	unlimited = ["japabocie", "japa_bocie"]
	
	# default: disallow fulltext, enable flood control, enable bot
	permissions = {"fulltext" : False, "flood" : True, "disallow" : False}
	
	if str(message.channel).startswith("Direct Message"):
		sh.debug("Private message", message)
		permissions["flood"] = False
		permissions["fulltext"] = True
		
	elif str(message.channel) in deny_all:
		sh.debug("Blacklisted channel", message)
		permissions["disallow"] = True
		
	elif str(message.channel) in allow_all:
		sh.debug("Whitelisted channel", message)
		permissions["fulltext"] = True
		
	elif str(message.channel) in unlimited:
		sh.debug("Flood control inactive", message)
		permissions["flood"] = False
		permissions["fulltext"] = True
		
	else:
		sh.debug("Default options", message)
	
	return permissions
Example #2
0
def update_msg_deletion(msg, db=stats, cursor=stats_c):
	data = (msg.id, )
	
	cursor.execute("UPDATE messages SET deleted = 1 WHERE message = ? LIMIT 1", data)
	remove_emojis(msg, True)
	
	sh.debug("Deleted message {} from the stat database".format(msg.id))
Example #3
0
def update_msg_function(msg, funct, db=stats, cursor=stats_c, tmp=False):
	data = (funct, msg.id)
	
	cursor.execute("UPDATE messages SET function = ? WHERE message = ?", data)
	if not tmp:
		db.commit()
	
	sh.debug("Updated message function: {}".format(funct))
Example #4
0
def insert_msg(msg, db=stats, cursor=stats_c, tmp=False):
	data = (msg.id, msg.server.id if msg.server is not None else 0, msg.channel.id if msg.channel is not None else None, msg.author.id, msg.timestamp, msg.mention_everyone, 0, 1 if msg.author.bot else 0, None)
	
	cursor.execute("INSERT INTO messages VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)", data)
	if not tmp:
		db.commit()
	
	sh.debug("Inserted message {} into the stat database".format(msg.id))
Example #5
0
def check_flood(client, message):
    sh.debug("Checking for flood...")
    msg_limit = 5  # messages per channel
    time_limit = 5  # minutes

    if message.channel not in check:
        # channel not tracked yet
        check[message.channel] = {}

    if message.author not in check[message.channel]:
        # author on channel not tracked yet:
        check[message.channel][message.author] = {}
        check[message.channel][message.author]["timestamps"] = []
        check[message.channel][message.author]["warning_issued"] = False

    if len(check[message.channel][message.author]["timestamps"]) < msg_limit:
        # less than limit messages sent
        sh.debug(
            "Message number " +
            str(len(check[message.channel][message.author]["timestamps"]) + 1))
        check[message.channel][message.author]["timestamps"].append(
            time.time())
        return False

    else:
        time_diff = time.time() - check[message.channel][
            message.author]["timestamps"][0]
        if time_diff > (time_limit * 60):
            # message older than time limit
            sh.debug(
                "Message number " + str(msg_limit) + "; deleted timestamp " +
                str(check[message.channel][message.author]["timestamps"][0]),
                message)
            check[message.channel][message.author]["timestamps"].pop(0)
            check[message.channel][message.author]["timestamps"].append(
                time.time())
            check[message.channel][message.author]["warning_issued"] = False
            return False
        else:
            if check[message.channel][
                    message.author]["warning_issued"] == False:
                # scold user for spamming
                sh.debug("This user is spamming!")
                yield from client.send_message(
                    message.channel,
                    sh.mention(message) + "zamknij pizdę przez " +
                    datetime.datetime.utcfromtimestamp(
                        (time_limit * 60) - time_diff).strftime("%Mmin %Ss") +
                    ". Spamuj w <#386148571529084929>")
                check[message.channel][message.author]["warning_issued"] = True
                return True
            else:
                # ...but just once
                sh.debug("Shut him up pls")
                return False
Example #6
0
def check_flood_channel(client, message):
    sh.debug("Checking for channel flood...")
    msg_limit = 5  # messages per channel
    time_limit = 1  # minutes

    if message.channel not in check:
        # channel not tracked yet
        check[message.channel] = {}
        check[message.channel]["timestamps"] = []
        check[message.channel]["warning_issued"] = False

    if len(check[message.channel]["timestamps"]) < msg_limit:
        # less than limit messages sent
        sh.debug("Channel message number " +
                 str(len(check[message.channel]["timestamps"]) + 1))
        check[message.channel]["timestamps"].append(time.time())
        return False

    else:
        time_diff = time.time() - check[message.channel]["timestamps"][0]
        if time_diff > (time_limit * 60):
            # message older than time limit
            sh.debug(
                "Channel message number " + str(msg_limit) +
                "; deleted timestamp " +
                str(check[message.channel]["timestamps"][0]), message)
            check[message.channel]["timestamps"].pop(0)
            check[message.channel]["timestamps"].append(time.time())
            check[message.channel]["warning_issued"] = False
            return False
        else:
            if check[message.channel]["warning_issued"] == False:
                # scold these idiots for spamming
                sh.debug("This channel is being flooded!")
                client.send_message(
                    message.channel, "Weźcie wszyscy sklejcie pizdy przez " +
                    datetime.datetime.utcfromtimestamp(
                        (time_limit * 60) - time_diff).strftime("%Mmin %Ss") +
                    ". Od spamowania jest <#386148571529084929>")
                check[message.channel]["warning_issued"] = True
                return True
            else:
                # ...but just once
                sh.debug("I CAN STILL HEAR VOICES")
                return False
Example #7
0
def on_message_delete(message):
    billy_c_stats.update_msg_deletion(message)

    content = sh.rm_leading_quotes(message)

    if message.author == client.user or not message.server or not re.match(
            client.command_prefix, content):
        return

    sh.debug("User message deleted: " + message.content, message)

    for log in reversed(
            list((yield from client.logs_from(message.channel,
                                              limit=50,
                                              after=message)))):
        if log.author != client.user or not log.content.startswith(
                "<@!" + message.author.id + ">:"):
            continue

        sh.debug("Found a message to delete: " + log.content, message)

        yield from client.delete_message(log)
        return

    sh.debug("No matching bot response found")
Example #8
0
def check_channel_whitelist(client, message):
    deny_all = []
    # pecetgej, politbiuro, luzna_jazda, japabocie, japa_bocie
    allow_fulltext = [
        "318733700160290826", "174449535811190785", "316323075622961152",
        "319056762814595076", "386148571529084929"
    ]
    # japabocie, japa_bocie, sesje_rpg, sun_world, kanau_fela
    unlimited = [
        "316323075622961152", "319056762814595076", "386148571529084929",
        "174541542923436032", "539154754631106584", "232881423604776960"
    ]

    # default: disallow fulltext, enable flood control, enable bot
    permissions = {"fulltext": False, "flood": True, "disallow": False}

    if str(message.channel).startswith("Direct Message"):
        sh.debug("Private message", message)
        permissions["flood"] = False
        permissions["fulltext"] = True

    else:
        if message.channel.id in deny_all:
            #sh.debug("Blacklisted channel", message)
            permissions["disallow"] = True

        if message.channel.id in allow_fulltext:
            #sh.debug("Whitelisted channel", message)
            permissions["fulltext"] = True

        if message.channel.id in unlimited:
            #sh.debug("Flood control inactive", message)
            permissions["flood"] = False

    sh.debug("Received message", message)

    return permissions
Example #9
0
def parse_message(message, edited=False):
    if not edited:
        billy_c_stats.insert_msg(message)

    # Track used emojis

    emoji_list = list(
        c for c in message.clean_content if c in emoji.UNICODE_EMOJI) or []
    custom_emoji_list = re.findall(r"<:\S+?:\d+>", message.clean_content,
                                   re.IGNORECASE) or []
    billy_c_stats.insert_emojis_post(message, emoji_list, custom_emoji_list,
                                     edited)

    # ignore bot messages

    if message.author == client.user:
        return

    perm = af.check_channel_whitelist(client, message)

    # channel blacklisted

    if perm["disallow"]:
        return

    # strip quotes
    content = sh.rm_leading_quotes(message)

    if not edited:
        # fulltext search
        for f in f_functions:
            c = getattr(f, "command", False)
            p = getattr(f, "prob", False)

            if c and p and re.search(c, content, re.IGNORECASE) and (
                    p >= 1.0 or (perm["fulltext"] and random.random() < p)):
                sh.debug("Triggered " + f.__name__ + "...")
                try:
                    yield from f(client, message)
                except Exception:
                    print_warning("An error occured in " + f.__name__ +
                                  "!!! (" + content + ")")
                    raise

    # commands

    if re.match(client.command_prefix, content):
        sh.debug("This seems to be a command: " + sh.get_command(message))

        # check antiflood

        #if perm["flood"] and ((yield from af.check_flood_channel(client, message)) or (yield from af.check_flood(client, message))):
        #	sh.debug("Anti-flood kicked in yo")
        #	return

        # help

        # not needed here

        #else:
        # iterate over functions
        for f in c_functions:
            c = getattr(f, "command", False)
            r = re.match(compile_command(c), content, re.IGNORECASE)
            if c and r:
                sh.debug("Executing " + f.__name__ + "...")
                yield from client.send_typing(message.channel)
                try:
                    yield from f(client, message)
                    billy_c_stats.update_msg_function(message, f.__name__)
                except Exception:
                    yield from client.send_message(
                        message.channel,
                        "Oho, chyba jakiś błąd w kodzie. <@307949259658100736> to kiedyś naprawi, jak się skończy bawić pociągami."
                    )
                    print_warning("An error occured in " + f.__name__ +
                                  "!!! (" + content + ")")
                    #CZEMU TY CHUJU NIE DZIALASZ
                    #logging.exception("An error occured in " + f.__name__ + "!!! (" + content + ")")
                    raise
                    continue
                break
Example #10
0
        module = sys.modules[module_name]

        for name, val in module.__dict__.items():
            if callable(val) and name.startswith(("c_", "f_", "t_")):
                if name.startswith("t_"):
                    if hasattr(val, "time") and not re.match(
                            r"([0-9]|[0-1][0-9]|2[0-3])\:[0-5][0-9]",
                            getattr(val, "time")):
                        print("Invalid specified time: " + name)
                        continue
                    elif not hasattr(val, "channels"):
                        print("No channels specified: " + name)
                        continue
                    else:
                        # Timer functions functions
                        sh.debug("Imported timer: " + name)
                        t_functions.append(val)
                        i += 1

                elif not hasattr(val, "command"):
                    # Omit functions without specified .command
                    print("Missing command regex: " + name)
                    continue

                elif name.startswith("f_") and not hasattr(val, "prob"):
                    # Omit fulltext without specified .prob
                    print("Missing fulltext probability: " + name)
                    continue

                elif name.startswith("c_"):
                    # Called functions
Example #11
0
def parse_message(message, edited=False):
    #if not edited:
    #	billy_c_stats.insert_msg(message)

    # Track used emojis

    #emoji_list = list(c for c in message.clean_content if c in emoji.UNICODE_EMOJI) or []
    #custom_emoji_list = re.findall(r"<:\S+?:\d+>", message.clean_content, re.IGNORECASE) or []
    #billy_c_stats.insert_emojis_post(message, emoji_list, custom_emoji_list, edited)

    # ignore bot messages

    if message.author == client.user:
        return

    perm = af.check_channel_whitelist(client, message)

    # channel blacklisted

    if perm["disallow"]:
        return

    # strip quotes
    content = unidecode.unidecode(sh.rm_leading_quotes(message))

    if not edited:
        # fulltext search
        for f in f_functions:
            c = getattr(f, "command", False)
            p = getattr(f, "prob", False)

            if c and p and re.search(c, content, re.IGNORECASE) and (
                    p >= 1.0 or (perm["fulltext"] and random.random() < p)):
                sh.debug("Triggered " + f.__name__ + "...")
                try:
                    yield from f(client, message)
                except Exception:
                    print_warning("An error occured in " + f.__name__ +
                                  "!!! (" + content + ")")
                    raise

    # commands

    if re.match(client.command_prefix, content):
        sh.debug("This seems to be a command: " + sh.get_command(message))

        # check antiflood

        if perm["flood"] and (
            (yield from af.check_flood_channel(client, message)) or
            (yield from af.check_flood(client, message))):
            sh.debug("Anti-flood kicked in yo")
            return

        # help

        if re.match(compile_command(r"(help|pomoc)"), content, re.IGNORECASE):
            sh.debug("What a noob")
            ret = "Witam witam, z tej strony Billy Mays z kolejnym fantastycznym produktem!\nDozwolone przedrostki funkcji: . , \ / ! ;\n\n"

            for f in c_functions:
                desc = getattr(f, "desc", False)

                if hasattr(f, "rhyme") or desc == "hidden":
                    continue

                command = getattr(f, "command", False)

                ret += "." + getattr(f, "command")

                params = getattr(f, "params", False)

                if params:
                    for p in params:
                        ret += " [" + p + "]"

                if desc:
                    ret += " - " + desc

                ret += "\n"

            ret += "\nRymy i inne bzdety: .rymy"
            ret += "\nZadzwoń teraz, a podwoimy ofertę!"

            if len(ret) > 2000:
                n = 40
                groups = ret.split("\n")
                help = ["\n".join(groups[:n]), "\n".join(groups[n:])]
            else:
                help = [ret]

            for m in help:
                yield from client.send_message(message.channel, m)

        elif re.match(compile_command(r"(rymy|rhymes)"), content,
                      re.IGNORECASE):
            sh.debug("What an utter pillock")
            ret = "Rymy i inne bzdety:\n"

            for f in c_functions:
                if not hasattr(f, "rhyme"):
                    continue
                command = getattr(f, "command", False)
                ret += "." + getattr(f, "command") + "\n"

            yield from client.send_message(message.channel, ret[:-1])

        else:
            # iterate over functions
            for f in c_functions:
                c = getattr(f, "command", False)
                r = re.match(compile_command(c), content, re.IGNORECASE)
                if c and r:
                    sh.debug("Executing " + f.__name__ + "...")
                    yield from client.send_typing(message.channel)
                    try:
                        yield from f(client, message)
                        #billy_c_stats.update_msg_function(message, f.__name__)
                    except Exception:
                        yield from client.send_message(
                            message.channel,
                            "Oho, chyba jakiś błąd w kodzie. <@307949259658100736> to kiedyś naprawi, jak się skończy bawić pociągami."
                        )
                        print_warning("An error occured in " + f.__name__ +
                                      "!!! (" + content + ")")
                        #CZEMU TY CHUJU NIE DZIALASZ
                        #logging.exception("An error occured in " + f.__name__ + "!!! (" + content + ")")
                        raise
                        continue
                    break
Example #12
0
def on_message(message):
    #ignore bot messages

    if message.author == client.user:
        return

    perm = af.check_channel_whitelist(client, message)

    # channel blacklisted

    if perm["disallow"]:
        return

    # strip quotes
    content = sh.rm_leading_quotes(message)

    if perm["fulltext"]:
        # fulltext search
        for f in f_functions:
            c = getattr(f, "command", False)
            p = getattr(f, "prob", False)

            if c and p and re.search(c, content,
                                     re.IGNORECASE) and random.random() < p:
                sh.debug("Triggered " + f.__name__ + "...")
                try:
                    yield from f(client, message)
                except Exception:
                    print("An error occured in " + f.__name__ + "!!! (" +
                          content + ")")
                    raise
                    continue

    # commands

    if re.match(client.command_prefix, content):
        sh.debug("This seems to be a command: " + sh.get_command(message))

        # check antiflood

        if perm["flood"] and (
            (yield from af.check_flood_channel(client, message)) or
            (yield from af.check_flood(client, message))):
            sh.debug("Anti-flood kicked in yo")
            return

        # help

        if re.match(compile_command(r"(help|pomoc)"), content, re.IGNORECASE):
            sh.debug("What a noob")
            ret = "Witam witam, z tej strony Billy Mays z kolejnym fantastycznym produktem!\nDozwolone przedrostki funkcji: . , \ / !\n\n"

            for f in c_functions:
                desc = getattr(f, "desc", False)

                if hasattr(f, "rhyme") or desc == "hidden":
                    continue

                command = getattr(f, "command", False)

                ret += "." + getattr(f, "command")

                params = getattr(f, "params", False)

                if params:
                    for p in params:
                        ret += " [" + p + "]"

                if desc:
                    ret += " - " + desc

                ret += "\n"

            ret += "\nRymy i inne bzdety: .rymy"
            ret += "\nZadzwoń teraz, a podwoimy ofertę!"

            yield from client.send_message(message.channel, ret)

        elif re.match(compile_command(r"(rymy|rhymes)"), content,
                      re.IGNORECASE):
            sh.debug("What an utter pillock")
            ret = "Rymy i inne bzdety:\n"

            for f in c_functions:
                if not hasattr(f, "rhyme"):
                    continue
                command = getattr(f, "command", False)
                ret += "." + getattr(f, "command") + "\n"

            yield from client.send_message(message.channel, ret[:-1])

        else:
            # iterate over functions
            for f in c_functions:
                c = getattr(f, "command", False)
                r = re.match(compile_command(c), content, re.IGNORECASE)
                if c and r:
                    sh.debug("Executing " + f.__name__ + "...")
                    yield from client.send_typing(message.channel)
                    try:
                        yield from f(client, message)
                    except Exception:
                        yield from client.send_message(
                            message.channel,
                            "Oho, chyba jakiś błąd w kodzie. <@307949259658100736> to kiedyś naprawi, jak się skończy bawić pociągami."
                        )
                        print("An error occured in " + f.__name__ + "!!! (" +
                              content + ")")
                        #CZEMU TY CHUJU NIE DZIALASZ
                        #logging.exception("An error occured in " + f.__name__ + "!!! (" + content + ")")
                        raise
                        continue
                    break

        return