mirror of
https://github.com/spalencsar/bearhub.git
synced 2026-07-07 01:14:15 +02:00
52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
import io
|
|
import os
|
|
import subprocess
|
|
|
|
from PyQt5.QtGui import QIcon
|
|
from PyQt5.QtWidgets import QInputDialog, QLineEdit
|
|
|
|
from bauh import ROOT_DIR
|
|
from bauh.api.abstract.view import MessageType
|
|
from bauh.commons import resource
|
|
from bauh.view.qt.dialog import show_message
|
|
|
|
|
|
def is_root():
|
|
return os.getuid() == 0
|
|
|
|
|
|
def ask_root_password(locale_keys: dict):
|
|
|
|
diag = QInputDialog()
|
|
diag.setInputMode(QInputDialog.TextInput)
|
|
diag.setTextEchoMode(QLineEdit.Password)
|
|
diag.setWindowIcon(QIcon(resource.get_path('img/lock.svg', ROOT_DIR)))
|
|
diag.setWindowTitle(locale_keys['popup.root.title'])
|
|
diag.setLabelText(locale_keys['popup.root.password'] + ':')
|
|
diag.setCancelButtonText(locale_keys['popup.button.cancel'])
|
|
diag.resize(400, 200)
|
|
|
|
ok = diag.exec_()
|
|
|
|
if ok:
|
|
if not validate_password(diag.textValue()):
|
|
show_message(title=locale_keys['popup.root.bad_password.title'],
|
|
body=locale_keys['popup.root.bad_password.body'],
|
|
type_=MessageType.ERROR)
|
|
ok = False
|
|
|
|
return diag.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())
|