blob: 7da4f0732af5e46337d5dee7f9fbd7684533272f (
plain)
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
|
import discord
from discord.ext import commands, tasks
import os
import requests
import config
class MyBot(commands.Bot):
def __init__(self):
super().__init__(
command_prefix="***",
activity=discord.Game(name="music!"),
intents=discord.Intents.default(),
)
async def setup_hook(self):
get_access_token.start()
config.LOG.info("Loading cogs...")
for ext in os.listdir("./code/cogs"):
if ext.endswith(".py"):
if ext[:-3] == "feedback" and config.FEEDBACK_CHANNEL_ID == None:
config.LOG.info("Skipped loading feedback cog - channel ID not provided")
continue
if ext[:-3] == "bug" and config.BUG_CHANNEL_ID == None:
config.LOG.info("Skipped loading bug cog - channel ID not provided")
continue
await self.load_extension(f"cogs.{ext[:-3]}")
for ext in os.listdir("./code/cogs/owner"):
if ext.endswith(".py"):
await self.load_extension(f"cogs.owner.{ext[:-3]}")
bot = MyBot()
bot.remove_command("help")
bot.temp_command_count = {} # command_name: count
bot.autoplay = [] # guild_id, guild_id, etc.
@bot.event
async def on_ready():
config.LOG.info(f"{bot.user} has connected to Discord.")
config.LOG.info("Startup complete. Sync slash commands by DMing the bot ***sync")
@tasks.loop(minutes=45)
async def get_access_token():
auth_url = "https://accounts.spotify.com/api/token"
data = {
"grant_type": "client_credentials",
"client_id": config.SPOTIFY_CLIENT_ID,
"client_secret": config.SPOTIFY_CLIENT_SECRET,
}
response = requests.post(auth_url, data=data)
access_token = response.json()["access_token"]
bot.spotify_headers = {"Authorization": f"Bearer {access_token}"}
if __name__ == "__main__":
config_contents = config.load_config()
config.validate_config(config_contents)
bot.run(config.TOKEN)
|
