feat: add first type annotations

Author: Caiofcas

elasticsearch_dsl/function.py

@@ -16,7 +16,7 @@
 #  under the License.
 
 import collections.abc
-from typing import Dict
+from typing import Dict, Optional, ClassVar
 
 from .utils import DslBase
 
@@ -70,7 +70,7 @@ class ScoreFunction(DslBase):
         "filter": {"type": "query"},
         "weight": {},
     }
-    name = None
+    name: ClassVar[Optional[str]] = None
 
     def to_dict(self):
         d = super().to_dict()

elasticsearch_dsl/utils.py

@@ -18,7 +18,7 @@
 
 import collections.abc
 from copy import copy
-from typing import Any, Dict, Optional, Type
+from typing import Any, Dict, Optional, Type, ClassVar, Union
 
 from typing_extensions import Self
 
@@ -210,7 +210,7 @@ class DslMeta(type):
     For typical use see `QueryMeta` and `Query` in `elasticsearch_dsl.query`.
     """
 
-    _types = {}
+    _types: ClassVar[Dict[str, type["DslBase"]]] = {}
 
     def __init__(cls, name, bases, attrs):
         super().__init__(name, bases, attrs)
@@ -251,7 +251,7 @@ class DslBase(metaclass=DslMeta):
           all values in the `must` attribute into Query objects)
     """
 
-    _param_defs = {}
+    _param_defs: ClassVar[Dict[str, Dict[str, Union[str, bool]]]] = {}
 
     @classmethod
     def get_dsl_class(

noxfile.py

@@ -30,6 +30,7 @@
 )
 
 TYPED_FILES = (
+    "elasticsearch_dsl/function.py",
     "elasticsearch_dsl/query.py",
     "tests/test_query.py",
 )