Minor cleaning of the code generator:

  - Adding some type annotations
  - Fixing some formatting issues
  - Removing unused imports
  - Replacing some code with more pythonic versions

Test: run the code generator, observe no change in generated code

BUG=223858461

Change-Id: I16e009982cccc0d44c3d11ec4b476d4742e40dd3
6 files changed