major 'performance tweaks' in henti.py

1 files changed, 22 insertions, 6 deletions

diff --git a/bot.py b/bot.py
index 3b1e971..6cc04d1 100644
--- a/bot.py
+++ b/bot.py
@@ -1,20 +1,36 @@
 import os
+import random
 
 from dotenv import load_dotenv
 from discord.ext import commands
 
-import henti as H
+from henti import Numbers as n
 
-load_dotenv()
+load_dotenv()  # load .env file
 TOKEN = os.getenv('DISCORD_TOKEN')
+bot = commands.Bot(command_prefix='z!')  # initialise the bot
+
+
 
-bot = commands.Bot(command_prefix='z!')
 
 # get title of henti
-@bot.command(name='title', help='Get the title of doujin using the holy numbers')
-async def nine_nine(ctx, sauce):
-    r =  H.get_title(sauce)
+@bot.command(name='info', help='Get the title of doujin using the holy numbers')
+async def getInfo(ctx, sauce):
+    r = n(sauce).title()
     await ctx.send(r)
 
+# short for the same
+@bot.command(name='i', help='short fot z!title')
+async def getInfo(ctx, sauce):
+    r = n(sauce).title()
+    await ctx.send(r)
+
+
+# @bot.command(name='', help='')
+# async def 
+
+
+
+
 bot.run(TOKEN)