Kanji now uses interactive
This commit is contained in:
Robert
parent
65fe559ece
commit
387fd75b38
161
cogs/kanji.py
161
cogs/kanji.py
|
|
@ -1,90 +1,85 @@
|
|||
import discord
|
||||
from discord.ext import commands
|
||||
from utils import jisho
|
||||
from utils import jisho, interactive
|
||||
|
||||
class JishoKanjiObject():
|
||||
def __init__(self, query, owner):
|
||||
self.response = jisho.JishoKanji(query)
|
||||
self.total_pages = self.response.entries
|
||||
self.page = 0
|
||||
self.owner = owner
|
||||
class KanjiEmbed(interactive.InteractiveEmbed):
|
||||
REACTIONS = {
|
||||
"prev_kanji": "⬅️",
|
||||
"next_kanji": "➡️"
|
||||
}
|
||||
|
||||
def prev(self):
|
||||
self.page -= 1
|
||||
if self.page < 0:
|
||||
self.page = self.total_pages - 1
|
||||
def __init__(self, parent, ctx, response):
|
||||
super(KanjiEmbed, self).__init__(parent.bot, ctx, 60.0)
|
||||
self.parent = parent
|
||||
self.owner = self.ctx.author
|
||||
self.response = response
|
||||
|
||||
def next(self):
|
||||
self.page += 1
|
||||
if self.page >= self.total_pages:
|
||||
self.page = 0
|
||||
self.current_kanji = 0
|
||||
|
||||
def prev_kanji(self):
|
||||
self.current_kanji -= 1
|
||||
if self.current_kanji < 0:
|
||||
self.current_kanji = self.response.entries - 1
|
||||
|
||||
def next_kanji(self):
|
||||
self.current_kanji += 1
|
||||
if self.current_kanji >= self.response.entries:
|
||||
self.current_kanji = 0
|
||||
|
||||
async def on_reaction(self, reaction, user):
|
||||
if reaction.emoji == KanjiEmbed.REACTIONS["prev_kanji"]:
|
||||
self.prev_kanji()
|
||||
await reaction.remove(user)
|
||||
|
||||
if reaction.emoji == KanjiEmbed.REACTIONS["next_kanji"]:
|
||||
self.next_kanji()
|
||||
await reaction.remove(user)
|
||||
|
||||
async def add_navigation(self, message):
|
||||
if self.response.entries > 1:
|
||||
await message.add_reaction(KanjiEmbed.REACTIONS["prev_kanji"])
|
||||
await message.add_reaction(KanjiEmbed.REACTIONS["next_kanji"])
|
||||
|
||||
def make_embed(self):
|
||||
node = self.response.nodes[self.current_kanji]
|
||||
|
||||
embed = discord.Embed(
|
||||
title = node.meaning,
|
||||
url = node.url,
|
||||
description = f"{node.strokes} strokes",
|
||||
colour = 0x56d926
|
||||
)
|
||||
|
||||
if node.kun:
|
||||
embed.add_field(name="Kun", value="、 ".join(node.kun), inline=False)
|
||||
if node.on:
|
||||
embed.add_field(name="On", value="、 ".join(node.on), inline=False)
|
||||
|
||||
embed.add_field(name=f"Radical: {node.radical[0]}", value=node.radical[1], inline=False)
|
||||
embed.set_thumbnail(url=node.image_url)
|
||||
embed.set_footer(text=f"Jōyō kanji (Grade {node.grade}) | JLPT level {node.jlpt}\t\t{self.current_kanji + 1}/{self.response.entries}")
|
||||
|
||||
return embed
|
||||
|
||||
async def on_close(self):
|
||||
self.parent.activeObjects.pop(self.ctx.channel.id)
|
||||
|
||||
|
||||
class Kanji(commands.Cog):
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
self.activeObject = None
|
||||
self.latestMessage = 0
|
||||
|
||||
async def createEmbed(self):
|
||||
if self.activeObject.total_pages == 0:
|
||||
embed = discord.Embed(
|
||||
title = "No search results",
|
||||
description = "The search returned nothing. Did you make a typo?",
|
||||
colour = 0x56d926
|
||||
)
|
||||
else:
|
||||
response = self.activeObject.response
|
||||
node = response.nodes[self.activeObject.page]
|
||||
|
||||
embed = discord.Embed(
|
||||
title = node.meaning,
|
||||
url = node.url,
|
||||
description = f"{node.strokes} strokes",
|
||||
colour = 0x56d926
|
||||
)
|
||||
|
||||
if node.kun:
|
||||
embed.add_field(name="Kun", value="、 ".join(node.kun), inline=False)
|
||||
if node.on:
|
||||
embed.add_field(name="On", value="、 ".join(node.on), inline=False)
|
||||
|
||||
embed.add_field(name=f"Radical: {node.radical[0]}", value=node.radical[1], inline=False)
|
||||
|
||||
embed.set_thumbnail(url=node.image_url)
|
||||
|
||||
embed.set_footer(text=f"Jōyō kanji (Grade {node.grade}) | JLPT level {node.jlpt}\t\t{self.activeObject.page + 1}/{self.activeObject.total_pages}")
|
||||
|
||||
return embed
|
||||
|
||||
@commands.Cog.listener()
|
||||
async def on_reaction_add(self, reaction, user):
|
||||
message = reaction.message
|
||||
if message.id != self.latestMessage:
|
||||
return
|
||||
|
||||
if user == self.bot.user:
|
||||
return
|
||||
|
||||
if user.id != self.activeObject.owner:
|
||||
return
|
||||
|
||||
if reaction.me:
|
||||
if reaction.emoji == "⬅️":
|
||||
self.activeObject.prev()
|
||||
await reaction.remove(user)
|
||||
|
||||
if reaction.emoji == "➡️":
|
||||
self.activeObject.next()
|
||||
await reaction.remove(user)
|
||||
|
||||
embed = await self.createEmbed()
|
||||
await message.edit(embed=embed)
|
||||
self.activeObjects = {}
|
||||
|
||||
@commands.command(name="kanji", description="Performs a Kanji search", usage="<kanji>", aliases=["k"])
|
||||
@commands.cooldown(1, 5)
|
||||
async def kanji(self, ctx, *, kanji: str = None):
|
||||
if kanji is None:
|
||||
embed = discord.Embed(
|
||||
title = "No search results",
|
||||
description = "The search returned nothing. Did you make a typo?",
|
||||
colour = 0x56d926
|
||||
)
|
||||
await ctx.send(embed=embed)
|
||||
return
|
||||
|
||||
embed = discord.Embed(
|
||||
|
|
@ -92,19 +87,23 @@ class Kanji(commands.Cog):
|
|||
description = "This might take a few seconds",
|
||||
colour = 0x56d926
|
||||
)
|
||||
|
||||
message = await ctx.send(embed=embed)
|
||||
|
||||
kanji = kanji[:5]
|
||||
response = jisho.JishoKanji(kanji)
|
||||
|
||||
if response.entries < 1:
|
||||
embed = discord.Embed(
|
||||
title = "No search results",
|
||||
description = "The search returned nothing. Did you make a typo?",
|
||||
colour = 0x56d926
|
||||
)
|
||||
await message.edit(embed=embed)
|
||||
return
|
||||
|
||||
self.activeObject = JishoKanjiObject(kanji, ctx.author.id)
|
||||
embed = await self.createEmbed()
|
||||
await message.edit(embed=embed)
|
||||
self.latestMessage = message.id
|
||||
|
||||
if self.activeObject.total_pages > 1:
|
||||
await message.add_reaction("⬅️")
|
||||
await message.add_reaction("➡️")
|
||||
await message.delete()
|
||||
self.activeObjects[ctx.channel.id] = KanjiEmbed(self, ctx, response)
|
||||
await self.activeObjects[ctx.channel.id].show_embed()
|
||||
|
||||
def setup(bot):
|
||||
bot.add_cog(Kanji(bot))
|
||||
Loading…
Reference in a new issue