done
This commit is contained in:
@@ -0,0 +1,73 @@
|
||||
|
||||
import discord
|
||||
from discord.ext import commands, tasks
|
||||
from discord import app_commands
|
||||
from datetime import datetime, timezone
|
||||
from base64 import b64decode
|
||||
import requests
|
||||
from random import shuffle
|
||||
from os import environ as env
|
||||
|
||||
# Channel ids
|
||||
CUSTOM_CHANNELS = set([
|
||||
1352404223127588924
|
||||
])
|
||||
|
||||
TOKEN = env['TOKEN']
|
||||
HOURS = 10
|
||||
ROLE = env['ROLE']
|
||||
CHANNELS = set([])
|
||||
|
||||
def decode(value):
|
||||
return b64decode(value).decode('utf-8')
|
||||
|
||||
OPENTDB_URL = 'https://opentdb.com/api.php?amount=1&category=28&encode=base64'
|
||||
|
||||
intents = discord.Intents.default()
|
||||
intents.message_content = True
|
||||
intents.guilds = True
|
||||
intents.messages = True
|
||||
bot = commands.Bot(command_prefix="!", intents=intents)
|
||||
|
||||
@bot.event
|
||||
async def on_ready():
|
||||
print(f"Logged in as {bot.user}")
|
||||
for cc in CUSTOM_CHANNELS:
|
||||
print(f'Custom channel: {cc}')
|
||||
CHANNELS.add(bot.get_channel(cc))
|
||||
check_inactivity.start()
|
||||
|
||||
@tasks.loop(seconds=HOURS)
|
||||
async def check_inactivity():
|
||||
if ROLE is not None:
|
||||
current_question = None
|
||||
for channel in CHANNELS:
|
||||
async for message in channel.history(limit=1):
|
||||
if (datetime.now(timezone.utc) - message.created_at).total_seconds() > HOURS:
|
||||
if current_question is None:
|
||||
current_question = requests.get(OPENTDB_URL).json()['results'][0]
|
||||
|
||||
decoded_data = {k: decode(v) if isinstance(v, str) else [decode(i) for i in v] for k, v in current_question.items()}
|
||||
|
||||
answers: list[str] = list(decoded_data['incorrect_answers'] + [decoded_data['correct_answer']])
|
||||
|
||||
shuffle(answers)
|
||||
|
||||
role_mention = f"<@&{ROLE}>"
|
||||
embedVar = discord.Embed(title=decoded_data['question'], description="", color=0x00ff00)
|
||||
embedVar.add_field(name="Difficulty", value=decoded_data['difficulty'].capitalize(), inline=False)
|
||||
embedVar.add_field(name="Answers", value='**'+('\n'.join(['- '+a for a in answers]))+'**', inline=False)
|
||||
msg = await channel.send(role_mention, embed=embedVar)
|
||||
|
||||
def check(m):
|
||||
return m.reference and m.reference.message_id == msg.id and m.author != bot.user
|
||||
|
||||
for _ in range(0, 10):
|
||||
reply = await bot.wait_for("message", check=check)
|
||||
|
||||
if reply.content.lower() == decoded_data['correct_answer'].lower():
|
||||
await reply.add_reaction("✅")
|
||||
else:
|
||||
await reply.add_reaction("❌")
|
||||
|
||||
bot.run(TOKEN)
|
||||
Reference in New Issue
Block a user