Merge pull request #396 from abe-101/add-type-hints

Add type hints

Author: faif

patterns/behavioral/iterator.py

@@ -7,7 +7,7 @@
 """
 
 
-def count_to(count):
+def count_to(count: int):
     """Counts by word numbers, up to a maximum of five"""
     numbers = ["one", "two", "three", "four", "five"]
     yield from numbers[:count]

patterns/behavioral/template.py

@@ -10,28 +10,28 @@
 """
 
 
-def get_text():
+def get_text() -> str:
     return "plain-text"
 
 
-def get_pdf():
+def get_pdf() -> str:
     return "pdf"
 
 
-def get_csv():
+def get_csv() -> str:
     return "csv"
 
 
-def convert_to_text(data):
+def convert_to_text(data: str) -> str:
     print("[CONVERT]")
     return f"{data} as text"
 
 
-def saver():
+def saver() -> None:
     print("[SAVE]")
 
 
-def template_function(getter, converter=False, to_save=False):
+def template_function(getter, converter=False, to_save=False) -> None:
     data = getter()
     print(f"Got `{data}`")