Added type hints to factory pattern

rednafi · rednafi · commit 997f283e9904 · 2020-06-22T01:44:14.000+06:00
diff --git a/patterns/creational/factory.py b/patterns/creational/factory.py
@@ -29,27 +29,29 @@
 class GreekLocalizer:
     """A simple localizer a la gettext"""
 
-    def __init__(self):
+    def __init__(self) -> None:
         self.translations = {"dog": "σκύλος", "cat": "γάτα"}
 
-    def localize(self, msg):
+    def localize(self, msg: str) -> str:
         """We'll punt if we don't have a translation"""
         return self.translations.get(msg, msg)
 
 
 class EnglishLocalizer:
     """Simply echoes the message"""
 
-    def localize(self, msg):
+    def localize(self, msg: str) -> str:
         return msg
 
 
-def get_localizer(language="English"):
+def get_localizer(language: str = "English",) -> object:
+
     """Factory"""
     localizers = {
         "English": EnglishLocalizer,
         "Greek": GreekLocalizer,
     }
+
     return localizers[language]()
 
 
@@ -70,4 +72,5 @@ def main():
 
 if __name__ == "__main__":
     import doctest
+
     doctest.testmod()
