selfprivacy-rest-api/selfprivacy_api/utils/root_interface.py

from typing import List

# from subprocess import check_output
from selfprivacy_api.root_daemon import SOCKET_PATH, socket_module
from tests.test_common import get_test_mode


def call_root_function(cmd: List[str]) -> str:
    if get_test_mode():
        return "done"
    else:
        return _call_root_daemon(cmd)


def _call_root_daemon(cmd: List[str]) -> str:
    return _write_to_daemon_socket(cmd)


def _write_to_daemon_socket(cmd: List[str]) -> str:
    sock = socket_module.socket(socket_module.AF_UNIX, socket_module.SOCK_STREAM)
    sock.connect(SOCKET_PATH)
    sock.send(" ".join(cmd).encode("utf-8") + b"\n")
    pipe = sock.makefile("rw")
    line = pipe.readline()
    return line