mirror of
https://github.com/jarulsamy/Plex-Bot.git
synced 2024-08-19 15:01:55 +02:00
53 lines
1.2 KiB
Python
53 lines
1.2 KiB
Python
"""
|
|
Main entrypoint script.
|
|
Sets up loggers and initiates bot.
|
|
"""
|
|
import logging
|
|
|
|
from discord.ext.commands import Bot
|
|
|
|
from . import load_config
|
|
from .bot import General
|
|
from .bot import Plex
|
|
|
|
# Load config from file
|
|
configdir = "config"
|
|
from os import geteuid
|
|
if geteuid() == 0:
|
|
configdir = "/config"
|
|
config = load_config(configdir,"config.yaml")
|
|
|
|
BOT_PREFIX = config["discord"]["prefix"]
|
|
TOKEN = config["discord"]["token"]
|
|
|
|
BASE_URL = config["plex"]["base_url"]
|
|
PLEX_TOKEN = config["plex"]["token"]
|
|
LIBRARY_NAME = config["plex"]["library_name"]
|
|
|
|
if config["lyrics"]:
|
|
LYRICS_TOKEN = config["lyrics"]["token"]
|
|
else:
|
|
LYRICS_TOKEN = None
|
|
|
|
# Set appropiate log level
|
|
root_log = logging.getLogger()
|
|
plex_log = logging.getLogger("Plex")
|
|
bot_log = logging.getLogger("Bot")
|
|
|
|
plex_log.setLevel(config["plex"]["log_level"])
|
|
bot_log.setLevel(config["discord"]["log_level"])
|
|
|
|
plex_args = {
|
|
"base_url": BASE_URL,
|
|
"plex_token": PLEX_TOKEN,
|
|
"lib_name": LIBRARY_NAME,
|
|
"lyrics_token": LYRICS_TOKEN,
|
|
}
|
|
|
|
bot = Bot(command_prefix=BOT_PREFIX)
|
|
# Remove help command, we have our own custom one.
|
|
bot.remove_command("help")
|
|
bot.add_cog(General(bot))
|
|
bot.add_cog(Plex(bot, **plex_args))
|
|
bot.run(TOKEN)
|