def exception_to_error_frame(stream_id: int, exception: Exception) -> ErrorFrame:
frame = ErrorFrame()
frame.stream_id = stream_id
if isinstance(exception, RSocketProtocolError):
frame.error_code = exception.error_code
frame.data = ensure_bytes(exception.data)
else:
frame.error_code = ErrorCode.APPLICATION_ERROR
frame.data = str(exception).encode()
return frame
async def generator() -> AsyncGenerator[Tuple[Payload, bool], None]:
item_count = 10
for j in range(item_count):
await asyncio.sleep(1)
is_complete = j == item_count - 1
yield Payload(ensure_bytes('Feed Item: %s' % j)), is_complete
def generator() -> Generator[Tuple[Payload, bool], None, None]:
for i in range(3):
yield Payload(
ensure_bytes('some long data which should be fragmented %s' %
i)), i == 2
def ensure_encoding_name(encoding) -> bytes:
if isinstance(encoding, WellKnownMimeTypes):
return encoding.value.name
return ensure_bytes(encoding)
def __init__(self, token: Optional[Union[str, bytes]] = None):
self.token = ensure_bytes(token)
def __init__(self,
username: Optional[str] = None,
password: Optional[str] = None):
self.username = ensure_bytes(username)
self.password = ensure_bytes(password)