OrthoBot/src/central.py

"""
    Copyright (c) 2018 Elliott Pardee <me [at] vypr [dot] xyz>
    This file is part of OrthoBot.

    OrthoBot is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    OrthoBot is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with OrthoBot.  If not, see <http://www.gnu.org/licenses/>.
"""

import configparser
import math
import os
import time
import numbers

import tinydb

from extensions.vylogger import VyLogger

dir_path = os.path.dirname(os.path.realpath(__file__))

config = configparser.ConfigParser()
config.read(dir_path + "/config.ini")

configVersion = configparser.ConfigParser()
configVersion.read(dir_path + "/config.example.ini")

version = "OrthoBot v" + configVersion["meta"]["version"]
icon = "https://cdn.discordapp.com/avatars/479471696093642762/a2deba76dda32dc874a75ec5c5025f37.png"  # noqa: E501

logger = VyLogger("default")

guildDB = tinydb.TinyDB(dir_path + "/../databases/guilddb")
banDB = tinydb.TinyDB(dir_path + "/../databases/bandb")


def capitalize_first_letter(string):
    return string[0].upper() + string[1:]


def splitter(s):
    middle = math.floor(len(s) / 2)
    before = s.rfind(" ", middle)
    after = s.index(" ", middle + 1)

    if (middle - before) < (after - middle):
        middle = after
    else:
        middle = before

    return {
        "first": s[0:middle],
        "second": s[middle + 1:]
    }


def log_message(level, shard, sender, source, msg):
    message = "[shard " + str(shard) + "] <" + \
        sender + "@" + source + "> " + msg

    if level == "warn":
        logger.warning(message)
    elif level == "err":
        logger.error(message)
    elif level == "info":
        logger.info(message)
    elif level == "debug":
        logger.debug(message)


def add_ban(entryid, reason):
    ideal_entry = tinydb.Query()
    result = banDB.search(ideal_entry.id == entryid)

    if len(result) > 0:
        return False
    else:
        banDB.insert({"id": entryid, "reason": reason})
        return True


def remove_ban(entryid):
    ideal_entry = tinydb.Query()
    result = banDB.search(ideal_entry.id == entryid)

    if len(result) > 0:
        banDB.remove(ideal_entry.id == entryid)
        return True
    else:
        return False


def is_banned(entryid):
    ideal_entry = tinydb.Query()
    result = banDB.search(ideal_entry.id == entryid)

    if len(result) > 0:
        if "reason" in result[0].keys():
            return True, result[0]["reason"]
        else:
            return True, "Unspecified"
    else:
        return False, None


def add_optout(entryid):
    ideal_entry = tinydb.Query()
    result = optoutDB.search(ideal_entry.id == entryid)

    if len(result) > 0:
        return False
    else:
        db.remove(ideal_entry.id == entryid)
        guildDB.remove(ideal_entry.id == entryid)

        optoutDB.insert({"id": entryid})
        return True


def remove_optout(entryid):
    ideal_entry = tinydb.Query()
    result = optoutDB.search(ideal_entry.id == entryid)

    if len(result) > 0:
        optoutDB.remove(ideal_entry.id == entryid)
        return True
    else:
        return False


def is_optout(entryid):
    ideal_entry = tinydb.Query()
    result = optoutDB.search(ideal_entry.id == entryid)

    if len(result) > 0:
        return True
    else:
        return False


def is_snowflake(snowflake):
    try:
        if isinstance(int(snowflake), numbers.Number):
            if int(snowflake) > 1.4200704e+16:
                return True
    except ValueError:
        return False


def sleep(milliseconds):
    time.sleep(milliseconds / 1000.0)