project renamed as 'bauh'

bauh/core/system.py

@@ -0,0 +1,51 @@
+import os
+import subprocess
+from typing import List
+
+from bauh import __app_name__
+from bauh.core import resource
+
+
+class BauhProcess:
+
+    def __init__(self, subproc: subprocess.Popen, success_phrase: str = None, wrong_error_phrase: str = '[sudo] password for'):
+        self.subproc = subproc
+        self.success_pgrase = success_phrase
+        self.wrong_error_phrase = wrong_error_phrase
+
+
+def run_cmd(cmd: str, expected_code: int = 0, ignore_return_code: bool = False, print_error: bool = True) -> str:
+    args = {
+        "shell": True,
+        "stdout": subprocess.PIPE,
+        "env": {'LANG': 'en'}
+    }
+
+    if not print_error:
+        args["stderr"] = subprocess.DEVNULL
+
+    res = subprocess.run(cmd, **args)
+    return res.stdout.decode() if ignore_return_code or res.returncode == expected_code else None
+
+
+def stream_cmd(cmd: List[str]):
+    return cmd_to_subprocess(cmd).stdout
+
+
+def cmd_to_subprocess(cmd: List[str]):
+    return subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env={'LANG': 'en'})
+
+
+def notify_user(msg: str, icon_path: str = resource.get_path('img/logo.svg')):
+    os.system("notify-send -a {} {} '{}'".format(__app_name__, "-i {}".format(icon_path) if icon_path else '', msg))
+
+
+def cmd_as_root(cmd: List[str], root_password: str) -> subprocess.Popen:
+    pwdin, final_cmd = None, []
+
+    if root_password is not None:
+        final_cmd.extend(['sudo', '-S'])
+        pwdin = stream_cmd(['echo', root_password])
+
+    final_cmd.extend(cmd)
+    return subprocess.Popen(final_cmd, stdin=pwdin, stdout=subprocess.PIPE, stderr=subprocess.PIPE)