Hans P. Möller
5 years ago
parent
2fc04fbdbc
commit
c0ec597d94
@ -1,355 +0,0 @@
|
|||||||
#!/usr/bin/python3
|
|
||||||
# deppend on
|
|
||||||
# aptdaemon
|
|
||||||
# debconf-kde-helper
|
|
||||||
import sys
|
|
||||||
import os
|
|
||||||
from PyQt5.QtWidgets import (QWidget, QApplication, QLabel, QPushButton,
|
|
||||||
QHBoxLayout, QVBoxLayout, QProgressBar,
|
|
||||||
QPlainTextEdit, QMessageBox)
|
|
||||||
from PyQt5.QtCore import Qt
|
|
||||||
from PyQt5.QtGui import QIcon, QTextCursor, QPalette
|
|
||||||
from optparse import OptionParser
|
|
||||||
from aptdaemon import client
|
|
||||||
from aptdaemon.errors import NotAuthorizedError, TransactionFailed
|
|
||||||
from aptdaemon.enums import (EXIT_SUCCESS,
|
|
||||||
EXIT_FAILED,
|
|
||||||
STATUS_COMMITTING,
|
|
||||||
get_error_description_from_enum,
|
|
||||||
get_error_string_from_enum,
|
|
||||||
get_status_string_from_enum)
|
|
||||||
from pathlib import Path
|
|
||||||
|
|
||||||
|
|
||||||
class DialogUpg(QWidget):
|
|
||||||
def __init__(self, options=None):
|
|
||||||
QWidget.__init__(self)
|
|
||||||
|
|
||||||
self.initUI()
|
|
||||||
self.closeBtn.clicked.connect(self.call_reject)
|
|
||||||
self.apt_client = client.AptClient()
|
|
||||||
self.downloadText = ""
|
|
||||||
self.detailText = ""
|
|
||||||
self.old_short_desc = ""
|
|
||||||
self.details = ""
|
|
||||||
self.status = ""
|
|
||||||
self.errors = []
|
|
||||||
# TODO make a terminal work to see more info
|
|
||||||
# self.master, self.slave = pty.openpty()
|
|
||||||
'''proc = subprocess.Popen(['qterminal'],
|
|
||||||
stdin=self.slave,
|
|
||||||
#stdout=subprocess.PIPE,
|
|
||||||
stdout=self.slave,
|
|
||||||
#stderr=subprocess.PIPE
|
|
||||||
stderr=self.slave)'''
|
|
||||||
|
|
||||||
if options.fullUpgrade:
|
|
||||||
self.trans2 = self.apt_client.upgrade_system(safe_mode=False)
|
|
||||||
self.setWindowTitle('Full Upgrade')
|
|
||||||
else:
|
|
||||||
self.trans2 = self.apt_client.upgrade_system(safe_mode=True)
|
|
||||||
|
|
||||||
if os.geteuid() == 0:
|
|
||||||
if options.cacheUpdate:
|
|
||||||
self.trans1 = self.apt_client.update_cache()
|
|
||||||
self.update_cache()
|
|
||||||
else:
|
|
||||||
self.upgrade()
|
|
||||||
|
|
||||||
def initUI(self):
|
|
||||||
self.label = QLabel()
|
|
||||||
self.label.setAlignment(Qt.AlignHCenter)
|
|
||||||
self.closeBtn = QPushButton("Close")
|
|
||||||
self.progressBar = QProgressBar()
|
|
||||||
self.plainTextEdit = QPlainTextEdit()
|
|
||||||
palette = self.plainTextEdit.palette()
|
|
||||||
palette.setColor(QPalette.Base, Qt.black)
|
|
||||||
palette.setColor(QPalette.Text, Qt.gray)
|
|
||||||
self.plainTextEdit.setPalette(palette)
|
|
||||||
|
|
||||||
hbox = QHBoxLayout()
|
|
||||||
hbox.addStretch(1)
|
|
||||||
hbox.addWidget(self.closeBtn)
|
|
||||||
hbox.addStretch(1)
|
|
||||||
|
|
||||||
vbox = QVBoxLayout()
|
|
||||||
vbox.addWidget(self.label)
|
|
||||||
vbox.addWidget(self.progressBar)
|
|
||||||
vbox.addWidget(self.plainTextEdit)
|
|
||||||
vbox.addLayout(hbox)
|
|
||||||
|
|
||||||
self.setLayout(vbox)
|
|
||||||
self.setGeometry(300, 300, 500, 150)
|
|
||||||
self.setWindowTitle('Upgrade')
|
|
||||||
self.progressBar.setVisible(False)
|
|
||||||
self.plainTextEdit.setReadOnly(True)
|
|
||||||
self.plainTextEdit.setVisible(False)
|
|
||||||
self.center()
|
|
||||||
|
|
||||||
def center(self):
|
|
||||||
frameGm = self.frameGeometry()
|
|
||||||
screen = QApplication.desktop().screenNumber(
|
|
||||||
QApplication.desktop().cursor().pos())
|
|
||||||
centerPoint = QApplication.desktop().screenGeometry(screen).center()
|
|
||||||
frameGm.moveCenter(centerPoint)
|
|
||||||
self.move(frameGm.topLeft())
|
|
||||||
|
|
||||||
def upgrade_progress(self, transaction, progress):
|
|
||||||
self.progressBar.setVisible(True)
|
|
||||||
self.progressBar.setValue(progress)
|
|
||||||
|
|
||||||
def update_progress(self, transaction, progress):
|
|
||||||
self.progressBar.setVisible(True)
|
|
||||||
self.progressBar.setValue(progress)
|
|
||||||
self.label.setText("Updating cache...")
|
|
||||||
|
|
||||||
def update_progress_download(self, transaction, uri, status, short_desc,
|
|
||||||
total_size, current_size, msg):
|
|
||||||
self.plainTextEdit.setVisible(True)
|
|
||||||
if self.old_short_desc == short_desc:
|
|
||||||
# if it's the same file we update the line, don't append new line
|
|
||||||
self.plainTextEdit.moveCursor(QTextCursor.End)
|
|
||||||
cursor = self.plainTextEdit.textCursor()
|
|
||||||
cursor.movePosition(QTextCursor.End, QTextCursor.MoveAnchor)
|
|
||||||
cursor.select(QTextCursor.LineUnderCursor)
|
|
||||||
cursor.removeSelectedText()
|
|
||||||
self.plainTextEdit.insertPlainText(str(current_size) + "/"
|
|
||||||
+ str(total_size) + " " + msg)
|
|
||||||
cursor.movePosition(QTextCursor.End, QTextCursor.MoveAnchor)
|
|
||||||
else:
|
|
||||||
self.plainTextEdit.moveCursor(QTextCursor.End)
|
|
||||||
self.plainTextEdit.appendPlainText(status + " " + short_desc +
|
|
||||||
"\n")
|
|
||||||
self.plainTextEdit.insertPlainText(str(current_size) + "/"
|
|
||||||
+ str(total_size) + " " + msg)
|
|
||||||
self.plainTextEdit.moveCursor(QTextCursor.End)
|
|
||||||
self.old_short_desc = short_desc
|
|
||||||
|
|
||||||
def upgrade_progress_download(self, transaction, uri, status, short_desc,
|
|
||||||
total_size, current_size, msg):
|
|
||||||
self.plainTextEdit.setVisible(True)
|
|
||||||
if self.status == "status-downloading":
|
|
||||||
# TODO it prints the last line after installation is complete.
|
|
||||||
if self.old_short_desc == short_desc:
|
|
||||||
# if it's the same file we update the line, don't append
|
|
||||||
self.plainTextEdit.moveCursor(QTextCursor.End)
|
|
||||||
cursor = self.plainTextEdit.textCursor()
|
|
||||||
cursor.movePosition(QTextCursor.End, QTextCursor.MoveAnchor)
|
|
||||||
cursor.select(QTextCursor.LineUnderCursor)
|
|
||||||
cursor.removeSelectedText()
|
|
||||||
self.plainTextEdit.insertPlainText(str(current_size) + "/"
|
|
||||||
+ str(total_size) + " "
|
|
||||||
+ msg)
|
|
||||||
cursor.movePosition(QTextCursor.End, QTextCursor.MoveAnchor)
|
|
||||||
else:
|
|
||||||
self.plainTextEdit.moveCursor(QTextCursor.End)
|
|
||||||
self.plainTextEdit.appendPlainText(status + " " + short_desc
|
|
||||||
+ "\n")
|
|
||||||
self.plainTextEdit.insertPlainText(str(current_size) + "/"
|
|
||||||
+ str(total_size) + " "
|
|
||||||
+ msg)
|
|
||||||
self.plainTextEdit.moveCursor(QTextCursor.End)
|
|
||||||
self.old_short_desc = short_desc
|
|
||||||
|
|
||||||
def update_progress_detail(self, transaction, current_items, total_items,
|
|
||||||
current_bytes, total_bytes, current_cps, eta):
|
|
||||||
if total_items > 0:
|
|
||||||
self.plainTextEdit.setVisible(True)
|
|
||||||
if self.detailText != "Fetching " + str(
|
|
||||||
current_items) + " of " + str(total_items):
|
|
||||||
self.detailText = (
|
|
||||||
"Fetching " + str(current_items) + " of "
|
|
||||||
+ str(total_items)
|
|
||||||
)
|
|
||||||
self.label.setText(self.detailText + "\n" + self.downloadText)
|
|
||||||
|
|
||||||
def upgrade_progress_detail(self, transaction, current_items, total_items,
|
|
||||||
current_bytes, total_bytes, current_cps, eta):
|
|
||||||
if total_items > 0:
|
|
||||||
self.plainTextEdit.setVisible(True)
|
|
||||||
if self.detailText != "Downloaded " + str(
|
|
||||||
current_items) + " of " + str(total_items):
|
|
||||||
self.detailText = (
|
|
||||||
"Downloaded " + str(current_items) + " of "
|
|
||||||
+ str(total_items)
|
|
||||||
)
|
|
||||||
self.label.setText(self.detailText + "\n" + self.downloadText)
|
|
||||||
|
|
||||||
def upgrade_finish(self, transaction, exit_state):
|
|
||||||
if exit_state == EXIT_FAILED:
|
|
||||||
error_string = get_error_string_from_enum(transaction.error.code)
|
|
||||||
error_desc = get_error_description_from_enum(
|
|
||||||
transaction.error.code)
|
|
||||||
|
|
||||||
text = "Upgrade finished"
|
|
||||||
|
|
||||||
reboot_required_path = Path("/var/run/reboot-required")
|
|
||||||
if reboot_required_path.exists():
|
|
||||||
text = text + "\n" + "Restart required"
|
|
||||||
self.progressBar.setVisible(False)
|
|
||||||
|
|
||||||
if(len(self.errors) > 0):
|
|
||||||
text = text + "\n With some Errors"
|
|
||||||
self.plainTextEdit.appendPlainText("Error Resume:\n")
|
|
||||||
for error in self.errors:
|
|
||||||
self.plainTextEdit.setEnabled(False)
|
|
||||||
self.plainTextEdit.insertPlainText(error + "\n")
|
|
||||||
self.plainTextEdit.insertPlainText(error_string + "\n")
|
|
||||||
self.plainTextEdit.insertPlainText(error_desc + "\n")
|
|
||||||
self.plainTextEdit.moveCursor(QTextCursor.End)
|
|
||||||
|
|
||||||
self.label.setText(text)
|
|
||||||
self.closeBtn.setVisible(True)
|
|
||||||
self.closeBtn.setEnabled(True)
|
|
||||||
self.plainTextEdit.setEnabled(True)
|
|
||||||
|
|
||||||
def upgrade_error(self, transaction, error_code, error_details):
|
|
||||||
self.plainTextEdit.setVisible(True)
|
|
||||||
self.errors.append("Eror Code: " + str(error_code))
|
|
||||||
self.errors.append("Error Detail: " + error_details)
|
|
||||||
self.plainTextEdit.setVisible(True)
|
|
||||||
self.closeBtn.setEnabled(True)
|
|
||||||
print("ECode: " + str(error_code) + "\n")
|
|
||||||
print("EDetail: " + error_details + "\n")
|
|
||||||
|
|
||||||
def upgrade_cancellable_changed(self, transaction, cancellable):
|
|
||||||
self.closeBtn.setEnabled(cancellable)
|
|
||||||
|
|
||||||
def update_cache(self):
|
|
||||||
self.closeBtn.setVisible(False)
|
|
||||||
try:
|
|
||||||
self.trans1.connect('finished', self.update_finish)
|
|
||||||
|
|
||||||
self.trans1.connect('progress-changed', self.update_progress)
|
|
||||||
self.trans1.connect('progress-details-changed',
|
|
||||||
self.update_progress_detail)
|
|
||||||
self.trans1.connect('progress-download-changed',
|
|
||||||
self.update_progress_download)
|
|
||||||
self.trans1.connect('error', self.upgrade_error)
|
|
||||||
self.trans1.connect("status-changed", self.status_changed)
|
|
||||||
# TODO make a terminal work to see more info
|
|
||||||
# self.trans1.set_terminal(os.ttyname(self.slave))
|
|
||||||
|
|
||||||
self.trans1.run()
|
|
||||||
|
|
||||||
except (NotAuthorizedError, TransactionFailed) as e:
|
|
||||||
print("Warning: install transaction not completed successfully:"
|
|
||||||
+ "{}".format(e))
|
|
||||||
|
|
||||||
def update_finish(self, transaction, exit_state):
|
|
||||||
self.label.setText("Update Cache Finished")
|
|
||||||
if exit_state == EXIT_FAILED:
|
|
||||||
error_string = get_error_string_from_enum(transaction.error.code)
|
|
||||||
error_desc = get_error_description_from_enum(
|
|
||||||
transaction.error.code)
|
|
||||||
self.plainTextEdit.setEnabled(False)
|
|
||||||
self.plainTextEdit.moveCursor(QTextCursor.End)
|
|
||||||
self.plainTextEdit.insertPlainText(error_string + "\n")
|
|
||||||
self.plainTextEdit.insertPlainText(error_desc + "\n")
|
|
||||||
self.plainTextEdit.moveCursor(QTextCursor.End)
|
|
||||||
self.plainTextEdit.setEnabled(True)
|
|
||||||
|
|
||||||
self.upgrade()
|
|
||||||
|
|
||||||
def status_changed(self, transaction, status):
|
|
||||||
self.status = status
|
|
||||||
self.label.setText("Status:" + get_status_string_from_enum(status))
|
|
||||||
print("Status:" + get_status_string_from_enum(status) + " " + status
|
|
||||||
+ "\n")
|
|
||||||
|
|
||||||
def status_details_changed(self, transaction, details):
|
|
||||||
self.plainTextEdit.setVisible(True)
|
|
||||||
if self.details != details:
|
|
||||||
self.details = details
|
|
||||||
|
|
||||||
if self.status != "status-downloading":
|
|
||||||
'''
|
|
||||||
if "Downloading xxxxx" is handled by
|
|
||||||
upgrade_progress_download" in short_desc
|
|
||||||
'''
|
|
||||||
self.plainTextEdit.appendPlainText(details)
|
|
||||||
self.plainTextEdit.moveCursor(QTextCursor.End)
|
|
||||||
self.label.setText(details)
|
|
||||||
else:
|
|
||||||
self.label.setText(self.detailText + "\n" + details)
|
|
||||||
# if is downloading put the "Downloaded x of y" text
|
|
||||||
# print("PTY:" + str(self.slave))
|
|
||||||
print("Status Details:" + details)
|
|
||||||
|
|
||||||
def upgrade(self):
|
|
||||||
try:
|
|
||||||
self.trans2.connect('progress-changed', self.upgrade_progress)
|
|
||||||
self.trans2.connect('cancellable-changed',
|
|
||||||
self.upgrade_cancellable_changed)
|
|
||||||
self.trans2.connect('progress-details-changed',
|
|
||||||
self.upgrade_progress_detail)
|
|
||||||
self.trans2.connect('progress-download-changed',
|
|
||||||
self.upgrade_progress_download)
|
|
||||||
self.trans2.connect('finished', self.upgrade_finish)
|
|
||||||
self.trans2.connect('error', self.upgrade_error)
|
|
||||||
self.trans2.connect("status-details-changed",
|
|
||||||
self.status_details_changed)
|
|
||||||
self.trans2.connect("status-changed", self.status_changed)
|
|
||||||
|
|
||||||
# TODO make a terminal work to see more info
|
|
||||||
# self.trans2.set_terminal(os.ttyname(self.slave))
|
|
||||||
|
|
||||||
'''
|
|
||||||
# TODO implement this
|
|
||||||
self.trans2.connect("medium-required", self._on_medium_required)
|
|
||||||
self.trans2.connect("config-file-conflict",
|
|
||||||
self._on_config_file_conflict)
|
|
||||||
remove_obsoleted_depends
|
|
||||||
'''
|
|
||||||
self.trans2.set_debconf_frontend('kde')
|
|
||||||
self.trans2.run()
|
|
||||||
|
|
||||||
except (NotAuthorizedError, TransactionFailed) as e:
|
|
||||||
print("Warning: install transaction not completed successfully:"
|
|
||||||
+ "{}".format(e))
|
|
||||||
|
|
||||||
def call_reject(self):
|
|
||||||
app.quit()
|
|
||||||
|
|
||||||
|
|
||||||
class App(QApplication):
|
|
||||||
def __init__(self, options, *args):
|
|
||||||
QApplication.__init__(self, *args)
|
|
||||||
self.dialog = DialogUpg(options)
|
|
||||||
self.dialog.show()
|
|
||||||
|
|
||||||
|
|
||||||
def main(args, options):
|
|
||||||
global app
|
|
||||||
app = App(options, args)
|
|
||||||
app.setWindowIcon(QIcon.fromTheme("system-software-update"))
|
|
||||||
|
|
||||||
# Check for root permissions
|
|
||||||
if os.geteuid() != 0:
|
|
||||||
text = "Please run this software with administrative rights."
|
|
||||||
text += "To do so, run this program with lxqt-sudo."
|
|
||||||
title = "Need administrative powers"
|
|
||||||
QMessageBox.critical(None, title, text)
|
|
||||||
sys.exit()
|
|
||||||
else:
|
|
||||||
app.exec_()
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
# check arguments
|
|
||||||
parser = OptionParser(usage='%prog [options]',
|
|
||||||
version='0.1',)
|
|
||||||
parser.add_option("-c",
|
|
||||||
"--cache-update",
|
|
||||||
action="store_true",
|
|
||||||
dest="cacheUpdate",
|
|
||||||
help="Update Cache Before Upgrade")
|
|
||||||
parser.add_option("-f",
|
|
||||||
"--full-upgrade",
|
|
||||||
action="store_true",
|
|
||||||
dest="fullUpgrade",
|
|
||||||
help="Full upgrade same as dist-upgrade")
|
|
||||||
(options, args) = parser.parse_args()
|
|
||||||
|
|
||||||
# run it
|
|
||||||
main(sys.argv, options)
|
|
||||||
Loading…
Reference in new issue