skins/tiamat-cogs
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Tiamat: Fix monster embed length too long

Browse Source
This commit is contained in:
skins 2021-08-07 16:48:57 +01:00
parent dcfe01b58f
commit 71a6e740fb
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tiamat/tiamat.py
View File

@ -1,6 +1,7 @@
from redbot.core import commands from redbot.core import commands
import discord import discord
import requests import requests
from itertools import islice
class Tiamat(commands.Cog): class Tiamat(commands.Cog):
""" Get information from D&D 5e's srd """ """ Get information from D&D 5e's srd """
@ -21,15 +22,19 @@ class Tiamat(commands.Cog):
initial_embed.add_field(name="AC", value=data["armor_class"]) initial_embed.add_field(name="AC", value=data["armor_class"])
for ability in ["strength", "dexterity", "constitution", "intelligence", "wisdom", "charisma"]: for ability in ["strength", "dexterity", "constitution", "intelligence", "wisdom", "charisma"]:
initial_embed.add_field(name=ability, value=data[ability], inline=True) initial_embed.add_field(name=ability, value=data[ability], inline=True)
attack_embed = discord.Embed(title="Actions") attack_embeds = []
for i, action_group in enumerate(chunk(data["actions"], 3)):
attack_embed = discord.Embed(title=f"Actions [{i}]")
for action in data["actions"]: for action in data["actions"]:
attack_embed.add_field(name=action["name"], value=action["desc"]) attack_embed.add_field(name=action["name"], value=action["desc"])
attack_embeds.append(attack_embed)
legendary_embed = None legendary_embed = None
if "legendary_actions" in data: if "legendary_actions" in data:
legendary_embed = discord.Embed(title="Legendary Actions") legendary_embed = discord.Embed(title="Legendary Actions")
for action in data["legendary_actions"]: for action in data["legendary_actions"]:
legendary_embed.add_field(name=action["name"], value=action["desc"]) legendary_embed.add_field(name=action["name"], value=action["desc"])
await ctx.send(embed=initial_embed) await ctx.send(embed=initial_embed)
for attack_embed in attack_embeds:
await ctx.send(embed=attack_embed) await ctx.send(embed=attack_embed)
if legendary_embed is not None: if legendary_embed is not None:
await ctx.send(embed=legendary_embed) await ctx.send(embed=legendary_embed)
@ -101,3 +106,8 @@ class Tiamat(commands.Cog):
await ctx.send(embed=embed) await ctx.send(embed=embed)
except KeyError: except KeyError:
await ctx.send("Error: item is not a weapon") await ctx.send("Error: item is not a weapon")
def chunk(it, size):
it = iter(it)
return iter(lambda: tuple(islice(it, size)), ())