CERBERUS/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty.git synced 2026-06-19 21:55:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Handle blocking io errors when writing all to an fd

Browse source
This commit is contained in:
Kovid Goyal 2021-11-17 11:45:15 +05:30
parent 18ed56b639
commit e68debc94e
No known key found for this signature in database
GPG key ID: 06BC317B515ACE7C
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
kitty/utils.py
View file

@ -430,11 +430,16 @@ def parse_address_spec(spec: str) -> Tuple[AddressFamily, Union[Tuple[str, int],
return family, address, socket_path
def write_all(fd: int, data: Union[str, bytes]) -> None:
def write_all(fd: int, data: Union[str, bytes], block_until_written: bool = True) -> None:
if isinstance(data, str):
data = data.encode('utf-8')
while data:
n = os.write(fd, data)
try:
n = os.write(fd, data)
except BlockingIOError:
if not block_until_written:
raise
continue
if not n:
break
data = data[n:]