Add __init__ to maps to improve imports

This commit is contained in:
Pragyansh Chaturvedi
2025-09-30 19:51:28 +05:30
parent 744aa3fbdf
commit d943b78a25
5 changed files with 7 additions and 5 deletions

View File

@ -1,7 +1,7 @@
from pythonbpf.decorators import bpf, map, section, bpfglobal
from ctypes import c_void_p, c_int64, c_int32, c_uint64
from pythonbpf.helpers import ktime
from pythonbpf.maps.maps import HashMap
from pythonbpf.maps import HashMap
@bpf

View File

@ -1,6 +1,6 @@
from pythonbpf import bpf, map, section, bpfglobal, compile
from pythonbpf.helpers import ktime, deref
from pythonbpf.maps.maps import HashMap
from pythonbpf.maps import HashMap
from ctypes import c_void_p, c_int64, c_int32, c_uint64

View File

@ -2,7 +2,7 @@ import ast
from llvmlite import ir
from .license_pass import license_processing
from .functions_pass import func_proc
from pythonbpf.maps.maps_pass import maps_proc
from pythonbpf.maps import maps_proc
from .structs.structs_pass import structs_proc
from .globals_pass import globals_processing
import os

View File

@ -0,0 +1,2 @@
from .maps import HashMap, PerfEventArray
from .maps_pass import maps_proc

View File

@ -256,8 +256,8 @@ def process_bpf_map(func_node, module):
if handler:
return handler(map_name, rval, module)
else:
logger.warning(f"Unknown map type {
rval.func.id}, defaulting to HashMap")
logger.warning(f"Unknown map type "
f"{rval.func.id}, defaulting to HashMap")
process_hash_map(map_name, rval, module)
else:
raise ValueError("Function under @map must return a map")