Add types to several exmaples

Author: eliben

examples/c-to-c.py

@@ -16,7 +16,7 @@
 from pycparser import parse_file, c_generator
 
 
-def translate_to_c(filename):
+def translate_to_c(filename: str) -> None:
     """Simply use the c_generator module to emit a parsed AST."""
     ast = parse_file(filename, use_cpp=True)
     generator = c_generator.CGenerator()

examples/construct_ast_from_scratch.py

@@ -21,7 +21,7 @@
 # }
 
 
-def empty_main_function_ast():
+def empty_main_function_ast() -> c_ast.FuncDef:
     constant_zero = c_ast.Constant(type="int", value="0")
     return_node = c_ast.Return(expr=constant_zero)
     compound_node = c_ast.Compound(block_items=[return_node])
@@ -46,7 +46,7 @@ def empty_main_function_ast():
     return main_func_node
 
 
-def generate_c_code(my_ast):
+def generate_c_code(my_ast: c_ast.Node) -> str:
     generator = c_generator.CGenerator()
     return generator.visit(my_ast)
 

examples/func_calls.py

@@ -18,18 +18,18 @@
 
 # A visitor with some state information (the funcname it's looking for)
 class FuncCallVisitor(c_ast.NodeVisitor):
-    def __init__(self, funcname):
+    def __init__(self, funcname: str) -> None:
         self.funcname = funcname
 
-    def visit_FuncCall(self, node):
-        if node.name.name == self.funcname:
+    def visit_FuncCall(self, node: c_ast.FuncCall) -> None:
+        if isinstance(node.name, c_ast.ID) and node.name.name == self.funcname:
             print("%s called at %s" % (self.funcname, node.name.coord))
         # Visit args in case they contain more func calls.
-        if node.args:
+        if node.args is not None:
             self.visit(node.args)
 
 
-def show_func_calls(filename, funcname):
+def show_func_calls(filename: str, funcname: str) -> None:
     ast = parse_file(filename, use_cpp=True)
     v = FuncCallVisitor(funcname)
     v.visit(ast)

examples/func_defs.py

@@ -22,11 +22,11 @@
 # A simple visitor for FuncDef nodes that prints the names and
 # locations of function definitions.
 class FuncDefVisitor(c_ast.NodeVisitor):
-    def visit_FuncDef(self, node):
+    def visit_FuncDef(self, node: c_ast.FuncDef) -> None:
         print("%s at %s" % (node.decl.name, node.decl.coord))
 
 
-def show_func_defs(filename):
+def show_func_defs(filename: str) -> None:
     # Note that cpp is used. Provide a path to your own cpp or
     # make sure one exists in PATH.
     ast = parse_file(filename, use_cpp=True, cpp_args=r"-Iutils/fake_libc_include")

examples/func_defs_add_param.py

@@ -23,7 +23,7 @@
 
 
 class ParamAdder(c_ast.NodeVisitor):
-    def visit_FuncDecl(self, node):
+    def visit_FuncDecl(self, node: c_ast.FuncDecl) -> None:
         ty = c_ast.TypeDecl(
             declname="_hidden", quals=[], align=[], type=c_ast.IdentifierType(["int"])
         )
@@ -38,7 +38,7 @@ def visit_FuncDecl(self, node):
             bitsize=None,
             coord=node.coord,
         )
-        if node.args:
+        if node.args is not None:
             node.args.params.append(newdecl)
         else:
             node.args = c_ast.ParamList(params=[newdecl])