type hints

Author: esblinov

transfunctions/transformer.py

@@ -1,12 +1,12 @@
 from sys import version_info
-from typing import Optional, Union, List, Any
+from typing import Optional, Union, List, Dict, Any
 from types import MethodType, FunctionType
 from collections.abc import Callable
 from inspect import isfunction, iscoroutinefunction, getsource, getfile
 from ast import parse, NodeTransformer, Expr, AST, FunctionDef, AsyncFunctionDef, increment_lineno, Await, Call, With, Return, Name, Load, Assign, Constant, Store, arguments
 from functools import wraps, update_wrapper
 
-from dill.source import getsource as dill_getsource
+from dill.source import getsource as dill_getsource  # type: ignore[import-untyped]
 
 from transfunctions.errors import CallTransfunctionDirectlyError, DualUseOfDecoratorError, WrongDecoratorSyntaxError
 
@@ -27,7 +27,7 @@ def __init__(self, function: Callable, decorator_lineno: int, decorator_name: st
         self.decorator_name = decorator_name
         self.extra_transformers = extra_transformers
         self.base_object = None
-        self.cache = {}
+        self.cache: Dict[str, Callable] = {}
 
     def __call__(self, *args: Any, **kwargs: Any) -> None:
         raise CallTransfunctionDirectlyError("You can't call a transfunction object directly, create a function, a generator function or a coroutine function from it.")
@@ -102,7 +102,7 @@ def extract_context(self, context_name: str, addictional_transformers: Optional[
         if context_name in self.cache:
             return self.cache[context_name]
         try:
-            source_code = getsource(self.function)
+            source_code: str = getsource(self.function)
         except OSError:
             source_code = dill_getsource(self.function)
 
@@ -127,7 +127,7 @@ def visit_With(self, node: With) -> Optional[Union[AST, List[AST]]]:
                 return node
 
         class DeleteDecorator(NodeTransformer):
-            def visit_FunctionDef(self, node: Expr) -> Optional[Union[AST, List[AST]]]:
+            def visit_FunctionDef(self, node: FunctionDef) -> Optional[Union[AST, List[AST]]]:
                 if node.name == original_function.__name__:
                     nonlocal transfunction_decorator
                     transfunction_decorator = None
@@ -161,7 +161,7 @@ def visit_FunctionDef(self, node: Expr) -> Optional[Union[AST, List[AST]]]:
             increment_lineno(tree, n=(self.decorator_lineno - transfunction_decorator.lineno - 1))
 
         code = compile(tree, filename=getfile(self.function), mode='exec')
-        namespace = {}
+        namespace: Dict[str, Callable] = {}
         exec(code, namespace)
         function_factory = namespace['wrapper']
         result = function_factory()