Skip to content

Commit 7e27fc3

Browse files
committed
Please typechecking
1 parent 7ea39e3 commit 7e27fc3

File tree

1 file changed

+22
-12
lines changed
  • opentelemetry-sdk/src/opentelemetry/sdk/_configuration

1 file changed

+22
-12
lines changed

opentelemetry-sdk/src/opentelemetry/sdk/_configuration/__init__.py

Lines changed: 22 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,14 @@
109109
]
110110

111111

112+
class ExporterSpanProcessor(SpanProcessor):
113+
def __init__(self, span_exporter: SpanExporter, *args, **kwargs): ...
114+
115+
116+
class ExporterLogRecordProcessor(LogRecordProcessor):
117+
def __init__(self, exporter: LogRecordExporter, *args, **kwargs): ...
118+
119+
112120
def _import_config_components(
113121
selected_components: Sequence[str], entry_point_name: str
114122
) -> list[tuple[str, Type]]:
@@ -214,8 +222,8 @@ def _init_tracing(
214222
sampler: Sampler | None = None,
215223
resource: Resource | None = None,
216224
exporter_args_map: ExporterArgsMap | None = None,
217-
span_processors: Sequence[Type[SpanProcessor]] | None = None,
218-
export_processor: Type[SpanProcessor] | None = None,
225+
span_processors: Sequence[SpanProcessor] | None = None,
226+
export_span_processor: Type[ExporterSpanProcessor] | None = None,
219227
):
220228
provider = TracerProvider(
221229
id_generator=id_generator,
@@ -226,7 +234,7 @@ def _init_tracing(
226234

227235
exporter_args_map = exporter_args_map or {}
228236
span_processors = span_processors or []
229-
export_processor = export_processor or BatchSpanProcessor
237+
export_processor = export_span_processor or BatchSpanProcessor
230238
for _, exporter_class in exporters.items():
231239
exporter_args = exporter_args_map.get(exporter_class, {})
232240

@@ -268,16 +276,17 @@ def _init_logging(
268276
resource: Resource | None = None,
269277
setup_logging_handler: bool = True,
270278
exporter_args_map: ExporterArgsMap | None = None,
271-
log_record_processors: Sequence[Type[LogRecordProcessor]] | None = None,
272-
export_processor: Type[LogRecordProcessor] | None = None,
279+
log_record_processors: Sequence[LogRecordProcessor] | None = None,
280+
export_log_record_processor: Type[ExporterLogRecordProcessor]
281+
| None = None,
273282
):
274283
provider = LoggerProvider(resource=resource)
275284
set_logger_provider(provider)
276285

277286
exporter_args_map = exporter_args_map or {}
278287

279288
log_record_processors = log_record_processors or []
280-
export_processor = export_processor or BatchLogRecordProcessor
289+
export_processor = export_log_record_processor or BatchLogRecordProcessor
281290
for _, exporter_class in exporters.items():
282291
exporter_args = exporter_args_map.get(exporter_class, {})
283292

@@ -450,10 +459,11 @@ def _initialize_components(
450459
id_generator: IdGenerator | None = None,
451460
setup_logging_handler: bool | None = None,
452461
exporter_args_map: ExporterArgsMap | None = None,
453-
span_processors: Sequence[Type[SpanProcessor]] | None = None,
454-
trace_export_processor: Type[SpanProcessor] | None = None,
455-
log_record_processors: Sequence[Type[LogRecordProcessor]] | None = None,
456-
log_export_processor: Type[LogRecordProcessor] | None = None,
462+
span_processors: Sequence[SpanProcessor] | None = None,
463+
export_span_processor: Type[ExporterSpanProcessor] | None = None,
464+
log_record_processors: Sequence[LogRecordProcessor] | None = None,
465+
export_log_record_processor: Type[ExporterLogRecordProcessor]
466+
| None = None,
457467
):
458468
# pylint: disable=too-many-locals
459469
if trace_exporter_names is None:
@@ -491,7 +501,7 @@ def _initialize_components(
491501
resource=resource,
492502
exporter_args_map=exporter_args_map,
493503
span_processors=span_processors,
494-
export_processor=trace_export_processor,
504+
export_span_processor=export_span_processor,
495505
)
496506
_init_metrics(
497507
metric_exporters, resource, exporter_args_map=exporter_args_map
@@ -511,7 +521,7 @@ def _initialize_components(
511521
setup_logging_handler,
512522
exporter_args_map=exporter_args_map,
513523
log_record_processors=log_record_processors,
514-
export_processor=log_export_processor,
524+
export_log_record_processor=export_log_record_processor,
515525
)
516526

517527

0 commit comments

Comments
 (0)