3 Commits

Author SHA1 Message Date
f4e46c6854 added config file 2020-11-20 21:03:03 +01:00
60d6bd2c40 Started creating config file system 2020-11-20 21:01:41 +01:00
ae2e63da56 Started creating config file system 2020-11-20 21:00:56 +01:00
7 changed files with 91 additions and 34 deletions

32
bot.py
View File

@@ -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
View 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

View File

@@ -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
View 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

View File

@@ -21,6 +21,6 @@ modPlusRoles = [roleAdmin, roleMod, roleOwner]
# Channels
ModLogs = 760807882899193867
# dev
#QuizChannelID = 774418250665951232
QuizChannelID = 775776550871498752
QuizChannelID = 774418250665951232
#QuizChannelID = 775776550871498752
DiscordLinkerID = 776944220119760927

1
test.py Normal file
View File

@@ -0,0 +1 @@
from tests.quiz_test import *

17
tests/quiz_test.py Normal file
View 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]")