aboutsummaryrefslogtreecommitdiff
path: root/bot.py
diff options
context:
space:
mode:
Diffstat (limited to 'bot.py')
-rw-r--r--bot.py16
1 files changed, 10 insertions, 6 deletions
diff --git a/bot.py b/bot.py
index 6cc04d1..0be1c79 100644
--- a/bot.py
+++ b/bot.py
@@ -4,25 +4,29 @@ import random
from dotenv import load_dotenv
from discord.ext import commands
-from henti import Numbers as n
+from henti import Numbers
load_dotenv() # load .env file
TOKEN = os.getenv('DISCORD_TOKEN')
-bot = commands.Bot(command_prefix='z!') # initialise the bot
+
+PREFIXES = ['z!', '?'] # prefixes for the bot's commands
+bot = commands.Bot(command_prefix=PREFIXES) # initialise the bot with pre-defined prefixes
# get title of henti
-@bot.command(name='info', help='Get the title of doujin using the holy numbers')
+@bot.command(name='info', help='Get the info about a doujin using the holy numbers')
async def getInfo(ctx, sauce):
- r = n(sauce).title()
+ doujin = Numbers(sauce)
+ r = doujin.title()
await ctx.send(r)
# short for the same
-@bot.command(name='i', help='short fot z!title')
+@bot.command(name='i', help='short fot z!info')
async def getInfo(ctx, sauce):
- r = n(sauce).title()
+ doujin = Numbers(sauce)
+ r = doujin.info()
await ctx.send(r)