161 lines
3.9 KiB
Python
161 lines
3.9 KiB
Python
"""
|
|
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)
|