service locator type hints (#121)

jivago/inject/service_locator.py

@@ -1,4 +1,4 @@
-from typing import Callable, Optional
+from typing import Callable, Optional, TypeVar, Type, Union, List
 
 from jivago.inject import typing_meta_helper
 from jivago.inject.exception.instantiation_exception import InstantiationException
@@ -8,6 +8,9 @@
 from jivago.lang.registry import Registry
 from jivago.lang.stream import Stream
 
+T = TypeVar("T")
+S = TypeVar("S")
+
 
 class ServiceLocator(object):
     def __init__(self, registry=Registry()):
@@ -17,7 +20,7 @@ def __init__(self, registry=Registry()):
         self.providers = {}
         self.scopeCaches = []
 
-    def bind(self, interface, implementation):
+    def bind(self, interface: Type[T], implementation: Union[Type[T], Type[S], T, Callable[..., T]]):
         if self.__is_provider_function(implementation):
             self.providers[interface] = implementation
         elif isinstance(implementation, type):
@@ -28,7 +31,7 @@ def bind(self, interface, implementation):
     def register_scope(self, scope_cache: ScopeCache):
         self.scopeCaches.append(scope_cache)
 
-    def get(self, interface: type):
+    def get(self, interface: Type[T]) -> T:
         if interface in self.literals.keys():
             return self.literals[interface]
         if typing_meta_helper.is_typing_meta_collection(interface):
@@ -54,7 +57,7 @@ def get(self, interface: type):
             scope.store(stored_component, instance)
         return instance
 
-    def get_all(self, clazz: type):
+    def get_all(self, clazz: Type[T]) -> List[T]:
         return Stream(self.literals.keys(), self.components.keys(), self.providers.keys()).filter(
             lambda k: issubclass(k, clazz)).map(lambda k: self.get(k)).toList()