Add Protocol to factory pattern

Mateus Furquim · Mateus Furquim · commit ad6fd4b8e3ae · 2023-01-20T13:24:38.000-03:00
diff --git a/patterns/creational/factory.py b/patterns/creational/factory.py
@@ -21,6 +21,14 @@
 *TL;DR
 Creates objects without having to specify the exact class.
 """
+from typing import Dict
+from typing import Protocol
+from typing import Type
+
+
+class Localizer(Protocol):
+ def localize(self, msg: str) -> str:
+ pass
 
 
 class GreekLocalizer:
@@ -41,10 +49,10 @@ def localize(self, msg: str) -> str:
 return msg
 
 
-def get_localizer(language: str = "English") -> object:
+def get_localizer(language: str = "English") -> Localizer:
 
 """Factory"""
- localizers ={
+ localizers: Dict[str, Type[Localizer]] ={
 "English": EnglishLocalizer,
 "Greek": GreekLocalizer,
 }

-Original file line number
+Diff line change
 *TL;DR
 Creates objects without having to specify the exact class.
 """
 +fromtypingimportDict
 +fromtypingimportProtocol
 +fromtypingimportType
++
++
 +classLocalizer(Protocol):
 +deflocalize(self, msg: str) ->str:
 +pass
 classGreekLocalizer:
 returnmsg
 -defget_localizer(language: str="English") ->object:
 +defget_localizer(language: str="English") ->Localizer:
 """Factory"""
 -localizers={
 +localizers: Dict[str, Type[Localizer]]={
 "English": EnglishLocalizer,
 "Greek": GreekLocalizer,
+ }