This commit is contained in:
Kovid Goyal 2022-01-02 09:38:14 +05:30
parent 0fcc446298
commit fe91b74ba7
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C

View File

@ -5,12 +5,12 @@ import json
import os import os
import re import re
import sys import sys
import types
from contextlib import suppress from contextlib import suppress
from functools import partial from functools import partial
from time import monotonic from time import monotonic
from types import GeneratorType
from typing import ( from typing import (
Any, Dict, Iterator, Iterable, List, Optional, Tuple, Union, cast Any, Dict, Iterable, Iterator, List, Optional, Tuple, Union, cast
) )
from .cli import emph, parse_args from .cli import emph, parse_args
@ -137,7 +137,7 @@ class RCIO(TTYIO):
def do_io(to: Optional[str], send: Dict[str, Any], no_response: bool, response_timeout: float) -> Dict[str, Any]: def do_io(to: Optional[str], send: Dict[str, Any], no_response: bool, response_timeout: float) -> Dict[str, Any]:
payload = send.get('payload') payload = send.get('payload')
if not isinstance(payload, types.GeneratorType): if not isinstance(payload, GeneratorType):
send_data: Union[bytes, Iterator[bytes]] = encode_send(send) send_data: Union[bytes, Iterator[bytes]] = encode_send(send)
else: else:
def send_generator() -> Iterator[bytes]: def send_generator() -> Iterator[bytes]: