mirror of
https://github.com/GNS3/gns3-gui.git
synced 2026-05-17 00:46:01 +03:00
106 lines
4.3 KiB
Python
106 lines
4.3 KiB
Python
#!/usr/bin/env python
|
|
#
|
|
# Copyright (C) 2015 GNS3 Technologies Inc.
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import shlex
|
|
import subprocess
|
|
import sys
|
|
|
|
from gns3.qt import QtWidgets
|
|
from gns3.utils.progress_dialog import ProgressDialog
|
|
from gns3.utils.wait_for_command_worker import WaitForCommandWorker
|
|
from gns3.utils.wait_for_runas_worker import WaitForRunAsWorker
|
|
|
|
import logging
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def sudo(*commands, parent=None, shell=False):
|
|
"""
|
|
Run a command as an administrator.
|
|
"""
|
|
|
|
if parent is None:
|
|
from ..main_window import MainWindow
|
|
parent = MainWindow.instance()
|
|
|
|
while True:
|
|
if sys.platform.startswith("win32"):
|
|
for command in commands:
|
|
worker = WaitForRunAsWorker(command)
|
|
progress_dialog = ProgressDialog(worker, "Run as administrator", "Executing command...", "Cancel", busy=True, parent=parent)
|
|
progress_dialog.show()
|
|
if not progress_dialog.exec():
|
|
return False
|
|
elif sys.platform.startswith("darwin"):
|
|
cmd = []
|
|
for command in commands:
|
|
command = [shlex.quote(c) for c in command]
|
|
cmd.append(' '.join(command))
|
|
command = ["osascript", "-e", "do shell script \"{}\" with administrator privileges".format(' && '.join(cmd))]
|
|
log.info("Execute as admin: {}".format(' '.join(command)))
|
|
worker = WaitForCommandWorker(command, shell=shell)
|
|
|
|
for line in worker.output().decode("utf-8", errors="ignore").splitlines():
|
|
log.info(line)
|
|
progress_dialog = ProgressDialog(worker, "Run as administrator", "Executing command...", "Cancel", busy=True, parent=parent)
|
|
progress_dialog.show()
|
|
if not progress_dialog.exec():
|
|
return False
|
|
|
|
else:
|
|
command_detail = []
|
|
for command in commands:
|
|
command = [shlex.quote(c) for c in command]
|
|
command_detail.append(' '.join(command))
|
|
|
|
password, ok = QtWidgets.QInputDialog.getText(parent,
|
|
"Run as administrator",
|
|
"Please enter your password to proceed.\nCommand:\n{}".format('\n'.join(command_detail)),
|
|
QtWidgets.QLineEdit.EchoMode.Password, "")
|
|
if not ok:
|
|
return False
|
|
|
|
try:
|
|
# check the password is valid
|
|
subprocess.check_output(["sudo", "-S", "id"], input=(password + "\n").encode(), timeout=30)
|
|
except subprocess.CalledProcessError:
|
|
continue
|
|
except (OSError, subprocess.SubprocessError) as e:
|
|
QtWidgets.QMessageBox.critical(parent, "Run as administrator", "Could not execute sudo: {}".format(e))
|
|
return False
|
|
|
|
# sudo shouldn't need the password again.
|
|
for command in commands:
|
|
waited_command = ["sudo"]
|
|
waited_command.extend(command)
|
|
worker = WaitForCommandWorker(waited_command, shell=shell)
|
|
|
|
for line in worker.output().decode("utf-8", errors="ignore").splitlines():
|
|
log.info(line)
|
|
progress_dialog = ProgressDialog(worker, "Run as administrator", "Executing command...", "Cancel", busy=True, parent=parent)
|
|
progress_dialog.show()
|
|
if not progress_dialog.exec():
|
|
return False
|
|
return True
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
app = QtWidgets.QApplication(sys.argv)
|
|
main = QtWidgets.QMainWindow()
|
|
sudo(["touch", "/tmp/a"], ["touch", "/tmp/b"], parent=main)
|