varun-r-mallya/python-bpf
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/Python-BPF.git synced 2025-12-31 21:06:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add iter to LocalSymbol

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-10-02 04:55:37 +05:30
parent 12ba3605e9
commit 71b97e3e20
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pythonbpf/functions_pass.py
View File

@ -18,6 +18,11 @@ class LocalSymbol:
ir_type: ir.Type ir_type: ir.Type
metadata: Any = None metadata: Any = None
def __iter__(self):
yield self.var
yield self.ir_type
yield self.metadata
def get_probe_string(func_node): def get_probe_string(func_node):
"""Extract the probe string from the decorator of the function node.""" """Extract the probe string from the decorator of the function node."""
@ -669,8 +674,7 @@ def infer_return_type(func_node: ast.FunctionDef):
if found_type is None: if found_type is None:
found_type = t found_type = t
elif found_type != t: elif found_type != t:
raise ValueError(f"Conflicting return types:{ raise ValueError(f"Conflicting return types:{found_type} vs {t}")
found_type} vs {t}")
return found_type or "None" return found_type or "None"