mirror of
https://codeberg.org/privacy1st/subprocess-util
synced 2024-12-22 22:06:05 +01:00
53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
import socket
|
|
from pathlib import Path
|
|
from typing import IO, AnyStr
|
|
|
|
from unix_sock_input import accept_loop_until_command_received
|
|
|
|
|
|
def receive_inform(in_pipe: IO[AnyStr],
|
|
socket_file: Path,
|
|
chunk_file_tmpl: Path) -> None:
|
|
"""
|
|
:param in_pipe:
|
|
:param chunk_file_tmpl:
|
|
:param cmd:
|
|
:param socket_file: Create a UNIX socket and wait for messages.
|
|
:return:
|
|
"""
|
|
print(f'Listening on socket {socket_file}')
|
|
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
sock.bind(str(socket_file))
|
|
sock.listen(1)
|
|
|
|
ct = 1
|
|
commands = [b'OK\n', b'EOF\n']
|
|
while True:
|
|
command = accept_loop_until_command_received(sock, commands)
|
|
if command not in commands:
|
|
raise ValueError("Invalid state")
|
|
|
|
chunk_file = chunk_file_tmpl.parent.joinpath(f'{chunk_file_tmpl.name}.{ct}')
|
|
chunk = chunk_file.read_bytes()
|
|
in_pipe.write(chunk)
|
|
# in_pipe.flush() # TODO: is this required?
|
|
chunk_file.unlink(missing_ok=False)
|
|
|
|
if command == b'OK\n':
|
|
ct += 1
|
|
elif command == b'EOF\n':
|
|
break
|
|
else:
|
|
raise ValueError("Invalid state")
|
|
|
|
print(f'Closing socket {socket_file}')
|
|
sock.close()
|
|
socket_file.unlink(missing_ok=False)
|
|
|
|
in_pipe.flush()
|
|
|
|
# TODO: Has this any effect? On stdin probably yes!
|
|
in_pipe.close()
|