CLI: api.register type preservation

This commit is contained in:
Johannes Kirschbauer
2024-09-16 13:16:57 +02:00
committed by kenji
parent f3a9a4e108
commit dc57c3d948
4 changed files with 7 additions and 5 deletions

View File

@@ -107,7 +107,9 @@ API.register(open_file)
self.register(wrapper)
return fn
def register(self, fn: Callable[..., T]) -> Callable[..., T]:
F = TypeVar("F", bound=Callable[..., Any])
def register(self, fn: F) -> F:
if fn.__name__ in self._registry:
msg = f"Function {fn.__name__} already registered"
raise ClanError(msg)