From dcc7dc15fb53d5841112f88c1f288c8b9058bf43 Mon Sep 17 00:00:00 2001
From: Alexander Sheykin <alexander.sheykin@it.adcombo.com>
Date: Mon, 3 Jan 2022 00:13:11 +0300
Subject: [PATCH] Add "forever" keyword for !ro and !ban aiogram#10

---
 aiogram_bot/handlers/simple_admin.py | 25 +++++++++++++++++--------
 aiogram_bot/utils/timedelta.py       |  6 ++++--
 2 files changed, 21 insertions(+), 10 deletions(-)

diff --git a/aiogram_bot/handlers/simple_admin.py b/aiogram_bot/handlers/simple_admin.py
index 25d6e7e..cbe8947 100644
--- a/aiogram_bot/handlers/simple_admin.py
+++ b/aiogram_bot/handlers/simple_admin.py
@@ -1,4 +1,5 @@
 import asyncio
+import datetime
 from contextlib import suppress
 from typing import List, Optional
 
@@ -84,12 +85,16 @@ async def cmd_ro(message: types.Message, chat: Chat):
         logger.error("Failed to restrict chat member: {error!r}", error=e)
         return False
 
+    if duration >= datetime.timedelta(days=367):
+        duration = "forever"
+    else:
+        duration = format_timedelta(
+            duration, locale=chat.language, granularity="seconds", format="short"
+        )
+
     await message.reply_to_message.answer(
         _("<b>Read-only</b> activated for user {user}. Duration: {duration}").format(
-            user=message.reply_to_message.from_user.get_mention(),
-            duration=format_timedelta(
-                duration, locale=chat.language, granularity="seconds", format="short"
-            ),
+            user=message.reply_to_message.from_user.get_mention(), duration=duration
         )
     )
     return True
@@ -120,12 +125,16 @@ async def cmd_ban(message: types.Message, chat: Chat):
         logger.error("Failed to kick chat member: {error!r}", error=e)
         return False
 
+    if duration >= datetime.timedelta(days=367):
+        duration = "forever"
+    else:
+        duration = format_timedelta(
+            duration, locale=chat.language, granularity="seconds", format="short"
+        )
+
     await message.reply_to_message.answer(
         _("User {user} <b>banned</b> for {duration}").format(
-            user=message.reply_to_message.from_user.get_mention(),
-            duration=format_timedelta(
-                duration, locale=chat.language, granularity="seconds", format="short"
-            ),
+            user=message.reply_to_message.from_user.get_mention(), duration=duration
         )
     )
     return True
diff --git a/aiogram_bot/utils/timedelta.py b/aiogram_bot/utils/timedelta.py
index 0637b47..fb4802a 100644
--- a/aiogram_bot/utils/timedelta.py
+++ b/aiogram_bot/utils/timedelta.py
@@ -4,8 +4,8 @@
 
 from aiogram import types
 
-PATTERN = re.compile(r"(?P<value>\d+)(?P<modifier>[wdhms])")
-LINE_PATTERN = re.compile(r"^(\d+[wdhms]){1,}$")
+PATTERN = re.compile(r"(?P<value>\d+)(?P<modifier>[wdhms])|^(?P<keyword>forever)$")
+LINE_PATTERN = re.compile(r"^(\d+[wdhms])+$|^(forever)$")
 
 MODIFIERS = {
     "w": datetime.timedelta(weeks=1),
@@ -26,6 +26,8 @@ def parse_timedelta(value: str) -> datetime.timedelta:
         raise TimedeltaParseError("Invalid time format")
 
     try:
+        if PATTERN.match(value).group("keyword"):
+            return datetime.timedelta(days=367)
         result = datetime.timedelta()
         for match in PATTERN.finditer(value):
             value, modifier = match.groups()