From ed4912e87a8ed20eaec11bdfd954fc59495a7aeb Mon Sep 17 00:00:00 2001
From: Trim21 <trim21.me@gmail.com>
Date: Wed, 21 Aug 2024 03:49:16 +0800
Subject: [PATCH] style: fix mypy

---
 server/tmpl.py | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/server/tmpl.py b/server/tmpl.py
index 96a2704..0266a81 100644
--- a/server/tmpl.py
+++ b/server/tmpl.py
@@ -1,3 +1,4 @@
+import typing
 from collections.abc import Callable
 from datetime import datetime, timedelta
 from typing import Any
@@ -16,12 +17,24 @@
     auto_reload=DEV,
 )
 
+P = typing.ParamSpec("P")
+T = typing.TypeVar("T")
 
-def add_filter(s: str | Callable):
-    def real_wrapper(name: str, fn: Callable):
+
+@typing.overload
+def add_filter(s: str) -> Callable[[Callable[P, T]], Callable[P, T]]: ...
+
+
+@typing.overload
+def add_filter(s: Callable[P, T]) -> Callable[P, T]: ...
+
+
+def add_filter(s: str | Callable[P, T]) -> Any:
+    def real_wrapper(name: str, fn: Callable[P, T]) -> Callable[P, T]:
         if name in engine.filters:
             raise ValueError(f"filter '{name}' already exists")
         engine.filters[name] = fn
+        return fn
 
     if isinstance(s, str):
         return lambda fn: real_wrapper(s, fn)
@@ -73,7 +86,7 @@ def format_duration(seconds: timedelta) -> str:
 
 
 @add_filter
-def subject_type_readable(s: int):
+def subject_type_readable(s: int) -> str:
     match s:
         case 1:
             return "书籍"