[PATCH v3 17/17] Add typing in decorator

[Muhammad Rizki <kiizuha@gnuweeb.org>]

I want to add code typing on @handle_flood, so when the function were
called, it display the hints.

Signed-off-by: Muhammad Rizki <kiizuha@gnuweeb.org>
---
 daemon/packages/decorator.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/daemon/packages/decorator.py b/daemon/packages/decorator.py
index 7d7dc39..c7a5f02 100644
--- a/daemon/packages/decorator.py
+++ b/daemon/packages/decorator.py
@@ -4,15 +4,20 @@
 #
 
 from pyrogram.errors.exceptions.flood_420 import FloodWait
-from typing import Any, Callable
+from pyrogram.types import Message
+from typing import Any, Callable, TypeVar
 from functools import wraps
 import re
 import asyncio
 
 __all__ = ["handle_flood"]
 
+T = TypeVar("T", bound=Message)
 
-def handle_flood(func: Callable[[Any], Any]) -> Callable[[Any], Any]:
+#
+# TODO(Muhammad Rizki): Add more typing for @handle_flood
+#
+def handle_flood(func: Callable[[T], T]) -> Callable[[T], T]:
 	@wraps(func)
 	async def callback(*args: Any) -> Any:
 		while True:
-- 
Muhammad Rizki