Initial commit
This commit is contained in:
commit
a20e7fdbef
4
.gitignore
vendored
Normal file
4
.gitignore
vendored
Normal file
|
@ -0,0 +1,4 @@
|
|||
*.env
|
||||
/.venv
|
||||
member_data.db
|
||||
discord.log
|
3
availcommands.txt
Normal file
3
availcommands.txt
Normal file
|
@ -0,0 +1,3 @@
|
|||
/help : to view all commands
|
||||
/nick : to view current nickname
|
||||
/setnick : to change nickname
|
111
main.py
Normal file
111
main.py
Normal file
|
@ -0,0 +1,111 @@
|
|||
import discord
|
||||
from discord.ext import commands
|
||||
import sqlite3
|
||||
from dotenv import load_dotenv
|
||||
import os
|
||||
import logging
|
||||
from datetime import datetime
|
||||
|
||||
# Set up logging handler. Passed in at the end in bot.run
|
||||
handler = logging.FileHandler(filename='discord.log', encoding='utf-8', mode='w')
|
||||
|
||||
# Load ENV file
|
||||
load_dotenv()
|
||||
|
||||
# Gather variables from .env file
|
||||
BOT_TOKEN = os.getenv('BOT_TOKEN')
|
||||
CHANNEL_ID = int(os.getenv('BOT_CHANNEL_ID'))
|
||||
botCommandsEnable = True
|
||||
|
||||
if BOT_TOKEN is None:
|
||||
print("BOT TOKEN NOT FOUND IN ENV")
|
||||
exit()
|
||||
if CHANNEL_ID is None:
|
||||
print("NO CHANNEL SELECTED, DISABLING BOT COMMANDS")
|
||||
botCommandsEnable = False
|
||||
else:
|
||||
print(f'Channel ID selected: {CHANNEL_ID}')
|
||||
|
||||
|
||||
intents = discord.Intents.all()
|
||||
intents.members = True # Enable the member update intent
|
||||
intents.message_content = True # Enable the Privileged Content intent
|
||||
|
||||
# Set discord end bot commands to / prefix
|
||||
bot = commands.Bot(command_prefix='/', intents=intents)
|
||||
|
||||
# Connect to SQLite
|
||||
conn = sqlite3.connect('member_data.db')
|
||||
cursor = conn.cursor()
|
||||
|
||||
# Create Dataset if it doesn't exist already
|
||||
cursor.execute('''
|
||||
CREATE TABLE IF NOT EXISTS members (
|
||||
user_id INTEGER PRIMARY KEY,
|
||||
username TEXT,
|
||||
nickname TEXT,
|
||||
join_date TEXT
|
||||
)
|
||||
''')
|
||||
conn.commit()
|
||||
|
||||
|
||||
@bot.event
|
||||
async def on_ready():
|
||||
# When API connects and logs in.
|
||||
print(f'Bot logged in as {bot.user.name}')
|
||||
|
||||
# Send help command to bot-commands channel
|
||||
# channel = bot.get_channel(int(CHANNEL_ID))
|
||||
# await channel.send(f'Use /help to view available commands')
|
||||
|
||||
|
||||
@bot.event
|
||||
async def on_member_join(member):
|
||||
# Print user's name and ID in the console on new join
|
||||
print(f'New member joined: {member.name} (ID: {member.id}), joined at {member.joined_at}')
|
||||
print(member)
|
||||
|
||||
# Store member data in the database
|
||||
cursor.execute('INSERT INTO members (user_id, username, join_date) VALUES (?, ?, ?)',
|
||||
(member.id, member.name, datetime.now().strftime('%Y-%m-%d %H:%M:%S')))
|
||||
conn.commit()
|
||||
|
||||
|
||||
@bot.command()
|
||||
async def commands(ctx):
|
||||
avail_commands = open("availcommands.txt", "r")
|
||||
await ctx.send(avail_commands.read())
|
||||
avail_commands.close()
|
||||
|
||||
|
||||
@bot.command()
|
||||
async def setnick(ctx, new_nickname: str):
|
||||
|
||||
# Update user's nickname in the database
|
||||
cursor.execute('UPDATE members SET nickname = ? WHERE user_id = ?', (new_nickname, ctx.author.id))
|
||||
conn.commit()
|
||||
|
||||
# Retrieve the updated nickname from the database
|
||||
cursor.execute('SELECT nickname FROM members WHERE user_id = ?', (ctx.author.id,))
|
||||
updated_nickname = cursor.fetchone()[0]
|
||||
|
||||
# Print the updated nickname to the console
|
||||
print(f'Nickname updated for {ctx.author.name} (ID: {ctx.author.id}): {updated_nickname}')
|
||||
|
||||
member = ctx.guild.get_member(ctx.author.id)
|
||||
if member:
|
||||
try:
|
||||
await member.edit(nick=updated_nickname)
|
||||
print(f'Nickname updated on the server for {ctx.author.name}')
|
||||
except discord.Forbidden:
|
||||
print(f'Failed to update nickname on the server for {ctx.author.name}: Missing permissions')
|
||||
except discord.HTTPException as e:
|
||||
print(f'Failed to update nickname on the server for {ctx.author.name}: {e}')
|
||||
|
||||
# Close the database connection when the bot is stopped
|
||||
@bot.event
|
||||
async def on_disconnect():
|
||||
conn.close()
|
||||
|
||||
bot.run(BOT_TOKEN, log_handler=handler, log_level=logging.DEBUG)
|
Loading…
Reference in New Issue
Block a user