def test_value_is_callable(): # pylint: disable=W0612
logger = Logger("test", handler=handler, template="{foo}: {message}")
func = lambda: "yay!"
logger.info("from lambda", foo=func)
output = get_output()
assert output == "yay!: from lambda\n"
def test_default_provided(): # pylint: disable=W0612
logger = Logger("test",
handler=handler,
template="{foo}: {message}",
foo="bar")
logger.info("ok")
output = get_output()
assert output == "bar: ok\n"
def method_level_gte_logger_level(): # pylint: disable=W0612
logger = Logger("test", level=LogLevel.INFO, handler=handler)
logger.info("this should print")
output = get_output()
assert re.match(
r"\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\.\d{6} INFO test: this should print\n",
output,
)
def test_value_not_provided(): # pylint: disable=W0612
logger = Logger("test", handler=handler, template="{foo}: {message}")
logger.info("not provided")
output = get_output()
assert output == "\b: not provided\n"
def test_value_provided_in_log_call(): # pylint: disable=W0612
logger = Logger("test", handler=handler, template="{foo}: {message}")
logger.info("ok", foo="baz")
output = get_output()
assert output == "baz: ok\n"
import psycopg2
import discord
from discord.ext import commands
from discord.ext.commands import has_role
import discord.ext.commands.errors as discord_errors
import typing
from simple_logging import Logger
from db_setup import setup_db
from config import *
log = Logger('ug_bot', log_folder=Path(LOG_FOLDER).resolve(),
log_level='debug', output_level='debug')
log.debug("Loaded all modules.")
log.info("Starting the bot...")
if not setup_db(log):
log.error("Unable to setup the database.")
sys.exit(1)
DB_URL = os.environ['DATABASE_URL']
db = psycopg2.connect(DB_URL)
log.debug("Connected to database.")
TOKEN = os.environ['DISCORD_TOKEN']
log.info("Token:", TOKEN)
bot = commands.Bot(command_prefix=COMMAND_PREFIX)
log.info("Bot created with prefix {}.".format(COMMAND_PREFIX))
