Improved class inheritance and type annotations

Author: WeirdCarrotMonster

yatq/worker/factory/base.py

@@ -8,6 +8,9 @@
 
 
 class BaseJobFactory(ABC):
+    def __init__(self, **kwargs):
+        super().__init__()
+
     @abstractmethod
     def create_job(self, task: "Task") -> BaseJob:  # pragma: no cover
         ...

yatq/worker/factory/simple.py

@@ -14,7 +14,9 @@ class SimpleJobFactory(BaseJobFactory, Generic[T_SimpleJobClass]):
     def __init__(
         self,
         handlers: Dict[str, Type[T_SimpleJobClass]],
+        **kwargs,
     ):
+        super().__init__(**kwargs)
         self.handlers = handlers
 
     def create_job(self, task: "Task") -> T_SimpleJobClass:

yatq/worker/worker_settings.py

@@ -3,6 +3,7 @@
 
 import aioredis
 
+from yatq.worker.factory.base import BaseJobFactory
 from yatq.worker.factory.simple import SimpleJobFactory
 from yatq.worker.job.base import BaseJob
 
@@ -11,7 +12,7 @@
 
 class WorkerSettings:
 
-    factory_cls: Type[SimpleJobFactory] = SimpleJobFactory
+    factory_cls: Type[BaseJobFactory] = SimpleJobFactory
     factory_kwargs: Optional[Dict] = None
 
     queue_namespace: Optional[str] = None