mirror of
https://github.com/spalencsar/bearhub.git
synced 2026-07-06 22:54:16 +02:00
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
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())
|