Add type hints in template

worker-{{cookiecutter.slug}}/setup.py

@@ -2,6 +2,7 @@
 # -*- coding: utf-8 -*-
 
 from pathlib import Path
+from typing import List
 
 from setuptools import find_packages, setup
 
@@ -9,20 +10,20 @@ MODULE = "worker_{{cookiecutter.slug}}"
 COMMAND = "worker-{{cookiecutter.slug}}"
 
 
-def parse_requirements_line(line):
+def parse_requirements_line(line) -> str:
     """Special case for git requirements"""
     if line.startswith("git+http"):
         assert "@" in line, "Branch should be specified with suffix (ex: @master)"
         assert (
             "#egg=" in line
         ), "Package name should be specified with suffix (ex: #egg=kraken)"
-        package_name = line.split("#egg=")[-1]
+        package_name: str = line.split("#egg=")[-1]
         return f"{package_name} @ {line}"
     else:
         return line
 
 
-def parse_requirements():
+def parse_requirements() -> List[str]:
     path = Path(__file__).parent.resolve() / "requirements.txt"
     assert path.exists(), f"Missing requirements: {path}"
     return list(

worker-{{cookiecutter.slug}}/tests/conftest.py

@@ -8,7 +8,7 @@ from arkindex_worker.worker.base import BaseWorker
 
 
 @pytest.fixture(autouse=True)
-def setup_environment(responses, monkeypatch):
+def setup_environment(responses, monkeypatch) -> None:
     """Setup needed environment variables"""
 
     # Allow accessing remote API schemas

worker-{{cookiecutter.slug}}/worker_{{cookiecutter.slug}}/worker.py

 # -*- coding: utf-8 -*-
-import logging
+from logging import Logger, getLogger
 
+from arkindex_worker.models import Element
 from arkindex_worker.worker import ElementsWorker
 
-logger = logging.getLogger(__name__)
+logger: Logger = getLogger(__name__)
 
 
 class Demo(ElementsWorker):
-    def process_element(self, element):
+    def process_element(self, element: Element) -> None:
         logger.info(f"Demo processing element ({element.id})")
 
 
-def main():
+def main() -> None:
     Demo(description="{{ cookiecutter.description }}").run()