async def add(self, ctx: commands.Context):
"""Interactively add a Visual Novel to the database."""
def interactive_command_check(msg):
return msg.author == ctx.author and ctx.channel == msg.channel
name = await input_name(self.bot, ctx, interactive_command_check)
abbreviations = await input_abbreviations(self.bot, ctx,
interactive_command_check)
authors = await input_authors(self.bot, ctx, interactive_command_check)
store = await input_store(self.bot, ctx, interactive_command_check)
image = await input_image(self.bot, ctx, interactive_command_check)
android = await input_android_support(self.bot, ctx,
interactive_command_check)
undetermined = await input_undetermined(self.bot, ctx,
interactive_command_check)
vn = VisualNovel(
database=self.db,
name=name,
abbreviations=abbreviations,
authors=authors,
store=store,
image=image,
android_support=android,
undetermined=undetermined,
)
doc_id = vn.add_to_db()
self.bot.log.info("VN added to DB with ID %s", doc_id)
await vn.post_to_list(self.bot.channels)
await ctx.reply("VN added successfully!")
async def on_raw_reaction_add(self, payload):
channel_id = payload.channel_id
message_id = payload.message_id
member = payload.member
emoji = payload.emoji
if member == self.bot.user:
return
channel = None
if channel_id == self.bot.channels["vn_list"].id:
channel = self.bot.channels["vn_list"]
if channel_id == self.bot.channels["vn_undetermined"].id:
channel = self.bot.channels["vn_undetermined"]
if not channel:
return
message = await channel.fetch_message(message_id)
await message.remove_reaction(emoji, member)
vn = VisualNovel(database=self.db)
vn.load_from_db(message_id=message_id)
Rating = Query()
query = (Rating.member_id == member.id) & (Rating.vn_id == vn.doc_id)
if emoji.name == "👍":
self.db.table(TABLE_RATING).upsert(
{
"member_id": member.id,
"vn_id": vn.doc_id,
"rating": 1
}, query)
if emoji.name == "👎":
self.db.table(TABLE_RATING).upsert(
{
"member_id": member.id,
"vn_id": vn.doc_id,
"rating": -1
}, query)
if emoji.name == "❌":
self.db.table(TABLE_RATING).remove(query)
await vn.update_entry_ratings(message)
async def rebuild(self, ctx: commands.Context):
"""Reposts all the VNs again in the VN channels."""
async for message in self.bot.channels["vn_undetermined"].history(
limit=None):
await message.delete()
async for message in self.bot.channels["vn_list"].history(limit=None):
await message.delete()
for entry in sorted(self.db.table(TABLE_VISUAL_NOVEL).all(),
key=lambda doc: doc.doc_id):
vn = VisualNovel(database=self.db)
vn.load_from_db(doc_id=entry.doc_id)
await vn.post_to_list(self.bot.channels)
async def search(self, ctx: commands.Context, *, name: str):
"""Search for a Visual Novel by name or abbreviation."""
name = name.lower()
vn = VisualNovel(database=self.db)
try:
vn.load_from_db(name=name, abbreviations=[name])
except FileNotFoundError:
return await ctx.send("VN not found.")
channel = self.bot.channels["vn_undetermined" if vn.
undetermined else "vn_list"]
message = await channel.fetch_message(vn.message_id)
await ctx.reply(f"The VN {vn.name} can be found at {message.jump_url}")
async def delete(self, ctx: commands.Context, *, name: str):
"""Delete a Visual Novel from the database."""
name = name.lower()
vn = VisualNovel(database=self.db)
try:
vn.load_from_db(name=name, abbreviations=[name])
except FileNotFoundError:
return await ctx.reply("VN not found.")
channel = self.bot.channels["vn_undetermined" if vn.
undetermined else "vn_list"]
message = await channel.fetch_message(vn.message_id)
await message.delete()
self.db.table(TABLE_VISUAL_NOVEL).remove(doc_ids=[vn.doc_id])
self.db.table(TABLE_RATING).remove(where("vn_id") == vn.doc_id)
await ctx.reply(f"The VN {vn.name} got successfully deleted!")
async def update(self, ctx: commands.Context):
"""Posts a VN update in the VN News Channel."""
def interactive_command_check(msg):
return msg.author == ctx.author and ctx.channel == msg.channel
await ctx.reply(
"What VN do you want to publish an update? Enter a name or an abbreviation."
)
try:
vn_name = await self.bot.wait_for("message",
timeout=60.0,
check=interactive_command_check)
except asyncio.TimeoutError:
return await ctx.reply("You took long. Aborting.")
vn_name = vn_name.content.lower()
vn = VisualNovel(database=self.db)
try:
vn.load_from_db(name=vn_name, abbreviations=[vn_name])
except FileNotFoundError:
return await ctx.reply("VN not found.")
await ctx.reply("What is the URL to the update?")
try:
url = await self.bot.wait_for("message",
timeout=60.0,
check=interactive_command_check)
except asyncio.TimeoutError:
return await ctx.reply("You took long. Aborting.")
url = url.content
await ctx.reply("What is the title of the update?")
try:
title = await self.bot.wait_for("message",
timeout=60.0,
check=interactive_command_check)
except asyncio.TimeoutError:
return await ctx.reply("You took long. Aborting.")
title = title.content
title = f"{vn.name}: {title}"
embed = discord.Embed(
colour=discord.Colour.blurple(),
title=title,
url=url,
)
embed.add_field(name="Current Ratings", value=vn.calculate_ratings())
embed.add_field(name="Abbreviations", value=vn.pretty_abbreviations())
embed.add_field(name="Android Support",
value="Yes" if vn.android_support else "No")
embed.set_image(url=vn.image)
embed.set_author(
name="FVN Bot",
icon_url=
"https://media.discordapp.net/attachments/729276573496246304/747178571834982431/bonkshinbookmirrored.png",
)
embed.set_footer(
text=
f"Brought to you by Furry Visual Novels server. Join us for vn-lists, development channels and more. discord.gg/GFjSPkh"
)
msg = await self.bot.channels["vn_news"].send(
f"{title}\n{self.bot.roles['update_notification'].mention}",
embed=embed)
await msg.publish()
async def edit(self, ctx: commands.Context):
"""Edits the information about a VN."""
def interactive_command_check(msg):
return msg.author == ctx.author and ctx.channel == msg.channel
await ctx.reply(
"What VN do you want to edit? Enter a name or an abbreviation.")
try:
vn_name = await self.bot.wait_for("message",
timeout=60.0,
check=interactive_command_check)
except asyncio.TimeoutError:
return await ctx.reply("You took long. Aborting.")
vn_name = vn_name.content.lower()
vn = VisualNovel(database=self.db)
try:
vn.load_from_db(name=vn_name, abbreviations=[vn_name])
except FileNotFoundError:
return await ctx.reply("VN not found.")
await ctx.reply(
f"""What do you want to edit about the VN "{vn.name}"? Input the corresponding number.
1. Name
2. Abbreviations
3. Authors
4. Store Link
5. Image
6. Android Support
7. Is undetermined?
""")
try:
choice = await self.bot.wait_for("message",
timeout=60.0,
check=interactive_command_check)
except asyncio.TimeoutError:
return await ctx.reply("You took long. Aborting.")
choice = int(choice.content.strip())
if choice == 1:
vn.name = await input_name(self.bot, ctx,
interactive_command_check)
if choice == 2:
vn.abbreviations = await input_abbreviations(
self.bot, ctx, interactive_command_check)
if choice == 3:
vn.authors = await input_authors(self.bot, ctx,
interactive_command_check)
if choice == 4:
vn.store = await input_store(self.bot, ctx,
interactive_command_check)
if choice == 5:
vn.image = await input_image(self.bot, ctx,
interactive_command_check)
if choice == 6:
vn.android_support = await input_android_support(
self.bot, ctx, interactive_command_check)
if choice == 7:
vn.undetermined = await input_undetermined(
self.bot, ctx, interactive_command_check)
vn.update_to_db()
await vn.update_to_list(self.bot.channels)
await ctx.reply(f"VN {vn.name} updated successfully.")