68 lines
1.8 KiB
Python
Executable File
68 lines
1.8 KiB
Python
Executable File
import discord
|
|
from discord.ext import commands
|
|
from database import save_result
|
|
|
|
class Scan(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
@commands.command()
|
|
async def scan(self, ctx, limit: int = 4500):
|
|
"""
|
|
Scan previous messages in the channel for Rule34dle posts
|
|
"""
|
|
|
|
found = 0
|
|
saved = 0
|
|
|
|
async for message in ctx.channel.history(limit=limit):
|
|
if message.author.bot:
|
|
continue
|
|
|
|
content = message.content.strip()
|
|
|
|
# Only process Rule34dle messages
|
|
if not content.startswith("Rule34dle"):
|
|
continue
|
|
|
|
found += 1
|
|
|
|
try:
|
|
lines = content.split("\n")
|
|
|
|
# Ensure correct format
|
|
if len(lines) < 3:
|
|
continue
|
|
|
|
# Extract date from first line
|
|
# Example: "Rule34dle Daily 2026-06-12"
|
|
date_line = lines[0]
|
|
date = date_line.split(" ")[2]
|
|
|
|
# Extract result
|
|
result = int(lines[1].split("/")[0])
|
|
|
|
# Extract squares
|
|
squares_line = lines[2].strip()
|
|
|
|
# Validate squares
|
|
if not all(c in ["🟩", "🟥"] for c in squares_line):
|
|
continue
|
|
|
|
# Convert to 0/1
|
|
score_values = [1 if c == "🟩" else 0 for c in squares_line]
|
|
|
|
user_id = message.author.id
|
|
|
|
save_result(user_id, result, score_values, date)
|
|
saved += 1
|
|
|
|
except Exception as e:
|
|
print("Scan parse error:", e)
|
|
|
|
await ctx.send(f"Scan complete!\nFound: {found}\nSaved: {saved}")
|
|
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(Scan(bot))
|