PlopBot/BotHead.py at main · plop91/PlopBot · 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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
#!/usr/bin/env python3
"""
PlopBot is a discord bot that can play sounds and has some additional features.
This is the main file for the bot.
"""
import asyncio
import os
import discord
from discord.ext import commands
import argparse
import settings


class PlopBot(commands.Bot):
    """
    The main bot class.
    """
    def __init__(self, *args, **kwargs):
        """
        Constructor for the bot.
        :arg args:
        :arg kwargs:
        :return: None
        """
        super().__init__(*args, **kwargs)

    async def setup_hook(self) -> None:
        """
        Setup hook for the bot, loads all cogs.
        :return: None
        """
        for f in os.listdir('./cogs'):
            if f.endswith('.py'):
                await self.load_extension(f'cogs.{f[:-3]}')

    # @commands.command(brief="Admin only command: Load a Cog.")
    # async def load(self, ctx, extension):
    #     """Loads a new cog into the bot while running"""
    #     settings.logger.info(f"load: {extension} :: from {ctx.author}")
    #     if ctx.author not in settings.info_json["admins"]:
    #         settings.logger.warning(f"User: {ctx.author} is not an admin but tried to load a cog!")
    #     elif str(ctx.message.channel) not in settings.info_json["command_channels"]:
    #         settings.logger.info(f"User: {ctx.author} is an admin but tried to load in a non command channel!")
    #     else:
    #         await self.load_extension(f'cogs.{extension}Cog')
    #         settings.logger.info(f"cog:{extension} loaded")
    #         await ctx.message.delete()
    #
    # @commands.command(brief="Admin only command: Unload a Cog.")
    # async def unload(self, ctx, extension):
    #     """Unloads an active cog from the bot"""
    #     settings.logger.info(f"unload: {extension} :: from {ctx.author}")
    #     if ctx.author not in settings.info_json["admins"]:
    #         settings.logger.warning(f"User: {ctx.author} is not an admin but tried to unload a cog!")
    #     elif str(ctx.message.channel) not in settings.info_json["command_channels"]:
    #         settings.logger.info(f"User: {ctx.author} is an admin but tried to unload in a non command channel!")
    #     else:
    #         await self.unload_extension(f'cogs.{extension}Cog')
    #         settings.logger.info(f"cog:{extension} unloaded")
    #     await ctx.message.delete()
    #
    # @commands.command(brief="Admin only command: Reload a Cog.")
    # async def reload(self, ctx, extension):
    #     """Unloads then reloads a cog"""
    #     settings.logger.info(f"reload: {extension} :: from {ctx.author}")
    #     if ctx.author not in settings.info_json["admins"]:
    #         settings.logger.warning(f"User: {ctx.author} is not an admin but tried to reload a cog!")
    #     elif str(ctx.message.channel) not in settings.info_json["command_channels"]:
    #         settings.logger.info(f"User: {ctx.author} is an admin but tried to reload in a non command channel!")
    #     else:
    #         await self.reload_extension(f'cogs.{extension}Cog')
    #         settings.logger.info(f"reload from {ctx.author}")
    #     await ctx.message.delete()
    #
    # @commands.event
    # async def on_ready(self):
    #     """Logs that the cog was loaded properly"""
    #     settings.logger.info(f"Logged on as {self.user}!")
    #     # await load_cogs()
    #     await self.change_presence(status=discord.Status.online, activity=discord.Game('listening'))
    #
    # @commands.event
    # async def on_disconnect(self):
    #     """Logs the bot turning off"""
    #     settings.logger.info(f"Logged off!")
    #     await self.change_presence(status=discord.Status.offline, activity=discord.Game('sleeping'))


async def main(args):
    """
    Main entry point for the bot.
    :arg args: arguments from argparse
    :return: None
    """
    settings.init(args)

    async with PlopBot(command_prefix='.', intents=discord.Intents.all()) as bot:
        await bot.start(settings.token)


if __name__ == '__main__':
    parser = argparse.ArgumentParser(prog='Discord Bot',
                                     description='a discord bot with a soundboard and some additional features.')

    parser.add_argument('--json', help='Path to json file', default='info/info.json')
    parser.add_argument('--db_host', help='Database hostname or ip by default taken from json file', default=None)
    parser.add_argument('--db_username', help='Database username by default taken from json file', default=None)
    parser.add_argument('--db_password', help='Database password by default taken from json file', default=None)
    parser.add_argument('--db_name', help='Database name by default taken from json file', default=None)

    arguments = parser.parse_args()

    # Initialize settings to connect to database, open json and setup logging

    asyncio.run(main(arguments))