feat(shard): ✨ Store unbanner.

This commit is contained in:
2025-03-16 00:01:04 -03:00
parent 515048f43d
commit 0067481528
4 changed files with 29 additions and 10 deletions
+9 -2
View File
@@ -304,14 +304,21 @@ class Payphone(commands.AutoShardedBot):
return ban return ban
async def unban(self, user_id: int) -> None: async def unban(self, user_id: int, expired_by: int | None) -> None:
"""Unbans a user. """Unbans a user.
Args: Args:
user_id (int): The user to unban's ID. user_id (int): The user to unban's ID.
expired_by (int | None): The user who expired the ban.
""" """
await models.Ban.filter(user_id=user_id).delete() await models.Ban.filter(
(
expressions.Q(expires_at__gt=datetime.now(timezone.utc))
| expressions.Q(expires_at=None)
)
& expressions.Q(user_id=user_id)
).update(expires_at=datetime.now(timezone.utc), expired_by_id=expired_by)
await self.redis.delete(f"banned:{user_id}") await self.redis.delete(f"banned:{user_id}")
+16 -3
View File
@@ -84,7 +84,7 @@ class Admin(base.Cog):
user = user or ctx.author user = user or ctx.author
await self.bot.unban(user.id) await self.bot.unban(user.id, ctx.author.id)
await ctx.send(f"{user.mention} was unbanned from Payphone.") await ctx.send(f"{user.mention} was unbanned from Payphone.")
@@ -163,7 +163,7 @@ class Admin(base.Cog):
bans = ( bans = (
await models.Ban.filter(user_id=user_id) await models.Ban.filter(user_id=user_id)
.prefetch_related("report", "created_by") .prefetch_related("report", "created_by", "expired_by")
.order_by("-created_at") .order_by("-created_at")
) )
@@ -209,13 +209,26 @@ class Admin(base.Cog):
thumbnail=ban.report.user_avatar_url, thumbnail=ban.report.user_avatar_url,
footer=discord.EmbedFooter( footer=discord.EmbedFooter(
text=f"Banned by {(await self.bot.get_or_fetch_user(ban.created_by.id)).display_name}" text=f"Banned by {(await self.bot.get_or_fetch_user(ban.created_by.id)).display_name}"
+ (
f" · Unbanned by {(await self.bot.get_or_fetch_user(ban.expired_by.id)).display_name}"
if ban.expired_by
else ""
)
), ),
) )
if ban.report if ban.report
else discord.Embed( else discord.Embed(
color=constants.PRIMARY_COLOR, color=constants.PRIMARY_COLOR,
title=f"Ban #{ban.id}", title=f"Ban #{ban.id}",
description=f"The user was banned manually by <@!{ban.created_by.id}> <t:{int(ban.created_at.timestamp())}:R>.", description=f"The user was banned manually by <@!{ban.created_by.id}> <t:{int(ban.created_at.timestamp())}:R>. It expires {f"<t:{int(ban.expires_at.timestamp())}:R>" if ban.expires_at else "never"}.",
footer=discord.EmbedFooter(
text=f"Banned by {(await self.bot.get_or_fetch_user(ban.created_by.id)).display_name}"
+ (
f" · Unbanned by {(await self.bot.get_or_fetch_user(ban.expired_by.id)).display_name}"
if ban.expired_by
else ""
)
),
) )
) )
for ban in bans for ban in bans
+1 -1
View File
@@ -149,7 +149,7 @@ class Ban(models.Model):
created_at = fields.DatetimeField(auto_now_add=True) created_at = fields.DatetimeField(auto_now_add=True)
expires_at = fields.DatetimeField(null=True) expires_at = fields.DatetimeField(null=True)
created_by = fields.ForeignKeyField("models.User", related_name="created_bans") created_by = fields.ForeignKeyField("models.User", related_name="created_bans")
expired_by = fields.ForeignKeyField("models.User", related_name="bans", null=True) expired_by = fields.ForeignKeyField("models.User", related_name="expired_bans", null=True)
class Meta: class Meta:
""" """
Generated
+3 -4
View File
@@ -1,5 +1,4 @@
version = 1 version = 1
revision = 1
requires-python = ">=3.12" requires-python = ">=3.12"
[[package]] [[package]]
@@ -128,7 +127,7 @@ version = "8.1.8"
source = { registry = "https://pypi.org/simple" } source = { registry = "https://pypi.org/simple" }
dependencies = [ dependencies = [
{ name = "anyio" }, { name = "anyio" },
{ name = "colorama", marker = "sys_platform == 'win32'" }, { name = "colorama", marker = "platform_system == 'Windows'" },
] ]
sdist = { url = "https://files.pythonhosted.org/packages/cb/b5/e1e5fdf1c1bb7e6e614987c120a98d9324bf8edfaa5f5cd16a6235c9d91b/asyncclick-8.1.8.tar.gz", hash = "sha256:0f0eb0f280e04919d67cf71b9fcdfb4db2d9ff7203669c40284485c149578e4c", size = 232900 } sdist = { url = "https://files.pythonhosted.org/packages/cb/b5/e1e5fdf1c1bb7e6e614987c120a98d9324bf8edfaa5f5cd16a6235c9d91b/asyncclick-8.1.8.tar.gz", hash = "sha256:0f0eb0f280e04919d67cf71b9fcdfb4db2d9ff7203669c40284485c149578e4c", size = 232900 }
wheels = [ wheels = [
@@ -189,7 +188,7 @@ name = "click"
version = "8.1.8" version = "8.1.8"
source = { registry = "https://pypi.org/simple" } source = { registry = "https://pypi.org/simple" }
dependencies = [ dependencies = [
{ name = "colorama", marker = "sys_platform == 'win32'" }, { name = "colorama", marker = "platform_system == 'Windows'" },
] ]
sdist = { url = "https://files.pythonhosted.org/packages/b9/2e/0090cbf739cee7d23781ad4b89a9894a41538e4fcf4c31dcdd705b78eb8b/click-8.1.8.tar.gz", hash = "sha256:ed53c9d8990d83c2a27deae68e4ee337473f6330c040a31d4225c9574d16096a", size = 226593 } sdist = { url = "https://files.pythonhosted.org/packages/b9/2e/0090cbf739cee7d23781ad4b89a9894a41538e4fcf4c31dcdd705b78eb8b/click-8.1.8.tar.gz", hash = "sha256:ed53c9d8990d83c2a27deae68e4ee337473f6330c040a31d4225c9574d16096a", size = 226593 }
wheels = [ wheels = [
@@ -405,7 +404,7 @@ wheels = [
[[package]] [[package]]
name = "payphone" name = "payphone"
version = "0.42.0" version = "0.43.0"
source = { virtual = "." } source = { virtual = "." }
dependencies = [ dependencies = [
{ name = "aerich" }, { name = "aerich" },