Make linters happy

src/pgtracer/ebpf/dwarf.py

@@ -496,7 +496,7 @@ class Struct:
     fields_defs: Dict[str, StructMemberDefinition]
     metadata: ProcessMetadata
     die: DIE
-    buffer_type: Type
+    buffer_type: Type[ct._CData]
     _fully_loaded: bool = False
     size: int
 
@@ -634,15 +634,6 @@ def as_dict(self, include_all: bool = False) -> Dict[str, Any]:
                 value.as_dict(include_all=True)
         return values
 
-    @classmethod
-    def size(cls) -> int:
-        """
-        Returns the sizeof() this struct as stored in the DWARF information.
-        """
-        size = cls.die.attributes["DW_AT_byte_size"].value
-        assert isinstance(size, int)
-        return size
-
     @classmethod
     def pointer_type(cls) -> Type[DWARFPointer]:
         """

src/pgtracer/model/plan.py

@@ -6,14 +6,13 @@
 import ctypes as ct
 from typing import TYPE_CHECKING, Dict, Optional
 
+from ..ebpf.collector.c_defs import plan_data, planstate_data
+from ..ebpf.dwarf import ProcessMetadata, Struct
 from ..utils import timespec_to_float
 
 if TYPE_CHECKING:
     from enum import IntEnum
 
-from ..ebpf.collector.c_defs import plan_data, planstate_data
-from ..ebpf.dwarf import ProcessMetadata, Struct
-
 
 def explain_dict_to_str(parts: Dict[str, str]) -> str:
     """

src/pgtracer/utils.py

@@ -7,17 +7,19 @@
 import re
 import subprocess
 from datetime import timedelta
-from typing import TYPE_CHECKING, BinaryIO, Optional
+from typing import TYPE_CHECKING, BinaryIO, Optional, Union
 
 from pypsutil import Process
 
+from pgtracer.ebpf.dwarf import Struct
+
 if TYPE_CHECKING:
     from ctypes import _CData
 else:
     _CData = object
 
 
-def timespec_to_timedelta(timespec: _CData) -> timedelta:
+def timespec_to_timedelta(timespec: Union[_CData, Struct]) -> timedelta:
     """
     Convert a timespec_t or instr_time struct to a timedelta.
     """