SantaBot/santa-bot.py at master · avchikar97/SantaBot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import logging
import os
import pathlib

from configobj import ConfigObj
from discord.ext import commands
from discord import Intents

import CONFIG
from cogs.SecretSanta import SecretSanta
from cogs.SantaAdministrative import SantaAdministrative
from cogs.SantaMiscellaneous import SantaMiscellaneous
from cogs.SantaUtilities import SantaUtilities
from helpers.SQLiteHelper import SQLiteHelper
from helpers.SantaCountdownHelper import SantaCountdownHelper

def start_santa_bot():
    #initialize config file
    config = None
    try:
        config = ConfigObj(CONFIG.cfg_path, file_error = True)
    except Exception as e:
        try:
            os.mkdir(CONFIG.bot_folder)
        except Exception as e:
            pass
        config = ConfigObj()
        config.filename = CONFIG.cfg_path
        config['programData'] = {'exchange_started': False}
        config['members'] = {}
        config.write()

    # initialize the SQLite db
    sqlitehelper = SQLiteHelper(CONFIG.sqlite_path)
    sqlitehelper.create_connection()

    #set up discord connection debug logging
    discord_api_logger = logging.getLogger('discord')
    discord_api_logger.setLevel(logging.DEBUG)
    discord_api_handler = logging.FileHandler(filename=os.path.join(str(pathlib.Path(__file__).parent), CONFIG.bot_folder, "discord_api.log"), encoding='utf-8', mode='w')
    discord_api_handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s: %(message)s'))
    discord_api_logger.addHandler(discord_api_handler)

    # set up SantaBot debug logging
    santabot_logger = logging.getLogger('SantaBot')
    santabot_logger.setLevel(logging.DEBUG)
    santabot_log_handler = logging.FileHandler(filename=CONFIG.dbg_path, encoding='utf-8', mode='w')
    santabot_log_handler.setLevel(logging.DEBUG)
    santabot_log_handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s: %(message)s'))
    santabot_logger.addHandler(santabot_log_handler)

    # add the cogs
    intents = Intents.default()
    intents.members = True
    bot = commands.Bot(command_prefix = CONFIG.prefix, intents=intents)
    bot.add_cog(SantaAdministrative(bot))
    bot.add_cog(SantaMiscellaneous(bot))
    bot.add_cog(SantaUtilities(bot, sqlitehelper))
    bot.add_cog(SecretSanta(bot, config))

    # kick off the bot
    bot.run(CONFIG.discord_token, reconnect = True)

if __name__ == '__main__':
    start_santa_bot()
else:
    print("santa-bot.py does nothing unless it is run as the main program")
    quit()