Function _wrap_encryption_errors

pymongo/asynchronous/encryption.py:127–136

Context manager to wrap encryption related errors.

()

@contextlib.contextmanager
def _wrap_encryption_errors() -> Iterator[None]:
    """Context manager to wrap encryption related errors."""
    try:
        yield
    except BSONError:
        # BSON encoding/decoding errors are unrelated to encryption so
        # we should propagate them unchanged.
        raise
    except Exception as exc:
        raise EncryptionError(exc) from exc


class _EncryptionIO(AsyncMongoCryptCallback):  # type: ignore[misc]