python: add type information

Signed-off-by: Axel Heider <axelheider@gmx.de>

tools/hardware/outputs/c_header.py

@@ -7,7 +7,7 @@
 ''' generate a c header file from the device tree '''
 import argparse
 import builtins
-
+from typing import Dict, List
 from jinja2 import Environment, BaseLoader
 
 from hardware import config, fdt
@@ -120,7 +120,7 @@ static const p_region_t BOOT_RODATA avail_p_regs[] = {
 '''
 
 
-def get_kernel_devices(tree: fdt.FdtParser, rules: rule.HardwareYaml):
+def get_kernel_devices(tree: fdt.FdtParser, rules: rule.HardwareYaml) -> (List, Dict):
     '''
     Given a device tree and a set of rules, returns a tuple (groups, offsets).
 
@@ -151,7 +151,7 @@ def get_kernel_devices(tree: fdt.FdtParser, rules: rule.HardwareYaml):
     return (groups, offsets)
 
 
-def get_interrupts(tree: fdt.FdtParser, rules: rule.HardwareYaml):
+def get_interrupts(tree: fdt.FdtParser, rules: rule.HardwareYaml) -> List:
     ''' Get dict of interrupts, {label: KernelInterrupt} from the DT and hardware rules. '''
     kernel_devices = tree.get_kernel_devices()
 

tools/hardware/utils/rule.py

@@ -78,7 +78,7 @@ class KernelRegionGroup:
         self.kernel_offset = offset
         return offset + self.size
 
-    def get_labelled_addresses(self):
+    def get_labelled_addresses(self) -> Dict:
         ''' Get a dict of address -> label for the kernel '''
         ret = {}
         for (k, v) in self.labels.items():