Compare commits
3 Commits
release_br
...
tests
| Author | SHA1 | Date | |
|---|---|---|---|
| f4e46c6854 | |||
| 60d6bd2c40 | |||
| ae2e63da56 |
32
bot.py
32
bot.py
@@ -1,27 +1,11 @@
|
||||
#!/usr/bin/python
|
||||
|
||||
import sys
|
||||
from discord.ext import commands
|
||||
import os
|
||||
import yaml
|
||||
from bots.discordbot import DiscordBot
|
||||
|
||||
class DiscordBot:
|
||||
def __init__(self, config):
|
||||
|
||||
prefixes = ["/"]
|
||||
|
||||
self.client = commands.Bot(command_prefix=prefixes, case_insensitive=True)
|
||||
self.client.prefixes = prefixes
|
||||
|
||||
# Load this cog first so the other cogs can use properties initialized here
|
||||
self.client.load_extension("cogs.events")
|
||||
|
||||
for file in os.listdir("./cogs"):
|
||||
if file.endswith(".py") and not (file.startswith(("events",))):
|
||||
self.client.load_extension(f"cogs.{file[:-3]}")
|
||||
|
||||
def run(self):
|
||||
self.client.run("NzYwNTI5NDY2MDI3MDE2MjUy.X3NYQg.Nz3bwxgltlayMStfT7F-OCbx9pE")
|
||||
|
||||
def get_client(self):
|
||||
return self.client
|
||||
|
||||
|
||||
bot = DiscordBot(None)
|
||||
bot.run()
|
||||
if __name__ == "__main__":
|
||||
bot = DiscordBot(sys.argv[1])
|
||||
bot.run()
|
||||
|
||||
45
bots/discordbot.py
Normal file
45
bots/discordbot.py
Normal file
@@ -0,0 +1,45 @@
|
||||
import sys
|
||||
from discord.ext import commands
|
||||
import os
|
||||
import yaml
|
||||
|
||||
class DiscordBot:
|
||||
def __init__(self, config):
|
||||
print("jajajah")
|
||||
self.constants = ["token"]
|
||||
self.modules = {"quiz": "quiz"}
|
||||
prefixes = ["/"]
|
||||
|
||||
self.client = commands.Bot(command_prefix=prefixes, case_insensitive=True)
|
||||
self.client.prefixes = prefixes
|
||||
self.load_config(config)
|
||||
# Load this cog first so the other cogs can use properties initialized here
|
||||
self.client.load_extension("cogs.events")
|
||||
|
||||
# for file in os.listdir("./cogs"):
|
||||
# if file.endswith(".py") and not (file.startswith(("events",))):
|
||||
# self.client.load_extension(f"cogs.{file[:-3]}")
|
||||
|
||||
def run(self):
|
||||
if "token" in self.client.config.keys():
|
||||
print(self.client.config["token"])
|
||||
self.client.run(self.client.config["token"])
|
||||
|
||||
def get_client(self):
|
||||
return self.client
|
||||
|
||||
def load_config(self, config):
|
||||
with open(config, 'r') as file:
|
||||
try:
|
||||
self.client.config = yaml.safe_load(file)
|
||||
for cog in self.client.config:
|
||||
if cog not in self.constants and self.loaded_cog(cog):
|
||||
print(f"Loaded cog {cog}")
|
||||
except yaml.YAMLError as exc:
|
||||
print(exc)
|
||||
|
||||
def loaded_cog(self, module):
|
||||
if os.path.isfile(f"/cogs/{self.modules[module]}.py"):
|
||||
self.client.load_extension(f"cogs.{self.modules[module]}")
|
||||
return True
|
||||
return False
|
||||
14
cogs/quiz.py
14
cogs/quiz.py
@@ -1,6 +1,5 @@
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
from data import constants
|
||||
import asyncio
|
||||
import threading
|
||||
import random
|
||||
@@ -143,11 +142,10 @@ class QuizQuestions(commands.Cog):
|
||||
class Quiz(commands.Cog):
|
||||
def __init__(self, client):
|
||||
self.client = client
|
||||
#self.interval = (1, 4)
|
||||
self.interval = (5*60, 20*60)
|
||||
self.auto = False
|
||||
self.timeout = 10*60
|
||||
|
||||
self.quiz_config = self.client.config["quiz"]
|
||||
self.interval = (int(self.quiz_config["interval_start"]) * 60, int(self.quiz_config["interval_end"]) * 60)
|
||||
self.timeout = int(self.quiz_config["timeout"])*60
|
||||
|
||||
@commands.group(name="quiz",case_insensitive=True, invoke_without_command=True)
|
||||
@commands.check(checks.isModPlus)
|
||||
@@ -218,7 +216,7 @@ class Quiz(commands.Cog):
|
||||
ranges = {10: 50, 25: 75, 50: 100, 75: 125, 100: 150}
|
||||
range = random.choice(list(ranges))
|
||||
random_number = random.randint(1, range)
|
||||
channel = self.client.get_channel(constants.QuizChannelID)
|
||||
channel = self.client.get_channel(self.quiz_config["channel"])
|
||||
await channel.send(embed=self.create_question_embed(f"Guess the number from 1 to {range}", ranges[range]))
|
||||
await channel.edit(slowmode_delay=1)
|
||||
try:
|
||||
@@ -238,7 +236,7 @@ class Quiz(commands.Cog):
|
||||
async def ask_question(self):
|
||||
question = self.get_random_question()
|
||||
self.increment_asked_count(question[0])
|
||||
channel = self.client.get_channel(constants.QuizChannelID)
|
||||
channel = self.client.get_channel(self.quiz_config["channel"])
|
||||
await channel.send(embed=self.create_question_embed(question[1], question[3]))
|
||||
try:
|
||||
message = await self.client.wait_for("message", check=lambda message: message.content.lower() == question[2].lower(), timeout=self.timeout)
|
||||
@@ -255,7 +253,7 @@ class Quiz(commands.Cog):
|
||||
async def start_word_game(self):
|
||||
word = self.get_random_word()
|
||||
self.increment_asked_word(int(word["id"]))
|
||||
channel = self.client.get_channel(constants.QuizChannelID)
|
||||
channel = self.client.get_channel(self.quiz_config["channel"])
|
||||
shaken_word = self.shake_word(word["word"].lower())
|
||||
await channel.send(embed=self.create_question_embed(f"Find the word from: {shaken_word}", word['reward']))
|
||||
try:
|
||||
|
||||
12
config.yaml
Normal file
12
config.yaml
Normal file
@@ -0,0 +1,12 @@
|
||||
quiz:
|
||||
channel: 779084465426399272
|
||||
# question interval between questions in minutes
|
||||
interval_start: 1
|
||||
interval_end: 1
|
||||
# timeout for answer in minutes
|
||||
timeout: 10
|
||||
streaks: False
|
||||
|
||||
# test channel
|
||||
token: "Nzc4NjQ0MjMxNDI1MDk3NzQx.X7U-8w.TOqNKPWQkNSIRByTRcgkjaZqbUE"
|
||||
#token: NzYwNTI5NDY2MDI3MDE2MjUy.X3NYQg.Nz3bwxgltlayMStfT7F-OCbx9pE
|
||||
@@ -21,6 +21,6 @@ modPlusRoles = [roleAdmin, roleMod, roleOwner]
|
||||
# Channels
|
||||
ModLogs = 760807882899193867
|
||||
# dev
|
||||
#QuizChannelID = 774418250665951232
|
||||
QuizChannelID = 775776550871498752
|
||||
QuizChannelID = 774418250665951232
|
||||
#QuizChannelID = 775776550871498752
|
||||
DiscordLinkerID = 776944220119760927
|
||||
|
||||
17
tests/quiz_test.py
Normal file
17
tests/quiz_test.py
Normal file
@@ -0,0 +1,17 @@
|
||||
from bots.discordbot import DiscordBot
|
||||
import discord.ext.test as dpytest
|
||||
import pytest
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_bot():
|
||||
bot = DiscordBot("config.yaml")
|
||||
|
||||
# Load any extensions/cogs you want to in here
|
||||
dpytest.configure(bot.get_client())
|
||||
config = dpytest.get_config()
|
||||
channel = config.channels[0]
|
||||
user = dpytest.backend.make_user("test", 1, 1)
|
||||
member = dpytest.backend.make_member(user, config.guilds[0])
|
||||
guild = config.guilds[0]
|
||||
await dpytest.message("hallo", channel, member)
|
||||
dpytest.verify_message("[Expected help output]")
|
||||
Reference in New Issue
Block a user