Context manager that yeilds a requests session and starts a span that already has the headers injected with the span's context so that this is passed into the request for tracing

make_traced_request.py

import requests

from typing import Generator
from contextlib import contextmanager
from dataclasses import dataclass

@dataclass
class Context:
    span: opentracing.Span
    session: requests.Session

@contextmanager
def make_traced_request(url: str) -> Generator[Context]:
    """
    Yeilds a requests session and starts a span that already has the
    headers injected with the span's context so that this is passed
    into the request for tracing
    """
    with opentracing.tracer.start_active_span(url) as scope:
        session = requests.session()
        headers = {}
        opentracing.tracer.inject(
            scope.span,
            opentracing.Format.HTTP_HEADERS,
            headers
        )
        session.headers.update(headers)

        yield Context(scope.span, session)