Adding type hints to your code

Type hints in Python are static annotations of variables and functions, to let humans more easily understand the code. See the standard library docs and this PyCascades 2018 talk.

Type hints are not required for all modules at the moment in Home Assistant, but we aim to have complete coverage of the core modules.

Adding type hints to an existing codebase can be a daunting task. To speed this up and help developers doing this, Instagram made the monkeytype program. It will analyze calls during runtime and try to assign the correct type hints to the code.

See this instagram blog post for a description of the workflow involved to use the monkeytype program.

We've added a script to start a run of our test suite or a test module and tell the monkeytype program to analyze the run.

Basic workflow

  1. Run script/monkeytype tests/path/to/your_test_module.py.
  2. Run monkeytype stub homeassistant.your_actual_module.
  3. Look at output from the monkeytyped typing stub. If not totally bad, apply the stub to your module. You most likely will need to manually edit the typing in the last step.
  4. Run monkeytype apply homeassistant.your_actual_module.
  5. Check the diff and manually correct the typing if needed. Commit, push the branch and make a PR.

Note: Applying a monkeytyped stub to a module that has existing typing annotations might error and not work. This tool is most useful for totally untyped modules.

Last updated on