From 8dd2746411b01a3d7876726b4862ddf913f0712a Mon Sep 17 00:00:00 2001 From: Pragyansh Chaturvedi Date: Wed, 1 Oct 2025 18:21:42 +0530 Subject: [PATCH] rename get_key_ptr to get_or_create_ptr_from_arg --- pythonbpf/helper/helper_utils.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/pythonbpf/helper/helper_utils.py b/pythonbpf/helper/helper_utils.py index 4b7cbbd..f3a89eb 100644 --- a/pythonbpf/helper/helper_utils.py +++ b/pythonbpf/helper/helper_utils.py @@ -37,14 +37,14 @@ def create_int_constant_ptr(value, builder, int_width=64): return ptr -def get_key_ptr(key_arg, builder, local_sym_tab): +def get_or_create_ptr_from_arg(arg, builder, local_sym_tab): """Extract key pointer from the call arguments.""" - if isinstance(key_arg, ast.Name): - key_ptr = get_var_ptr_from_name(key_arg.id, local_sym_tab) - elif isinstance(key_arg, ast.Constant) and isinstance(key_arg.value, int): - key_ptr = create_int_constant_ptr(key_arg.value, builder) + if isinstance(arg, ast.Name): + ptr = get_var_ptr_from_name(arg.id, local_sym_tab) + elif isinstance(arg, ast.Constant) and isinstance(arg.value, int): + ptr = create_int_constant_ptr(arg.value, builder) else: raise NotImplementedError( "Only simple variable names are supported as keys in map lookup.") - return key_ptr + return ptr