supporting snaps

fpakman/core/system.py

@@ -2,9 +2,18 @@ import os
 import subprocess
 from typing import List
 
+from fpakman import __app_name__
 from fpakman.core import resource
 
 
+class FpakmanProcess:
+
+    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,
@@ -28,4 +37,15 @@ def cmd_to_subprocess(cmd: List[str]):
 
 
 def notify_user(msg: str, icon_path: str = resource.get_path('img/logo.svg')):
-    os.system("notify-send {} '{}'".format("-i {}".format(icon_path) if icon_path else '', msg))
+    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)