Add missing type annotations

mypy only checks types in functions that have a return type annotation.

In `check_generated_files`, change from returning `None` to returning a
boolean, since the function is supposed to check some boolean-valued
assertion. So far, the function always returns `True`.

In `main`, explicitly return an `int` value as expected by the caller. When
calling `check_generated_files`, convert the boolean result into an exit
status.

Signed-off-by: Gilles Peskine <Gilles.Peskine@arm.com>

Author: gilles-peskine-arm

scripts/make_generated_files.py

@@ -28,7 +28,7 @@ class GenerationScript:
     def __init__(self, script: Path, files: List[Path],
                  output_dir_option: Optional[str] = None,
                  output_file_option: Optional[str] = None,
-                 optional: bool = False):
+                 optional: bool = False) -> None:
         # Path from the root of Mbed TLS or TF-PSA-Crypto of the generation script
         self.script = script
 
@@ -56,7 +56,7 @@ def __init__(self, script: Path, files: List[Path],
         # consuming repository hasn't been updated yet.
         self.optional = optional
 
-def get_generation_script_files(generation_script: str):
+def get_generation_script_files(generation_script: str) -> List[Path]:
     """
     Get the list of the default paths of the files that a given script
     generates. It is assumed that the script supports the "--list" option.
@@ -188,7 +188,7 @@ def get_generation_script_files(generation_script: str):
                 get_generation_script_files("scripts/generate_visualc_files.pl"),
                 "--directory", None))
 
-def get_generated_files(generation_scripts: List[GenerationScript]):
+def get_generated_files(generation_scripts: List[GenerationScript]) -> List[Path]:
     """
     List the generated files in Mbed TLS or TF-PSA-Crypto. The path from root
     is returned for each generated files.
@@ -199,19 +199,21 @@ def get_generated_files(generation_scripts: List[GenerationScript]):
 
     return files
 
-def make_generated_files(generation_scripts: List[GenerationScript]):
+def make_generated_files(generation_scripts: List[GenerationScript]) -> None:
     """
     Generate the configuration independent files in their default location in
     the Mbed TLS or TF-PSA-Crypto tree.
     """
     for generation_script in generation_scripts:
         subprocess.run([generation_script.exe, str(generation_script.script)], check=True)
 
-def check_generated_files(generation_scripts: List[GenerationScript], root: Path):
+def check_generated_files(generation_scripts: List[GenerationScript],
+                          root: Path) -> bool:
     """
     Check that the given root directory contains the generated files as expected/
     generated by this script.
     """
+    ok = True
     for generation_script in generation_scripts:
         for file in generation_script.files:
             file = root / file
@@ -254,8 +256,9 @@ def check_generated_files(generation_scripts: List[GenerationScript], root: Path
                 print(f"Generated file {file} not identical to the reference one {ref_file}.")
             file.unlink()
             bak_file.rename(file)
+    return ok
 
-def main():
+def main() -> int:
     """
     Main function of this program
     """
@@ -286,10 +289,13 @@ def main():
         files = get_generated_files(generation_scripts)
         for file in files:
             print(str(file))
+        return 0
     elif args.check:
-        check_generated_files(generation_scripts, Path(args.root or "."))
+        ok = check_generated_files(generation_scripts, Path(args.root or "."))
+        return 0 if ok else 1
     else:
         make_generated_files(generation_scripts)
+        return 0 # Any error causes an exception
 
 if __name__ == "__main__":
     sys.exit(main())