feat(shard): ✨ Add duration and reason arguments to p.ban.

payphone/db/migrations/models/13_20250318130207_update.py

@@ -0,0 +1,11 @@
+from tortoise import BaseDBAsyncClient
+
+
+async def upgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE "ban" ADD "reason" TEXT;"""
+
+
+async def downgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE "ban" DROP COLUMN "reason";"""

payphone/db/models.py

@@ -146,11 +146,20 @@ class Ban(models.Model):
 
     user_id = fields.BigIntField()
     report = fields.ForeignKeyField("models.Report", null=True)
+    reason = fields.TextField(null=True, default=None)
     created_at = fields.DatetimeField(auto_now_add=True)
     expires_at = fields.DatetimeField(null=True)
     created_by = fields.ForeignKeyField("models.User", related_name="created_bans")
     expired_by = fields.ForeignKeyField("models.User", related_name="expired_bans", null=True)
 
+    @property
+    def is_expired(self) -> bool:
+        """
+        Whether the ban is expired.
+        """
+
+        return self.expires_at is not None and self.expires_at < datetime.now(timezone.utc)
+
     class Meta:
         """
         Meta information.