merge 'staging'

bauh/view/qt/root.py

@@ -0,0 +1,45 @@
+import io
+import os
+import subprocess
+
+from PyQt5.QtWidgets import QInputDialog, QLineEdit
+
+from bauh.view.qt.dialog import show_error
+
+
+def is_root():
+    return os.getuid() == 0
+
+
+def ask_root_password(locale_keys: dict):
+
+    dialog_pwd = QInputDialog()
+    dialog_pwd.setInputMode(QInputDialog.TextInput)
+    dialog_pwd.setTextEchoMode(QLineEdit.Password)
+    dialog_pwd.setWindowTitle(locale_keys['popup.root.title'])
+    dialog_pwd.setLabelText(locale_keys['popup.root.password'] + ':')
+    dialog_pwd.setCancelButtonText(locale_keys['popup.button.cancel'])
+    dialog_pwd.resize(400, 200)
+
+    ok = dialog_pwd.exec_()
+
+    if ok:
+        if not validate_password(dialog_pwd.textValue()):
+            show_error(title=locale_keys['popup.root.bad_password.title'],
+                       body=locale_keys['popup.root.bad_password.body'])
+            ok = False
+
+    return dialog_pwd.textValue(), ok
+
+
+def validate_password(password: str) -> bool:
+    proc = subprocess.Popen('sudo -k && echo {} | sudo -S whoami'.format(password),
+                            shell=True,
+                            stdout=subprocess.PIPE,
+                            stderr=subprocess.DEVNULL,
+                            bufsize=-1)
+    stream = os._wrap_close(io.TextIOWrapper(proc.stdout), proc)
+    res = stream.read()
+    stream.close()
+
+    return bool(res.strip())