.config/coc/extensions/node_modules/coc-pyright/node_modules/pyright/dist/typeshed-fallback/stubs/aws-xray-sdk/aws_xray_sdk/core/models/subsegment.pyi

   1 import time
   2 from typing import Any
   3
   4 from ...core import AWSXRayRecorder
   5 from ..exceptions.exceptions import SegmentNotFoundException as SegmentNotFoundException
   6 from .entity import Entity as Entity
   7 from .segment import Segment
   8
   9 SUBSEGMENT_RECORDING_ATTRIBUTE: str
  10
  11 def set_as_recording(decorated_func, wrapped) -> None: ...
  12 def is_already_recording(func): ...
  13 def subsegment_decorator(wrapped, instance, args, kwargs): ...
  14
  15 class SubsegmentContextManager:
  16     name: str | None
  17     subsegment_kwargs: dict[str, Any] | None
  18     recorder: AWSXRayRecorder
  19     subsegment: Subsegment
  20     def __init__(self, recorder: AWSXRayRecorder, name: Any | None = ..., **subsegment_kwargs) -> None: ...
  21     def __call__(self, wrapped, instance, args: list[Any], kwargs: dict[str, Any]): ...
  22     def __enter__(self) -> Subsegment: ...
  23     def __exit__(self, exc_type, exc_val, exc_tb) -> None: ...
  24
  25 class Subsegment(Entity):
  26     parent_segment: Segment
  27     trace_id: str
  28     type: str
  29     namespace: str
  30     sql: dict[str, Any]
  31     def __init__(self, name: str, namespace: str, segment: Segment) -> None: ...
  32     def add_subsegment(self, subsegment: Subsegment) -> None: ...
  33     def remove_subsegment(self, subsegment: Subsegment) -> None: ...
  34     def close(self, end_time: time.struct_time | None = ...) -> None: ...
  35     def set_sql(self, sql: dict[str, Any]) -> None: ...
  36     def to_dict(self) -> dict[str, Any]: ...