summaryrefslogtreecommitdiff
path: root/bitsandbytes/utils.py
blob: 4256a8742dd8c0272aeff1d135264c613951b942 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import shlex
import subprocess
import sys
from typing import Tuple


def execute_and_return(command_string: str) -> Tuple[str, str]:
    def _decode(subprocess_err_out_tuple):
        return tuple(
            to_decode.decode("UTF-8").strip()
            for to_decode in subprocess_err_out_tuple
        )

    def execute_and_return_decoded_std_streams(command_string):
        return _decode(
            subprocess.Popen(
                shlex.split(command_string),
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
            ).communicate()
        )

    std_out, std_err = execute_and_return_decoded_std_streams(command_string)
    return std_out, std_err


def print_stderr(s: str) -> None:
    print(s, file=sys.stderr)


def warn_of_missing_prerequisite(s: str) -> None:
    print_stderr("WARNING, missing pre-requisite: " + s)