first commit

This commit is contained in:
ObeseTermite 2025-01-13 07:04:33 -08:00
commit 8209208439
3 changed files with 171 additions and 0 deletions

1
.gitignore vendored Normal file
View file

@ -0,0 +1 @@
.env

166
bot.py Normal file
View file

@ -0,0 +1,166 @@
# bot.py
import os
import discord
import re
from collections import Counter
from discord.ext import commands
from dotenv import load_dotenv
WORDLE_CHANNEL = 1317916234863480832
load_dotenv()
TOKEN = os.getenv('DISCORD_TOKEN')
bot = commands.Bot()
@bot.slash_command(
name="stats",
description="Prints out your stats in a daily game, examples include wordle",
guild_ids=[261345598987436033]
)
async def stats(interaction: discord.Interaction, game, user: discord.User = None):
if game.lower() == 'wordle':
channel = bot.get_channel(WORDLE_CHANNEL)
if user is None:
user = interaction.user
WORDLE_PATTERN = r"^Wordle (\d{1,3}(?:,\d{3})*) (X|\d+)/(\d+)"
total_guesses = 0
total_games = 0
wins = 0
async for message in channel.history(limit=1000): # Limit can be adjusted
if message.author == user:
for line in message.content.splitlines():
match = re.match(WORDLE_PATTERN, line.strip())
if match:
total_games += 1
guesses = match.group(2)
if guesses != 'X':
total_guesses += int(guesses)
wins += 1
break
if total_games == 0:
await interaction.response.send_message(
"No Wordle results found.",
ephemeral=True
)
return
average_guesses = total_guesses / wins
win_rate = (wins / total_games) * 100
await interaction.response.send_message(
f"Wordle stats for {user.mention}:\n"
f"Total Games Played: {total_games}\n"
f"Total Guesses: {total_guesses}\n"
f"Average Guesses per Winning Game: {average_guesses:.2f}\n"
f"Win Rate: {win_rate:.2f}%\n",
ephemeral=False # Send message only to the user who called the command
)
elif game.lower() == 'connections':
channel = bot.get_channel(WORDLE_CHANNEL)
if user is None:
user = interaction.user
CONNECTIONS_PATTERN = r"^Connections\nPuzzle #(\d+)\n((?:[\u2B1B\u2B1C\u2B1D\u2B1E\u2B20]{4}\n?)+)$"
total_guesses = 0
total_games = 0
wins = 0
perfects = 0
async for message in channel.history(limit=1000): # Limit can be adjusted
if message.author == user:
message_content = message.content.strip()
if message_content.startswith("Connections") and "Puzzle #" in message_content:
# Extract the puzzle number
puzzle_number_start = message_content.find("Puzzle #") + len("Puzzle #")
puzzle_number_end = message_content.find("\n", puzzle_number_start)
puzzle_number = message_content[puzzle_number_start:puzzle_number_end].strip()
# Extract the grid (assume the grid starts after the first two lines)
grid_start = message_content.find("\n", message_content.find("Puzzle #")) + 1
grid_content = message_content[grid_start:].strip()
# Split the grid into lines
color_grid = grid_content.split("\n")
guesses = check_connections_win(color_grid)
total_games += 1
if guesses > 0:
total_guesses += guesses
wins += 1
perfect_game = is_perfect_game(color_grid)
if perfect_game:
perfects += 1
if total_games == 0:
await interaction.response.send_message(
"No Connections results found.",
ephemeral=True
)
return
average_guesses = total_guesses / wins
win_rate = (wins / total_games) * 100
await interaction.response.send_message(
f"Connections stats for {user.mention}:\n"
f"Total Games Played: {total_games}\n"
f"Average Guesses per Winning Game: {average_guesses:.2f}\n"
f"Win Rate: {win_rate:.2f}%\n"
f"Perfect Games: {perfects}\n",
ephemeral=False # Send message only to the user who called the command
)
else:
await interaction.response.send_message(
"Not a game.",
ephemeral=True
)
def check_connections_win(color_grid):
"""
Check if the Connections game is a win and return the number of guesses.
A win is when there are exactly 4 groups of 4 colors each.
"""
# Flatten the color grid into a list of colors
uniform_rows = [row for row in color_grid if len(set(row)) == 1]
flattened_grid = ''.join(uniform_rows)
# Count the occurrences of each color in the grid
color_counts = Counter(flattened_grid)
# Check if there are exactly 4 different colors with 4 squares each
if len(color_counts) == 4 and all(count == 4 for count in color_counts.values()):
# A win: return the number of guesses (lines in the grid)
return len(color_grid)
return 0 # Not a win
def is_perfect_game(color_grid):
"""
Check if the Connections game is a perfect game (solved in 4 guesses, no mistakes).
"""
# A perfect game is one where the first 4 lines form perfect groups
if len(color_grid) == 4 and all(len(set(line)) == 1 for line in color_grid):
return True # It's a perfect game
return False
@bot.event
async def on_ready():
print('Ready!')
bot.run(TOKEN)

4
config.json Normal file
View file

@ -0,0 +1,4 @@
{
"token": "MTMyODA2NzMwNTU5NDQyNTUyNg.GwMJI7.JcNhYkxT_D0WS9hm46MVnSqZu9Oz1DllEqBP-4",
"clientId": "1328067305594425526"
}