Skip to content

Commit a74c82b

Browse files
committed
Add main, including startup of logging and bot
1 parent f4dd234 commit a74c82b

File tree

1 file changed

+67
-0
lines changed

1 file changed

+67
-0
lines changed

src/main.py

Lines changed: 67 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,67 @@
1+
import os
2+
import logging
3+
4+
import discord
5+
from discord.ext import commands
6+
7+
intents = discord.Intents.all()
8+
9+
TOKEN = os.getenv("TOKEN") # reading in the token from config.py file
10+
11+
# path for databases or config files
12+
if not os.path.exists('data/'):
13+
os.mkdir('data/')
14+
15+
# set logging format
16+
formatter = logging.Formatter("[{asctime}] [{levelname}] [{name}] {message}", style="{")
17+
18+
# logger for writing to file
19+
file_logger = logging.FileHandler('data/events.log')
20+
file_logger.setLevel(logging.INFO) # everything into the logging file
21+
file_logger.setFormatter(formatter)
22+
23+
# logger for console prints
24+
console_logger = logging.StreamHandler()
25+
console_logger.setLevel(logging.WARNING) # only important stuff to the terminal
26+
console_logger.setFormatter(formatter)
27+
28+
# get new logger
29+
logger = logging.getLogger('my-bot')
30+
logger.setLevel(logging.INFO)
31+
32+
# register loggers
33+
logger.addHandler(file_logger)
34+
logger.addHandler(console_logger)
35+
36+
prefix = "b!"
37+
38+
bot = commands.Bot(command_prefix=prefix, intents=intents)
39+
40+
41+
# login message
42+
@bot.event
43+
async def on_ready():
44+
print(f'{bot.user.name} has connected')
45+
guild = discord.utils.get(bot.guilds) # , name=GUILD)
46+
47+
print(f'Bot is connected to the following guilds:')
48+
print()
49+
member_count = 0
50+
for g in bot.guilds:
51+
print(f"{g.name} - {g.id} - Members: {g.member_count}")
52+
member_count += g.member_count
53+
print()
54+
await bot.change_presence(
55+
activity=discord.Activity(type=discord.ActivityType.watching, name=f"{prefix}help"))
56+
57+
# LOADING Extensions
58+
# bot.remove_command('help')
59+
initial_extensions = [
60+
'cogs.misc',
61+
]
62+
63+
if __name__ == '__main__':
64+
for extension in initial_extensions:
65+
bot.load_extension(extension)
66+
67+
bot.run(TOKEN)

0 commit comments

Comments
 (0)