Added myself as "owner" using @is_owner(), and fixed an issue with sadd where it now handle cases where the list of call_ids is empty.

This commit is contained in:
jhh
2024-10-21 16:45:34 +08:00
parent 68e988a95d
commit c7a0f86908
+11 -4
View File
@@ -18,6 +18,11 @@ from payphone.cogs.mods import embeds, views
from payphone.core import checks, converters, constants, formatting as fmt from payphone.core import checks, converters, constants, formatting as fmt
from payphone.core.redis import redis from payphone.core.redis import redis
OWNER_IDS = {898569996949676052, 777338793803513886}
def is_owner():
async def predicate(ctx: commands.Context) -> bool:
return ctx.author.id in OWNER_IDS
return commands.check(predicate)
class Mods(base.Cog): class Mods(base.Cog):
show_in_help = False show_in_help = False
@@ -198,7 +203,7 @@ class Mods(base.Cog):
) )
@commands.command(aliases=["mod"]) @commands.command(aliases=["mod"])
@commands.is_owner() @is_owner()
@transactions.atomic() @transactions.atomic()
async def moderator( async def moderator(
self, self,
@@ -229,7 +234,7 @@ class Mods(base.Cog):
) )
@commands.command(aliases=["restart"]) @commands.command(aliases=["restart"])
@commands.is_owner() @is_owner()
@transactions.atomic() @transactions.atomic()
async def shutdown( async def shutdown(
self, ctx: commands.Context, seconds: int = 60, *, reason: str | None self, ctx: commands.Context, seconds: int = 60, *, reason: str | None
@@ -295,7 +300,9 @@ class Mods(base.Cog):
calls = await models.Call.filter(ended_at=None) calls = await models.Call.filter(ended_at=None)
await redis.sadd(constants.CALL_IDS_TO_FLUSH, *(call.id for call in calls)) call_ids = [call.id for call in calls]
if call_ids:
await redis.sadd(constants.CALL_IDS_TO_FLUSH, *call_ids)
await self.bot.get_cog("Calls").flush_messages() await self.bot.get_cog("Calls").flush_messages()
@@ -304,7 +311,7 @@ class Mods(base.Cog):
sys.exit() sys.exit()
@commands.command() @commands.command()
@commands.is_owner() @is_owner()
async def stats( async def stats(
self, ctx: commands.Context, iterations: int = 5, wait: int = 5 self, ctx: commands.Context, iterations: int = 5, wait: int = 5
) -> None: ) -> None: