Skip to content

Commit

Permalink
Merge pull request #2032 from LazyDope/fix-save-death
Browse files Browse the repository at this point in the history
make !save death a subcommand
  • Loading branch information
SeanStoves committed Apr 7, 2024
2 parents ae4214a + 579e8ad commit 1cd4ec2
Showing 1 changed file with 16 additions and 13 deletions.
29 changes: 16 additions & 13 deletions cogs5e/sheetManager.py
Original file line number Diff line number Diff line change
Expand Up @@ -238,26 +238,15 @@ async def attack_delete(self, ctx, name):
await character.commit(ctx)
await ctx.send(f"Okay, deleted attack {attack.name}.")

@commands.command(
@commands.group(
aliases=["s"],
invoke_without_command=True,
help=f"""
Rolls a save for your current active character.
{VALID_SAVE_ARGS}
""",
)
async def save(self, ctx, skill, *, args=""):
if skill == "death":
base_cmd = "game deathsave"
if args and (sub_cmd := args.split()[0].lower()) in ("fail", "success", "reset"):
base_cmd += f" {sub_cmd}"
args = ""
ds_cmd = self.bot.get_command(base_cmd)
if ds_cmd is None:
return await ctx.send("Error: GameTrack cog not loaded.")
if args:
return await ctx.invoke(ds_cmd, args=args)
return await ctx.invoke(ds_cmd)

char: Character = await ctx.get_character()

args = await self.new_arg_stuff(args, ctx, char, base_args=[skill])
Expand All @@ -277,6 +266,20 @@ async def save(self, ctx, skill, *, args=""):
if gamelog := self.bot.get_cog("GameLog"):
await gamelog.send_save(ctx, char, result.skill_name, result.rolls)

@save.command(name="death", help="Equivalent to `!game deathsave`")
async def save_death(self, ctx, *, args=""):
base_cmd = "game deathsave"
if args and (sub_cmd := args.split()[0].lower()) in ("fail", "success", "reset"):
base_cmd += f" {sub_cmd}"
ds_cmd = self.bot.get_command(base_cmd)
if ds_cmd is None:
return await ctx.send("Error: GameTrack cog not loaded.")

if base_cmd == "game deathsave":
return await ctx.invoke(ds_cmd, args=args)
else:
return await ctx.invoke(ds_cmd)

@commands.command(
aliases=["c"],
help=f"""
Expand Down

0 comments on commit 1cd4ec2

Please sign in to comment.