wlinator/tux
1
Fork 0
mirror of https://github.com/allthingslinux/tux.git synced 2024-10-03 00:53:12 +00:00
Code Issues Releases Wiki Activity

Merge branch 'allthingslinux:main' into main

Browse source
This commit is contained in:
Biko 2024-07-11 03:21:07 +03:00 committed by GitHub
commit b6bfd29000
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 27 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
tux/cogs/utility/snippets.py
View file

@ -15,6 +15,7 @@ class Snippets(commands.Cog):
def __init__(self, bot: commands.Bot) -> None:
self.bot = bot
self.db_controller = DatabaseController().snippet
self.config_db = DatabaseController().guild_config
@commands.command(
name="snippets",
@ -123,13 +124,32 @@ class Snippets(commands.Cog):
# Check if the author of the snippet is the same as the user who wants to delete it and if theres no author don't allow deletion
author_id = snippet.snippet_user_id or 0
if author_id != ctx.author.id:
embed = EmbedCreator.create_error_embed(
title="Error",
description="You can only delete your own snippets.",
ctx=ctx,
)
await ctx.send(embed=embed)
return
conf = await self.config_db.get_guild_config_by_id(ctx.guild.id)
# TODO: this was quick and dirty, needs to be refactored
user_roles = [role.id for role in ctx.author.roles] # type: ignore
if not conf:
embed = EmbedCreator.create_error_embed(
title="Error",
description="You can only delete your own snippets.",
ctx=ctx,
)
await ctx.send(embed=embed)
return
if (
conf.guild_base_staff_role_id is not None
and conf.guild_base_staff_role_id in user_roles
or conf.guild_dev_role_id is not None
and conf.guild_dev_role_id in user_roles
):
pass
else:
embed = EmbedCreator.create_error_embed(
title="Error",
description="You can only delete your own snippets.",
ctx=ctx,
)
await ctx.send(embed=embed)
return
await self.db_controller.delete_snippet_by_id(snippet.snippet_id)