Class WhateverIO

celery/utils/text.py:189–198

StringIO that takes bytes or str.

class WhateverIO(StringIO):
    """StringIO that takes bytes or str."""

    def __init__(
            self, v: bytes | str | None = None, *a: Any, **kw: Any) -> None:
        _SIO_init(self, v.decode() if isinstance(v, bytes) else v, *a, **kw)

    def write(self, data: bytes | str) -> int:
        return _SIO_write(self, data.decode()
                          if isinstance(data, bytes) else data)