advtech/selena
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

FEAT: Started adding profiles

Browse Source 
FIX: Knucklebones now works as best as it can (need to stress test it)
This commit is contained in:
Dan 2024-07-04 13:55:30 -04:00
parent 2b621ae030
commit c9255143cb
4 changed files with 170 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
config.py
View File

@ -24,6 +24,7 @@ config = {
'update': {'enabled': True},
'data_privacy': {'enabled': True},
'terms_privacy': {'enabled': True},
'knucklebones': {'enabled': True}
'knucklebones': {'enabled': True},
'profiles': {'enabled': True}
}
}

6
main.py
View File

@ -95,6 +95,12 @@ class Selena(discord.Client):
knucklebones_setup(self)
logging.info("Knucklebones module loaded")
if config['modules']['profiles']['enabled']:
from modules.user.profiles import Profiles
profiles = Profiles(self)
profiles.setup(self.tree)
logging.info("Profiles module loaded")
bot = Selena()

25
modules/games/knucklebones.py
View File

@ -146,16 +146,27 @@ class Knucklebones:
column = random.randint(1, 3)
game.place_dice(self.bot.user, dice, column)
if game.is_game_over():
await self.end_game(channel, game)
else:
game.next_turn()
await self.update_game_message(game, channel, f"{self.bot.user.mention} rolled a {dice} and placed it in column {column}.\nIt's now {game.current_player().mention}'s turn!\n{game.render_board()}", view=RollDiceView(self.bot))
async def end_game(self, channel, game):
winner = game.winner()
if winner:
await self.award_kibble(winner.id, game.bet * 2)
await self.bot.profiles.record_win(winner.id, "Knucklebones")
loser = game.other_player()
await self.bot.profiles.record_loss(loser.id, "Knucklebones")
await self.update_game_message(game, channel, f"{winner.mention} wins the game and {game.bet * 2} Kibble!\n{game.render_board()}")
else:
await self.update_game_message(game, channel, f"The game is a tie!\n{game.render_board()}")
del self.games[channel.id]
else:
game.next_turn()
await self.update_game_message(game, channel, f"{self.bot.user.mention} rolled a {dice} and placed it in column {column}.\nIt's now {game.current_player().mention}'s turn!\n{game.render_board()}", view=RollDiceView(self.bot))
await self.schedule_thread_deletion(channel)
async def schedule_thread_deletion(self, channel):
await discord.utils.sleep_until(discord.utils.utcnow() + discord.timedelta(minutes=2))
await channel.delete()
async def has_enough_kibble(self, user_id, amount):
conn = sqlite3.connect(self.db_path)
@ -226,13 +237,7 @@ class PlaceDiceView(discord.ui.View):
return
game.place_dice(interaction.user, self.dice, column)
if game.is_game_over():
winner = game.winner()
if winner:
await self.bot.knucklebones_module.award_kibble(winner.id, game.bet * 2)
await self.bot.knucklebones_module.update_game_message(game, interaction, f"{winner.mention} wins the game and {game.bet * 2} Kibble!\n{game.render_board()}")
else:
await self.bot.knucklebones_module.update_game_message(game, interaction, f"The game is a tie!\n{game.render_board()}")
del self.bot.knucklebones_module.games[interaction.channel_id]
await self.bot.knucklebones_module.end_game(interaction.channel, game)
else:
game.next_turn()
await self.bot.knucklebones_module.update_game_message(game, interaction, f"{interaction.user.mention} placed {self.dice} in column {column}.\nIt's now {game.current_player().mention}'s turn!\n{game.render_board()}", view=RollDiceView(self.bot))

143
modules/user/profiles.py Normal file
View File

@ -0,0 +1,143 @@
import discord
import sqlite3
import logging
class Profiles:
def __init__(self, bot):
self.bot = bot
self.db_path = 'data/selena.db'
self.logger = logging.getLogger('Profiles')
self.logger.setLevel(logging.DEBUG)
handler = logging.FileHandler(filename='log/selena.log', encoding='utf-8', mode='w')
handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s:%(message)s'))
self.logger.addHandler(handler)
self.ensure_tables_exist()
def ensure_tables_exist(self):
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute("""
CREATE TABLE IF NOT EXISTS profiles (
user_id TEXT,
guild_id TEXT,
pronouns TEXT,
age INTEGER,
xp INTEGER DEFAULT 0,
level INTEGER DEFAULT 1,
is_global BOOLEAN DEFAULT 0,
PRIMARY KEY (user_id, guild_id)
);
""")
cursor.execute("""
CREATE TABLE IF NOT EXISTS game_stats (
user_id TEXT NOT NULL,
guild_id TEXT NOT NULL,
game TEXT NOT NULL,
wins INTEGER DEFAULT 0,
losses INTEGER DEFAULT 0,
PRIMARY KEY (user_id, guild_id, game)
);
""")
conn.commit()
conn.close()
self.logger.info('Profile and game stats tables ensured in database')
async def record_win(self, user_id, guild_id, game):
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute("""
INSERT INTO game_stats (user_id, guild_id, game, wins)
VALUES (?, ?, ?, 1)
ON CONFLICT(user_id, guild_id, game)
DO UPDATE SET wins = wins + 1
""", (user_id, guild_id, game))
conn.commit()
conn.close()
self.logger.info(f'Recorded win for user {user_id} in game {game} in guild {guild_id}')
async def record_loss(self, user_id, guild_id, game):
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute("""
INSERT INTO game_stats (user_id, guild_id, game, losses)
VALUES (?, ?, ?, 1)
ON CONFLICT(user_id, guild_id, game)
DO UPDATE SET losses = losses + 1
""", (user_id, guild_id, game))
conn.commit()
conn.close()
self.logger.info(f'Recorded loss for user {user_id} in game {game} in guild {guild_id}')
async def update_profile(self, user_id, guild_id, pronouns=None, age=None, is_global=False):
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute("""
INSERT INTO profiles (user_id, guild_id, pronouns, age, is_global)
VALUES (?, ?, ?, ?, ?)
ON CONFLICT(user_id, guild_id)
DO UPDATE SET pronouns = COALESCE(?, pronouns), age = COALESCE(?, age), is_global = ?
""", (user_id, guild_id, pronouns, age, is_global, pronouns, age, is_global))
conn.commit()
conn.close()
self.logger.info(f'Updated profile for user {user_id} in guild {guild_id}')
async def get_profile(self, user_id, guild_id):
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute("""
SELECT pronouns, age, xp, level, is_global FROM profiles
WHERE user_id = ? AND (guild_id = ? OR is_global = 1)
""", (user_id, guild_id))
row = cursor.fetchone()
conn.close()
return row
async def get_game_stats(self, user_id, guild_id):
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute("SELECT game, wins, losses FROM game_stats WHERE user_id = ? AND guild_id = ?", (user_id, guild_id))
rows = cursor.fetchall()
conn.close()
return rows
def setup(self, tree: discord.app_commands.CommandTree):
@tree.command(name="set_profile", description="Set your profile information")
async def set_profile_command(interaction: discord.Interaction, pronouns: str = None, age: int = None, is_global: bool = False):
user_id = str(interaction.user.id)
guild_id = str(interaction.guild.id)
await self.update_profile(user_id, guild_id, pronouns, age, is_global)
await interaction.response.send_message("Your profile has been updated.", ephemeral=True)
@tree.command(name="profile", description="View your profile")
async def profile_command(interaction: discord.Interaction):
user_id = str(interaction.user.id)
guild_id = str(interaction.guild.id)
profile = await self.get_profile(user_id, guild_id)
if profile:
pronouns, age, xp, level, is_global = profile
game_stats = await self.get_game_stats(user_id, guild_id)
games_info = "\n".join([f"{game}: {wins}W/{losses}L" for game, wins, losses in game_stats])
embed = discord.Embed(title=f"{interaction.user.display_name}'s Profile")
embed.add_field(name="Pronouns", value=pronouns or "Not set", inline=True)
embed.add_field(name="Age", value=age or "Not set", inline=True)
embed.add_field(name="XP", value=xp, inline=True)
embed.add_field(name="Level", value=level, inline=True)
embed.add_field(name="Global Profile", value="Yes" if is_global else "No", inline=True)
embed.add_field(name="Games", value=games_info or "No games played", inline=False)
await interaction.response.send_message(embed=embed)
else:
await interaction.response.send_message("Profile not found. Please set your profile using /set_profile.", ephemeral=True)
if not tree.get_command("set_profile"):
tree.add_command(set_profile_command)
if not tree.get_command("profile"):
tree.add_command(profile_command)
def setup(bot):
profiles = Profiles(bot)
profiles.setup(bot.tree)
bot.profiles = profiles