Croc/croccybot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(nowlist): Implement edit cmd.

Browse Source
This commit is contained in:
Interitio
2024-10-07 12:19:03 +10:00
parent 79645177bd
commit 8991b1a641
1 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/modules/nowdoing/cog.py
View File

@@ -194,16 +194,17 @@ class NowDoingCog(LionCog):
else: else:
await ctx.send(f"Hello {ctx.author.name}! I don't think you have permission to test that.") await ctx.send(f"Hello {ctx.author.name}! I don't think you have permission to test that.")
async def now(self, ctx: commands.Context | LionContext, profile: UserProfile, args: Optional[str] = None): async def now(self, ctx: commands.Context | LionContext, profile: UserProfile, args: Optional[str] = None, edit=False):
args = args.strip() if args else None args = args.strip() if args else None
userid = profile.profileid userid = profile.profileid
if args: if args:
existing = self.tasks.get(userid, None)
await self.data.Task.table.delete_where(userid=userid) await self.data.Task.table.delete_where(userid=userid)
task = await self.data.Task.create( task = await self.data.Task.create(
userid=userid, userid=userid,
name=ctx.author.display_name, name=ctx.author.display_name,
task=args, task=args,
started_at=utc_now(), started_at=existing.started_at if (existing and edit) else utc_now(),
) )
self.tasks[task.userid] = task self.tasks[task.userid] = task
await self.channel.send_set(*self.channel.task_args(task)) await self.channel.send_set(*self.channel.task_args(task))
@@ -241,6 +242,20 @@ class NowDoingCog(LionCog):
profile = await self.bot.get_cog('ProfileCog').fetch_profile_discord(ctx.author) profile = await self.bot.get_cog('ProfileCog').fetch_profile_discord(ctx.author)
await self.now(ctx, profile, args) await self.now(ctx, profile, args)
@commands.command(
name='edit',
)
async def twi_edit(self, ctx: commands.Context, *, args: Optional[str] = None):
profile = await self.bot.get_cog('ProfileCog').fetch_profile_twitch(ctx.author)
await self.now(ctx, profile, args, edit=True)
@cmds.hybrid_command(
name='edit',
)
async def disc_edit(self, ctx: LionContext, *, args: Optional[str] = None):
profile = await self.bot.get_cog('ProfileCog').fetch_profile_discord(ctx.author)
await self.now(ctx, profile, args, edit=True)
async def nownext(self, ctx: commands.Context | LionContext, profile: UserProfile, args: Optional[str]): async def nownext(self, ctx: commands.Context | LionContext, profile: UserProfile, args: Optional[str]):
userid = profile.profileid userid = profile.profileid
task = self.tasks.get(userid, None) task = self.tasks.get(userid, None)