def removeOldKernels(self, response):
if response:
if self.progressView.lineCount('Kernels') > 0:
self.logger.info("Starting kernel removal...")
self.labels[(5, 1)].setMovie(self.movie)
self.setCursor(QtCore.Qt.BusyCursor)
self.progress.setValue(0)
try:
self.logger.info("Removing old kernels...")
self.install = Install("Kernels", "removing old kernels", False)
self.install.show()
self.install.exec_()
self.labels[(5, 1)].setPixmap(self.pixmap2)
self.unsetCursor()
self.lbl1.setText("Finished")
except Exception as arg:
self.logger.error("Kernel removal failed [{}]".format(str(arg)))
print "Sorry, kernel removal failed [{}]".format(str(arg))
self.removeUsers(response)
self.showUserInfo()
self.progress.setValue(1)
else:
self.labels[(5, 1)].setPixmap(self.pixmap2)
self.removeUsers(response)
self.showUserInfo()
self.progress.setValue(1)
else:
self.lbl1.setText("Finished")
self.removeUsers(response)
self.progress.setValue(1)
self.showUserInfo()
self.logger.info("Old kernel removal option not chosen")
def installPackages(self):
self.logger.info("Starting kernel removal...")
self.labels[(4, 1)].setMovie(self.movie)
self.movie.start()
self.install = Install("custom-install", "Installing packages", True)
self.install.show()
self.install.exec_()
self.labels[(4, 1)].setPixmap(self.pixmap2)
def installPackages(self):
self.logger.info("Starting package installation...")
self.labels[(4, 1)].setMovie(self.movie)
self.movie.start()
self.install = Install("apps-to-install", "Installing packages", True)
self.install.show()
self.install.exec_()
self.labels[(4, 1)].setPixmap(self.pixmap2)
self.removeUsers()
def installPackages(self):
self.logger.info("Starting installations...")
self.labels[(4, 1)].setMovie(self.movie)
if self.progressView.lineCount('custom-install') > 0:
self.install = Install("custom-install", "Installing packages", True)
self.install.show()
self.install.exec_()
self.labels[(4, 1)].setPixmap(self.pixmap2)
else:
self.labels[(4, 1)].setPixmap(self.pixmap2)
def installPackages(self):
self.logger.info("Starting installations...")
self.labels[(4, 1)].setMovie(self.movie)
if self.progressView.lineCount('apps-to-install') > 0:
self.install = Install('apps-to-install', 'Installing packages', True)
self.install.show()
self.install.exec_()
self.labels[(4, 1)].setPixmap(self.pixmap2)
self.removeUsers()
else:
self.labels[(4, 1)].setPixmap(self.pixmap2)
self.removeUsers()
def installPackages(self):
if len(self.unavailable) > 0:
self.keptBack(self.unavailable)
model = self.model
for index in xrange(model.rowCount()):
item = model.item(index)
if self.isWritten:
mode = "a"
else:
mode = "w"
self.isWritten = True
if item.isCheckable() and item.checkState() == QtCore.Qt.Checked:
with open("install", mode) as f_out:
to_install = str(item.text()).split(':')[0]
f_out.write("{}\n".format(to_install))
self.install = Install("install", "Installing packages", True)
self.install.show()
self.install.exec_()
def installPackages(self):
self.btnInstall.setEnabled(False)
model = self.model
for index in xrange(model.rowCount()):
item = model.item(index)
if self.isWritten:
mode = 'a'
else:
mode = 'w'
self.isWritten = True
if item.isCheckable() and item.checkState() == QtCore.Qt.Checked:
with open('install', mode) as f_out:
to_install = str(item.text()).split(':')[0]
f_out.write('{}\n'.format(to_install))
self.install = Install('install', 'Installing packages', True)
self.install.show()
self.install.exec_()
self.btnInstall.setEnabled(True)
self.removeItems()
def installPackages(self):
self.logger.info("Starting installations...")
self.labels[(4, 1)].setMovie(self.movie)
if self.progressView.lineCount('apps-to-install') > 0:
self.install = Install('apps-to-install', 'Installing packages', True)
self.install.show()
self.install.exec_()
self.labels[(4, 1)].setPixmap(self.pixmap2)
self.removeUsers()
else:
self.labels[(4, 1)].setPixmap(self.pixmap2)
self.removeUsers()
def installPackages(self):
self.btnInstall.setEnabled(False)
model = self.model
for index in xrange(model.rowCount()):
item = model.item(index)
if self.isWritten:
mode = 'a'
else:
mode = 'w'
self.isWritten = True
if item.isCheckable() and item.checkState() == QtCore.Qt.Checked:
with open('install', mode) as f_out:
to_install = str(item.text()).split(':')[0]
f_out.write('{}\n'.format(to_install))
self.install = Install('install', 'Installing packages', True)
self.install.show()
self.install.exec_()
self.btnInstall.setEnabled(True)
self.removeItems()
class Apply(QtGui.QDialog):
def __init__(self, file_in, response, rsu, parent=None):
super(Apply, self).__init__(parent)
self.setMinimumSize(400, 250)
self.file_in = file_in
self.response = response
self.rsu = rsu
self.custom_user = bool
self.remaining = 0
self.no_show = False
self.setWindowTitle("Applying")
self.buttonCancel = QtGui.QPushButton()
self.buttonCancel.setText("Cancel")
self.buttonCancel.clicked.connect(self.finished)
self.progress = QtGui.QProgressBar()
self.progress2 = QtGui.QProgressBar()
self.progress2.setVisible(False)
self.lbl1 = QtGui.QLabel()
gif = os.path.abspath(
"/usr/lib/resetter/data/icons/chassingarrows.gif")
self.movie = QtGui.QMovie(gif)
self.movie.setScaledSize(QtCore.QSize(20, 20))
self.pixmap = QtGui.QPixmap(
"/usr/lib/resetter/data/icons/checkmark.png")
self.cuser = '******'
self.pixmap2 = self.pixmap.scaled(20, 20)
verticalLayout = QtGui.QVBoxLayout(self)
verticalLayout.addWidget(self.lbl1)
verticalLayout.addWidget(self.progress)
verticalLayout.addWidget(self.progress2)
gridLayout = QtGui.QGridLayout()
self.labels = {}
for i in range(1, 7):
for j in range(1, 3):
self.labels[(i, j)] = QtGui.QLabel()
self.labels[(i, j)].setMinimumHeight(20)
gridLayout.addWidget(self.labels[(i, j)], i, j)
gridLayout.setAlignment(QtCore.Qt.AlignCenter)
self.labels[(1, 2)].setText("Loading packages")
self.labels[(2, 2)].setText("Removing packages")
self.labels[(3, 2)].setText("Cleaning Up")
self.labels[(4, 2)].setText("Installing packages")
if self.response:
self.labels[(5, 2)].setText("Removing old kernels")
self.labels[(6, 2)].setText("Deleting Users")
else:
self.labels[(5, 2)].setText("Deleting Users")
verticalLayout.addSpacing(20)
verticalLayout.addLayout(gridLayout)
verticalLayout.addWidget(self.buttonCancel, 0, QtCore.Qt.AlignRight)
self.logger = logging.getLogger(__name__)
self.logger.setLevel(logging.DEBUG)
handler = logging.FileHandler('/var/log/resetter/resetter.log')
handler.setLevel(logging.DEBUG)
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(funcName)s - %(levelname)s - %(message)s'
)
handler.setFormatter(formatter)
self.logger.addHandler(handler)
self.progressView = ProgressThread(self.file_in, False)
self.account = AccountDialog()
self.connect(self.progressView,
QtCore.SIGNAL("updateProgressBar(int, bool)"),
self.updateProgressBar)
self.connect(self.progressView.aprogress,
QtCore.SIGNAL("updateProgressBar2(int, bool, QString)"),
self.updateProgressBar2)
self.connect(self.progressView.iprogress,
QtCore.SIGNAL("updateProgressBar2(int, bool, QString)"),
self.updateProgressBar2)
self.connect(self.progressView,
QtCore.SIGNAL("showError(QString, QString)"),
self.showError)
self.addUser1()
def updateProgressBar(self, percent, isdone):
self.lbl1.setText("Loading Package List")
self.progress.setValue(percent)
self.labels[(1, 1)].setMovie(self.movie)
self.movie.start()
if isdone:
self.movie.stop()
def updateProgressBar2(self, percent, isdone, status):
self.progress.setVisible(False)
self.progress2.setVisible(True)
self.labels[(1, 1)].setPixmap(self.pixmap2)
self.lbl1.setText(status)
self.labels[(2, 1)].setMovie(self.movie)
self.movie.start()
self.progress2.setValue(percent)
if isdone:
self.progressView.conclude_op.connect(self.finished)
self.labels[(2, 1)].setPixmap(self.pixmap2)
self.movie.stop()
self.fixBroken()
def fixBroken(self):
self.labels[(3, 1)].setMovie(self.movie)
self.movie.start()
self.lbl1.setText("Cleaning up...")
self.logger.info("Cleaning up...")
self.progress2.setRange(0, 0)
self.setCursor(QtCore.Qt.BusyCursor)
self.process = QtCore.QProcess()
self.process.finished.connect(self.onFinished)
self.process.start('bash',
['/usr/lib/resetter/data/scripts/fix-broken.sh'])
def onFinished(self, exit_code, exit_status):
if exit_code or exit_status != 0:
self.progress2.setRange(0, 1)
self.logger.error(
"fixBroken() finished with exit code: {} and exit_status {}.".
format(exit_code, exit_status))
else:
self.progress2.setRange(0, 1)
self.progress2.setValue(1)
self.logger.debug(
"Cleanup finished with exit code: {} and exit_status {}.".
format(exit_code, exit_status))
self.labels[(3, 1)].setPixmap(self.pixmap2)
self.unsetCursor()
self.lbl1.setText("Done Cleanup")
self.installPackages()
self.removeOldKernels(self.response)
def addUser1(self):
choice = QtGui.QMessageBox.question \
(self, 'Would you like set your new account?',
"Set your own account?",
QtGui.QMessageBox.Yes | QtGui.QMessageBox.No)
if choice == QtGui.QMessageBox.Yes:
self.custom_user = True
self.show()
self.account.exec_()
self.start()
self.showMinimized()
elif choice == QtGui.QMessageBox.No:
self.custom_user = False
self.no_show = True
with open('users') as u, open('custom-users-to-delete.sh') as du:
converted_du = []
i = 0
for line in du:
line = line.split(' ')[-1]
converted_du.append(line)
if len(converted_du) > 0:
diff = set(u).difference(converted_du)
for x in diff:
i += 1
else:
i = len(u.read().strip().splitlines())
self.remaining = i
self.start()
def addUser2(
self
): # determine to add a backup user if all normal users are marked for deletion.
if self.custom_user:
p = subprocess.check_output(['bash', self.cuser])
print p
else:
if self.remaining == 0:
self.no_show = False
p = subprocess.check_output(
['bash', '/usr/lib/resetter/data/scripts/new-user.sh'])
print p
def removeOldKernels(self, response):
if response:
if self.progressView.lineCount('Kernels') > 0:
self.logger.info("Starting kernel removal...")
self.labels[(5, 1)].setMovie(self.movie)
self.setCursor(QtCore.Qt.BusyCursor)
self.progress.setValue(0)
try:
self.logger.info("Removing old kernels...")
self.install = Install("Kernels", "removing old kernels",
False)
self.install.show()
self.install.exec_()
self.labels[(5, 1)].setPixmap(self.pixmap2)
self.unsetCursor()
self.lbl1.setText("Finished")
except Exception as arg:
self.logger.error("Kernel removal failed [{}]".format(
str(arg)))
print "Sorry, kernel removal failed [{}]".format(str(arg))
self.removeUsers(response)
self.addUser2()
self.showUserInfo()
self.progress.setValue(1)
else:
self.labels[(5, 1)].setPixmap(self.pixmap2)
self.removeUsers(response)
self.addUser2()
self.showUserInfo()
self.progress.setValue(1)
else:
self.lbl1.setText("Finished")
self.removeUsers(response)
self.addUser2()
self.progress.setValue(1)
self.showUserInfo()
self.logger.info("Old kernel removal option not chosen")
def installPackages(self):
self.logger.info("Starting installations...")
self.labels[(4, 1)].setMovie(self.movie)
if self.progressView.lineCount('custom-install') > 0:
self.install = Install("custom-install", "Installing packages",
True)
self.install.show()
self.install.exec_()
self.labels[(4, 1)].setPixmap(self.pixmap2)
else:
self.labels[(4, 1)].setPixmap(self.pixmap2)
def start(self):
self.progressView.start()
@QtCore.pyqtSlot()
def finished(self):
self.logger.warning("finished apt operation")
self.progressView.thread1.finished.connect(
self.progressView.thread1.exit)
self.progressView.thread2.finished.connect(
self.progressView.thread2.exit)
self.progressView.conclude_op.connect(self.progressView.exit)
def removeSystemUsers(self, rsu):
if rsu:
self.logger.info("Starting user removal")
with open("non-default-users") as f_in, open(
'custom-users-to-delete.sh', 'a') as output:
for line in f_in:
line = ("userdel -rf ", line)
output.writelines(line)
else:
pass
def removeUsers(self, response):
self.removeSystemUsers(self.rsu)
if response:
self.logger.info("Starting user removal")
self.labels[(6, 1)].setMovie(self.movie)
try:
p = subprocess.check_output(
['bash', 'custom-users-to-delete.sh'])
print p
except subprocess.CalledProcessError, e:
self.logger.error("unable removing user [{}]".format(e.output))
else:
self.logger.debug("user removal completed successfully")
self.labels[(6, 1)].setPixmap(self.pixmap2)
else:
class Apply(QtGui.QDialog):
def __init__(self, file_in, parent=None):
super(Apply, self).__init__(parent)
self.setMinimumSize(400, 250)
self.file_in = file_in
self.setWindowTitle("Applying")
self.error_msg = QtGui.QMessageBox()
self.error_msg.setIcon(QtGui.QMessageBox.Critical)
self.error_msg.setWindowTitle("Error")
self.buttonCancel = QtGui.QPushButton()
self.buttonCancel.setText("Cancel")
self.buttonCancel.clicked.connect(self.finished)
self.progress = QtGui.QProgressBar()
self.progress2 = QtGui.QProgressBar()
self.progress2.setVisible(False)
self.lbl1 = QtGui.QLabel()
gif = os.path.abspath(
"/usr/lib/resetter/data/icons/chassingarrows.gif")
self.movie = QtGui.QMovie(gif)
self.movie.setScaledSize(QtCore.QSize(20, 20))
self.pixmap = QtGui.QPixmap(
"/usr/lib/resetter/data/icons/checkmark.png")
self.pixmap2 = self.pixmap.scaled(20, 20)
verticalLayout = QtGui.QVBoxLayout(self)
verticalLayout.addWidget(self.lbl1)
verticalLayout.addWidget(self.progress)
verticalLayout.addWidget(self.progress2)
gridLayout = QtGui.QGridLayout()
self.labels = {}
for i in range(1, 7):
for j in range(1, 3):
self.labels[(i, j)] = QtGui.QLabel()
self.labels[(i, j)].setMinimumHeight(20)
gridLayout.addWidget(self.labels[(i, j)], i, j)
gridLayout.setAlignment(QtCore.Qt.AlignCenter)
self.labels[(1, 2)].setText("Loading packages")
self.labels[(2, 2)].setText("Removing packages")
self.labels[(3, 2)].setText("Cleaning Up")
self.labels[(4, 2)].setText("Installing packages")
self.labels[(5, 2)].setText("Deleting Users")
verticalLayout.addSpacing(20)
verticalLayout.addLayout(gridLayout)
verticalLayout.addWidget(self.buttonCancel, 0, QtCore.Qt.AlignRight)
self.logger = logging.getLogger(__name__)
self.logger.setLevel(logging.DEBUG)
handler = logging.FileHandler('/var/log/resetter/resetter.log')
handler.setLevel(logging.DEBUG)
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(funcName)s - %(levelname)s - %(message)s'
)
handler.setFormatter(formatter)
self.logger.addHandler(handler)
self.progressView = ProgressThread(self.file_in, False)
self.connect(self.progressView,
QtCore.SIGNAL("updateProgressBar(int, bool)"),
self.updateProgressBar)
self.connect(self.progressView.aprogress,
QtCore.SIGNAL("updateProgressBar2(int, bool, QString)"),
self.updateProgressBar2)
self.connect(self.progressView.iprogress,
QtCore.SIGNAL("updateProgressBar2(int, bool, QString)"),
self.updateProgressBar2)
self.connect(self.progressView,
QtCore.SIGNAL("showError(QString, QString)"),
self.showError)
self.addUser()
self.start()
def addUser(self):
try:
self.logger.info("Adding default user...")
p = subprocess.check_output(
['bash', '/usr/lib/resetter/data/scripts/new-user.sh'])
print p
except subprocess.CalledProcessError as e:
self.logger.error("unable to add user Error: {}".format(e.output))
else:
self.logger.info("Default user added")
def updateProgressBar(self, percent, isdone):
self.lbl1.setText("Loading Package List")
self.progress.setValue(percent)
self.labels[(1, 1)].setMovie(self.movie)
self.movie.start()
if isdone:
self.movie.stop()
def updateProgressBar2(self, percent, isdone, status):
self.progress.setVisible(False)
self.progress2.setVisible(True)
self.labels[(1, 1)].setPixmap(self.pixmap2)
self.lbl1.setText(status)
self.labels[(2, 1)].setMovie(self.movie)
self.movie.start()
self.progress2.setValue(percent)
if isdone:
self.progressView.conclude_op.connect(self.finished)
self.labels[(2, 1)].setPixmap(self.pixmap2)
self.movie.stop()
self.fixBroken()
def fixBroken(self):
self.labels[(3, 1)].setMovie(self.movie)
self.movie.start()
self.lbl1.setText("Cleaning up...")
self.logger.info("Cleaning up...")
self.progress2.setRange(0, 0)
self.setCursor(QtCore.Qt.BusyCursor)
self.process = QtCore.QProcess()
self.process.finished.connect(self.onFinished)
self.process.start('bash',
['/usr/lib/resetter/data/scripts/fix-broken.sh'])
def onFinished(self, exit_code, exit_status):
if exit_code or exit_status != 0:
self.progress2.setRange(0, 1)
self.logger.error(
"fixBroken() finished with exit code: {} and exit_status {}.".
format(exit_code, exit_status))
else:
self.progress2.setRange(0, 1)
self.progress2.setValue(1)
self.logger.debug(
"Cleanup finished with exit code: {} and exit_status {}.".
format(exit_code, exit_status))
self.labels[(3, 1)].setPixmap(self.pixmap2)
self.unsetCursor()
self.lbl1.setText("Done Cleanup")
self.installPackages()
def installPackages(self):
self.logger.info("Starting installations...")
self.labels[(4, 1)].setMovie(self.movie)
if self.progressView.lineCount('apps-to-install') > 0:
self.install = Install('apps-to-install', 'Installing packages',
True)
self.install.show()
self.install.exec_()
self.labels[(4, 1)].setPixmap(self.pixmap2)
self.removeUsers()
else:
self.labels[(4, 1)].setPixmap(self.pixmap2)
self.removeUsers()
def start(self):
self.progressView.start()
@QtCore.pyqtSlot()
def finished(self):
self.logger.warning("finished apt operation")
self.progressView.thread1.finished.connect(
self.progressView.thread1.exit)
self.progressView.thread2.finished.connect(
self.progressView.thread2.exit)
self.progressView.conclude_op.connect(self.progressView.exit)
def removeUsers(self):
self.logger.info("Starting user removal")
self.labels[(5, 1)].setMovie(self.movie)
with open('users') as f_in, open('non-default-users',
'r') as ndu, open(
"users-to-delete.sh",
"w") as output:
for line in f_in:
line = ('userdel -rf ', line)
output.writelines(line)
for s_user in ndu:
s_user = ('userdel -rf ', s_user)
output.writelines(s_user)
try:
subprocess.Popen(['bash', 'users-to-delete.sh'],
stderr=subprocess.STDOUT,
stdout=subprocess.PIPE)
except subprocess.CalledProcessError as e:
print "error: {}".format(e.output)
else:
self.movie.stop()
self.labels[(5, 1)].setPixmap(self.pixmap2)
self.lbl1.setText("Finished")
self.showUserInfo()
def rebootMessage(self):
choice = QtGui.QMessageBox.information \
(self, 'Please reboot to complete system changes',
"Reboot now?",
QtGui.QMessageBox.Yes | QtGui.QMessageBox.No)
if choice == QtGui.QMessageBox.Yes:
self.logger.info("system rebooted after package removals")
os.system('reboot')
else:
self.logger.info("reboot was delayed.")
def showMessage(self):
msg = QtGui.QMessageBox(self)
msg.setWindowTitle("Packages kept back")
msg.setIcon(QtGui.QMessageBox.Information)
msg.setText(
"These packages could cause problems if removed so they've been kept back."
)
text = "\n".join(self.progressView.broken_list)
msg.setInformativeText(text)
msg.exec_()
def showError(self, error, m_type):
self.movie.stop()
msg = QtGui.QMessageBox(self)
msg.setWindowTitle(m_type)
msg.setIcon(QtGui.QMessageBox.Critical)
msg.setText("Something went wrong, please check details.")
msg.setDetailedText(error)
msg.exec_()
def showUserInfo(self):
msg = QtGui.QMessageBox(self)
msg.setWindowTitle("User Credentials")
msg.setIcon(QtGui.QMessageBox.Information)
msg.setText("Please use these credentials the next time you log-in")
msg.setInformativeText(
"USERNAME: <b>default</b><br/> PASSWORD: <b>NewLife3!</b>")
msg.setDetailedText(
"This username was automatically created as your backup user")
msg.exec_()
self.logger.info("Credential message info shown")
self.rebootMessage()
class Apply(QtGui.QDialog):
def __init__(self, file_in, response, rsu, parent=None):
super(Apply, self).__init__(parent)
self.setMinimumSize(400, 250)
self.file_in = file_in
self.response = response
self.rsu = rsu
self.setWindowTitle("Applying")
self.error_msg = QtGui.QMessageBox()
self.error_msg.setIcon(QtGui.QMessageBox.Critical)
self.error_msg.setWindowTitle("Error")
self.buttonCancel = QtGui.QPushButton()
self.buttonCancel.setText("Cancel")
self.buttonCancel.clicked.connect(self.finished)
self.progress = QtGui.QProgressBar()
self.progress2 = QtGui.QProgressBar()
self.progress2.setVisible(False)
self.lbl1 = QtGui.QLabel()
gif = os.path.abspath("/usr/lib/resetter/data/icons/chassingarrows.gif")
self.movie = QtGui.QMovie(gif)
self.movie.setScaledSize(QtCore.QSize(20, 20))
self.pixmap = QtGui.QPixmap("/usr/lib/resetter/data/icons/checkmark.png")
self.pixmap2 = self.pixmap.scaled(20, 20)
verticalLayout = QtGui.QVBoxLayout(self)
verticalLayout.addWidget(self.lbl1)
verticalLayout.addWidget(self.progress)
verticalLayout.addWidget(self.progress2)
gridLayout = QtGui.QGridLayout()
self.labels = {}
for i in range(1, 7):
for j in range(1, 3):
self.labels[(i, j)] = QtGui.QLabel()
self.labels[(i, j)].setMinimumHeight(20)
gridLayout.addWidget(self.labels[(i, j)], i, j)
gridLayout.setAlignment(QtCore.Qt.AlignCenter)
self.labels[(1, 2)].setText("Loading packages")
self.labels[(2, 2)].setText("Removing packages")
self.labels[(3, 2)].setText("Cleaning Up")
self.labels[(4, 2)].setText("Installing packages")
if self.response:
self.labels[(5, 2)].setText("Removing old kernels")
self.labels[(6, 2)].setText("Deleting Users")
else:
self.labels[(5, 2)].setText("Deleting Users")
verticalLayout.addSpacing(20)
verticalLayout.addLayout(gridLayout)
verticalLayout.addWidget(self.buttonCancel, 0, QtCore.Qt.AlignRight)
self.logger = logging.getLogger(__name__)
self.logger.setLevel(logging.DEBUG)
handler = logging.FileHandler('/var/log/resetter/resetter.log')
handler.setLevel(logging.DEBUG)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(funcName)s - %(levelname)s - %(message)s')
handler.setFormatter(formatter)
self.logger.addHandler(handler)
self.progressView = ProgressThread(self.file_in, False)
self.account = AccountDialog()
self.connect(self.progressView, QtCore.SIGNAL("updateProgressBar(int, bool)"), self.updateProgressBar)
self.connect(self.progressView.aprogress, QtCore.SIGNAL("updateProgressBar2(int, bool, QString)"), self.updateProgressBar2)
self.connect(self.progressView.iprogress, QtCore.SIGNAL("updateProgressBar2(int, bool, QString)"), self.updateProgressBar2)
self.connect(self.progressView, QtCore.SIGNAL("showError(QString, QString)"), self.showError)
self.addUser()
def updateProgressBar(self, percent, isdone):
self.lbl1.setText("Loading Package List")
self.progress.setValue(percent)
self.labels[(1, 1)].setMovie(self.movie)
self.movie.start()
if isdone:
self.movie.stop()
def updateProgressBar2(self, percent, isdone, status):
self.progress.setVisible(False)
self.progress2.setVisible(True)
self.labels[(1, 1)].setPixmap(self.pixmap2)
self.lbl1.setText(status)
self.labels[(2, 1)].setMovie(self.movie)
self.movie.start()
self.progress2.setValue(percent)
if isdone:
self.progressView.conclude_op.connect(self.finished)
self.labels[(2, 1)].setPixmap(self.pixmap2)
self.movie.stop()
self.fixBroken()
def fixBroken(self):
self.labels[(3, 1)].setMovie(self.movie)
self.movie.start()
self.lbl1.setText("Cleaning up...")
self.logger.info("Cleaning up...")
self.progress2.setRange(0, 0)
self.setCursor(QtCore.Qt.BusyCursor)
self.process = QtCore.QProcess()
self.process.finished.connect(self.onFinished)
self.process.start('bash', ['/usr/lib/resetter/data/scripts/fix-broken.sh'])
def onFinished(self, exit_code, exit_status):
if exit_code or exit_status != 0:
self.progress2.setRange(0, 1)
self.logger.error("fixBroken() finished with exit code: {} and exit_status {}."
.format(exit_code, exit_status))
else:
self.progress2.setRange(0, 1)
self.progress2.setValue(1)
self.logger.debug("Cleanup finished with exit code: {} and exit_status {}.".format(exit_code, exit_status))
self.labels[(3, 1)].setPixmap(self.pixmap2)
self.unsetCursor()
self.lbl1.setText("Done Cleanup")
self.installPackages()
self.removeOldKernels(self.response)
def addUser(self):
choice = QtGui.QMessageBox.question \
(self, 'Would you like set your new account?',
"Set your own account? Click 'No' so that I can create a default account for you instead",
QtGui.QMessageBox.Yes | QtGui.QMessageBox.No)
if choice == QtGui.QMessageBox.Yes:
self.show()
self.account.exec_()
self.start()
self.showMinimized()
print "Adding custom user"
if choice == QtGui.QMessageBox.No:
print "Adding default user"
try:
p = subprocess.Popen(['bash', '/usr/lib/resetter/data/scripts/new-user.sh'], stderr=subprocess.STDOUT,
stdout=subprocess.PIPE)
self.logger.info("Default user added")
p.wait()
self.start()
except subprocess.CalledProcessError as e:
self.logger.error("unable to add default user [{}]".format(e.output), exc_info=True)
print e.output
def removeOldKernels(self, response):
if response:
if self.progressView.lineCount('Kernels') > 0:
self.logger.info("Starting kernel removal...")
self.labels[(5, 1)].setMovie(self.movie)
self.setCursor(QtCore.Qt.BusyCursor)
self.progress.setValue(0)
try:
self.logger.info("Removing old kernels...")
self.install = Install("Kernels", "removing old kernels", False)
self.install.show()
self.install.exec_()
self.labels[(5, 1)].setPixmap(self.pixmap2)
self.unsetCursor()
self.lbl1.setText("Finished")
except Exception as arg:
self.logger.error("Kernel removal failed [{}]".format(str(arg)))
print "Sorry, kernel removal failed [{}]".format(str(arg))
self.removeUsers(response)
self.showUserInfo()
self.progress.setValue(1)
else:
self.labels[(5, 1)].setPixmap(self.pixmap2)
self.removeUsers(response)
self.showUserInfo()
self.progress.setValue(1)
else:
self.lbl1.setText("Finished")
self.removeUsers(response)
self.progress.setValue(1)
self.showUserInfo()
self.logger.info("Old kernel removal option not chosen")
def installPackages(self):
self.logger.info("Starting installations...")
self.labels[(4, 1)].setMovie(self.movie)
if self.progressView.lineCount('custom-install') > 0:
self.install = Install("custom-install", "Installing packages", True)
self.install.show()
self.install.exec_()
self.labels[(4, 1)].setPixmap(self.pixmap2)
else:
self.labels[(4, 1)].setPixmap(self.pixmap2)
def start(self):
self.progressView.start()
@QtCore.pyqtSlot()
def finished(self):
self.logger.warning("finished apt operation")
self.progressView.thread1.finished.connect(self.progressView.thread1.exit)
self.progressView.thread2.finished.connect(self.progressView.thread2.exit)
self.progressView.conclude_op.connect(self.progressView.exit)
self.close()
def removeSystemUsers(self, rsu):
if rsu:
self.logger.info("Starting user removal")
with open("non-default-users") as f_in, open('custom-users-to-delete.sh', 'a') as output:
for line in f_in:
line = ("userdel -rf ", line)
output.writelines(line)
else:
pass
def removeUsers(self, response):
self.removeSystemUsers(self.rsu)
if response:
self.logger.info("Starting user removal")
self.labels[(6, 1)].setMovie(self.movie)
try:
subprocess.Popen(['bash', 'custom-users-to-delete.sh'], stderr=subprocess.STDOUT,
stdout=subprocess.PIPE)
self.logger.debug("user removal completed successfully: [{}]".format(subprocess.STDOUT))
except subprocess.CalledProcessError, e:
self.logger.error("unable removing user [{}]".format(e.output))
else:
self.labels[(6, 1)].setPixmap(self.pixmap2)
else:
class Apply(QtGui.QDialog):
def __init__(self, file_in, parent=None):
super(Apply, self).__init__(parent)
self.setMinimumSize(400, 250)
self.file_in = file_in
self.setWindowTitle("Applying")
self.error_msg = QtGui.QMessageBox()
self.error_msg.setIcon(QtGui.QMessageBox.Critical)
self.error_msg.setWindowTitle("Error")
self.buttonCancel = QtGui.QPushButton()
self.buttonCancel.setText("Cancel")
self.buttonCancel.clicked.connect(self.finished)
self.progress = QtGui.QProgressBar()
self.progress2 = QtGui.QProgressBar()
self.progress2.setVisible(False)
self.lbl1 = QtGui.QLabel()
gif = os.path.abspath("/usr/lib/resetter/data/icons/chassingarrows.gif")
self.movie = QtGui.QMovie(gif)
self.movie.setScaledSize(QtCore.QSize(20, 20))
self.pixmap = QtGui.QPixmap("/usr/lib/resetter/data/icons/checkmark.png")
self.pixmap2 = self.pixmap.scaled(20, 20)
verticalLayout = QtGui.QVBoxLayout(self)
verticalLayout.addWidget(self.lbl1)
verticalLayout.addWidget(self.progress)
verticalLayout.addWidget(self.progress2)
gridLayout = QtGui.QGridLayout()
self.labels = {}
for i in range(1, 7):
for j in range(1, 3):
self.labels[(i, j)] = QtGui.QLabel()
self.labels[(i, j)].setMinimumHeight(20)
gridLayout.addWidget(self.labels[(i, j)], i, j)
gridLayout.setAlignment(QtCore.Qt.AlignCenter)
self.labels[(1, 2)].setText("Loading packages")
self.labels[(2, 2)].setText("Removing packages")
self.labels[(3, 2)].setText("Cleaning Up")
self.labels[(4, 2)].setText("Installing packages")
self.labels[(5, 2)].setText("Deleting Users")
verticalLayout.addSpacing(20)
verticalLayout.addLayout(gridLayout)
verticalLayout.addWidget(self.buttonCancel, 0, QtCore.Qt.AlignRight)
self.logger = logging.getLogger(__name__)
self.logger.setLevel(logging.DEBUG)
handler = logging.FileHandler('/var/log/resetter/resetter.log')
handler.setLevel(logging.DEBUG)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(funcName)s - %(levelname)s - %(message)s')
handler.setFormatter(formatter)
self.logger.addHandler(handler)
self.progressView = ProgressThread(self.file_in, False)
self.connect(self.progressView, QtCore.SIGNAL("updateProgressBar(int, bool)"), self.updateProgressBar)
self.connect(self.progressView.aprogress, QtCore.SIGNAL("updateProgressBar2(int, bool, QString)"), self.updateProgressBar2)
self.connect(self.progressView.iprogress, QtCore.SIGNAL("updateProgressBar2(int, bool, QString)"), self.updateProgressBar2)
self.connect(self.progressView, QtCore.SIGNAL("showError(QString, QString)"), self.showError)
self.addUser()
self.start()
def addUser(self):
try:
self.logger.info("Adding default user...")
p = subprocess.check_output(['bash', '/usr/lib/resetter/data/scripts/new-user.sh'])
print p
self.logger.info("Default user added")
except subprocess.CalledProcessError as e:
self.logger.error("unable to add user Error: {}".format(e.output))
def updateProgressBar(self, percent, isdone):
self.lbl1.setText("Loading Package List")
self.progress.setValue(percent)
self.labels[(1, 1)].setMovie(self.movie)
self.movie.start()
if isdone:
self.movie.stop()
def updateProgressBar2(self, percent, isdone, status):
self.progress.setVisible(False)
self.progress2.setVisible(True)
self.labels[(1, 1)].setPixmap(self.pixmap2)
self.lbl1.setText(status)
self.labels[(2, 1)].setMovie(self.movie)
self.movie.start()
self.progress2.setValue(percent)
if isdone:
self.progressView.conclude_op.connect(self.finished)
self.labels[(2, 1)].setPixmap(self.pixmap2)
self.movie.stop()
self.fixBroken()
def fixBroken(self):
self.labels[(3, 1)].setMovie(self.movie)
self.movie.start()
self.lbl1.setText("Cleaning up...")
self.logger.info("Cleaning up...")
self.progress2.setRange(0, 0)
self.setCursor(QtCore.Qt.BusyCursor)
self.process = QtCore.QProcess()
self.process.finished.connect(self.onFinished)
self.process.start('bash', ['/usr/lib/resetter/data/scripts/fix-broken.sh'])
def onFinished(self, exit_code, exit_status):
if exit_code or exit_status != 0:
self.progress2.setRange(0, 1)
self.logger.error("fixBroken() finished with exit code: {} and exit_status {}."
.format(exit_code, exit_status))
else:
self.progress2.setRange(0, 1)
self.progress2.setValue(1)
self.logger.debug("Cleanup finished with exit code: {} and exit_status {}.".format(exit_code, exit_status))
self.labels[(3, 1)].setPixmap(self.pixmap2)
self.unsetCursor()
self.lbl1.setText("Done Cleanup")
self.installPackages()
def installPackages(self):
self.logger.info("Starting installations...")
self.labels[(4, 1)].setMovie(self.movie)
if self.progressView.lineCount('apps-to-install') > 0:
self.install = Install('apps-to-install', 'Installing packages', True)
self.install.show()
self.install.exec_()
self.labels[(4, 1)].setPixmap(self.pixmap2)
self.removeUsers()
else:
self.labels[(4, 1)].setPixmap(self.pixmap2)
self.removeUsers()
def start(self):
self.progressView.start()
@QtCore.pyqtSlot()
def finished(self):
self.logger.warning("finished apt operation")
self.progressView.thread1.finished.connect(self.progressView.thread1.exit)
self.progressView.thread2.finished.connect(self.progressView.thread2.exit)
self.progressView.conclude_op.connect(self.progressView.exit)
self.close()
def removeUsers(self):
self.logger.info("Starting user removal")
self.labels[(5, 1)].setMovie(self.movie)
with open('users') as f_in, open('non-default-users', 'r') as ndu, open("users-to-delete.sh", "w") as output:
for line in f_in:
line = ('userdel -rf ', line)
output.writelines(line)
for s_user in ndu:
s_user = ('userdel -rf ', s_user)
output.writelines(s_user)
try:
subprocess.Popen(['bash', 'users-to-delete.sh'], stderr=subprocess.STDOUT, stdout=subprocess.PIPE)
except subprocess.CalledProcessError as e:
print "error: {}".format(e.output)
else:
self.movie.stop()
self.labels[(5, 1)].setPixmap(self.pixmap2)
self.lbl1.setText("Finished")
self.showUserInfo()
def getDependencies(self):
try:
self.setCursor(QtCore.Qt.WaitCursor)
sq = '\''
col = ':'
with open("deplist2", "w") as dl:
for pkg in self._cache.get_changes():
dependencies = pkg.versions[0].dependencies
for dependency in dependencies:
dependency = str(dependency).split(sq, 1)[1].split(sq, 1)[0]
if col in dependency:
dependency = dependency.split(col, 1)[0]
dl.write('{}\n'.format(dependency))
with open("keep", "w") as output, open("deplist2", "r") as dl, open(self.file_in, "r") as apps:
diff = set(dl).difference(apps)
for line in diff:
output.writelines(line)
self.unsetCursor()
except Exception as e:
self.unsetCursor()
self.logger.error("getting Dependencies failed: {}".format(e))
def rebootMessage(self):
choice = QtGui.QMessageBox.information \
(self, 'Please reboot to complete system changes',
"Reboot now?",
QtGui.QMessageBox.Yes | QtGui.QMessageBox.No)
if choice == QtGui.QMessageBox.Yes:
self.logger.info("system rebooted after package removals")
os.system('reboot')
else:
self.logger.info("reboot was delayed.")
def showMessage(self):
msg = QtGui.QMessageBox(self)
msg.setWindowTitle("Packages kept back")
msg.setIcon(QtGui.QMessageBox.Information)
msg.setText("These packages could cause problems if removed so they've been kept back.")
text = "\n".join(self.progressView.broken_list)
msg.setInformativeText(text)
msg.exec_()
def showError(self, error, m_type):
msg = QtGui.QMessageBox(self)
msg.setWindowTitle(m_type)
msg.setIcon(QtGui.QMessageBox.Critical)
text = "If you're running another package manager such as synaptic or USC please close them and try again."
msg.setText(text)
msg.setDetailedText(error)
msg.exec_()
def showUserInfo(self):
msg = QtGui.QMessageBox(self)
msg.setWindowTitle("User Credentials")
msg.setIcon(QtGui.QMessageBox.Information)
msg.setText("Please use these credentials the next time you log-in")
msg.setInformativeText("USERNAME: <b>default</b><br/> PASSWORD: <b>NewLife3!</b>")
msg.setDetailedText("This username was automatically created as your backup user")
msg.exec_()
self.logger.info("Credential message info shown")
self.rebootMessage()
class Apply(QtGui.QDialog):
def __init__(self, file_in, parent=None):
super(Apply, self).__init__(parent)
self.setMinimumSize(400, 250)
self.setWindowTitle("Applying")
self.buttonCancel = QtGui.QPushButton()
self.buttonCancel.setText("Cancel")
self.progress = QtGui.QProgressBar(self)
self.lbl1 = QtGui.QLabel()
gif = os.path.abspath(
"/usr/lib/resetter/data/icons/chassingarrows.gif")
self.movie = QtGui.QMovie(gif)
self.movie.setScaledSize(QtCore.QSize(20, 20))
self.pixmap = QtGui.QPixmap(
"/usr/lib/resetter/data/icons/checkmark.png")
self.pixmap2 = self.pixmap.scaled(20, 20)
verticalLayout = QtGui.QVBoxLayout(self)
gridLayout = QtGui.QGridLayout()
self.labels = {}
for i in range(1, 6):
for j in range(1, 3):
self.labels[(i, j)] = QtGui.QLabel()
self.labels[(i, j)].setMinimumHeight(20)
gridLayout.addWidget(self.labels[(i, j)], i, j)
gridLayout.setAlignment(QtCore.Qt.AlignCenter)
self.labels[(1, 2)].setText("Loading Apps")
self.labels[(2, 2)].setText("Removing Apps")
self.labels[(3, 2)].setText("Cleaning Up")
self.labels[(4, 2)].setText("Installing packages")
self.labels[(5, 2)].setText("Deleting Users")
verticalLayout.addWidget(self.lbl1)
verticalLayout.addWidget(self.progress)
verticalLayout.addSpacing(20)
verticalLayout.addLayout(gridLayout)
self.logger = logging.getLogger(__name__)
self.logger.setLevel(logging.DEBUG)
self.error_msg = QtGui.QMessageBox()
self.error_msg.setIcon(QtGui.QMessageBox.Critical)
self.error_msg.setWindowTitle("Error")
handler = logging.FileHandler('/var/log/resetter/resetter.log')
handler.setLevel(logging.DEBUG)
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(funcName)s - %(levelname)s - %(message)s'
)
handler.setFormatter(formatter)
self.logger.addHandler(handler)
verticalLayout.addWidget(self.buttonCancel, 0, QtCore.Qt.AlignRight)
self.file_in = file_in
self.buttonCancel.clicked.connect(self.cancel)
self.progressView = ProgressThread(self.file_in)
self.connect(self.progressView,
QtCore.SIGNAL("updateProgressBar(int, bool)"),
self.updateProgressBar)
self._cache = self.progressView._cache
self.aprogress = UIAcquireProgress(self.progress, self.lbl1, False)
self.iprogress = UIInstallProgress(self.progress, self.lbl1)
self.start()
@QtCore.pyqtSlot()
def updateProgressBar(self, percent, isDone):
self.lbl1.setText("Loading Package List")
self.progress.setValue(percent)
self.labels[(1, 1)].setMovie(self.movie)
self.movie.start()
if isDone:
self.buttonCancel.setDisabled(True)
self.movie.stop()
self.labels[(1, 1)].setPixmap(self.pixmap2)
self.removePackages()
def start(self):
# self.progressView.end_of_thread.connect(self.updateProgressBar)
self.progressView.start()
def cancel(self):
self.logger.warning("Progress thread was cancelled")
self.progressView.terminate()
self.close()
def addUser(self):
time.sleep(2)
self.lbl1.setText("Creating Default user")
try:
self.logger.info("Adding default user...")
p = subprocess.Popen(
['bash', '/usr/lib/resetter/data/scripts/new-user.sh'],
stderr=subprocess.STDOUT,
stdout=subprocess.PIPE)
p.wait()
self.logger.info("Default user added")
except subprocess.CalledProcessError as e:
self.logger.error("unable to add user Error: {}".format(e.output))
def removePackages(self):
print "removing packages..."
self.labels[(2, 1)].setMovie(self.movie)
self.movie.start()
try:
self.logger.info("Removing packages...")
self.getDependencies()
if len(self.progressView.broken_list) > 0:
self.showMessage()
self.logger.info("Keep Count before commit: {}".format(
self._cache.keep_count))
self.logger.info("Delete Count before commit: {}".format(
self._cache.delete_count))
self._cache.commit(self.aprogress, self.iprogress)
self.movie.stop()
self.labels[(2, 1)].setPixmap(self.pixmap2)
self.progress.setValue(100)
self.addUser()
self.fixBroken()
except Exception as arg:
self.logger.error("Package removal failed [{}]".format(str(arg)))
self.error_msg.setText(
"Something went wrong... please check details")
self.error_msg.setDetailedText(
"Package removal failed [{}]".format(str(arg)))
self.error_msg.exec_()
def fixBroken(self):
self.progress.setRange(0, 0)
self.logger.info("Cleaning up...")
self.lbl1.setText("Cleaning up...")
self.labels[(3, 1)].setMovie(self.movie)
self.movie.start()
self.setCursor(QtCore.Qt.BusyCursor)
self.prcs = QtCore.QProcess()
self.prcs.finished.connect(self.onFinished)
self.prcs.start('bash',
['/usr/lib/resetter/data/scripts/fix-broken.sh'])
def onFinished(self, exit_code, exit_status):
if exit_code or exit_status != 0:
self.progress.setRange(0, 1)
self.logger.error(
"fixBroken() finished with exit code: {} and exit_status {}.".
format(exit_code, exit_status))
self.error_msg.setText("Error occured, unable to continue.")
self.error_msg.exec_()
else:
self.progress.setRange(0, 1)
self.progress.setValue(1)
self.logger.debug(
"Cleanup finished with exit code: {} and exit_status {}.".
format(exit_code, exit_status))
self.movie.stop()
self.labels[(3, 1)].setPixmap(self.pixmap2)
self.unsetCursor()
self.lbl1.setText("Cleanup done.")
self.installPackages()
def installPackages(self):
self.logger.info("Starting package installation...")
self.labels[(4, 1)].setMovie(self.movie)
self.movie.start()
self.install = Install("apps-to-install", "Installing packages", True)
self.install.show()
self.install.exec_()
self.labels[(4, 1)].setPixmap(self.pixmap2)
self.removeUsers()
def removeUsers(self):
self.logger.info("Starting user removal")
self.labels[(5, 1)].setMovie(self.movie)
self.movie.start()
with open("users") as f_in, open("users-to-delete.sh", "w") as output:
for line in f_in:
line = ("userdel -rf ", line)
output.writelines(line)
try:
subprocess.Popen(['bash', 'users-to-delete.sh'],
stderr=subprocess.STDOUT,
stdout=subprocess.PIPE)
except subprocess.CalledProcessError as e:
print "error: {}".format(e.output)
self.movie.stop()
self.labels[(5, 1)].setPixmap(self.pixmap2)
self.lbl1.setText("Finished")
self.showUserInfo()
def showMessage(self):
msg = QtGui.QMessageBox(self)
msg.setWindowTitle("Packages kept back")
msg.setIcon(QtGui.QMessageBox.Information)
msg.setText(
"These packages could cause problems if removed so they've been kept back."
)
text = "\n".join(self.progressView.broken_list)
msg.setInformativeText(text)
msg.exec_()
def showUserInfo(self):
msg = QtGui.QMessageBox(self)
msg.setWindowTitle("User Credentials")
msg.setIcon(QtGui.QMessageBox.Information)
msg.setText("Please use these credentials the next time you log-in")
msg.setInformativeText(
"USERNAME: <b>default</b><br/> PASSWORD: <b>NewLife3!</b>")
msg.setDetailedText(
"This username was automatically created as your backup user")
msg.exec_()
self.logger.info("Credential message info shown")
self.rebootMessage()
def rebootMessage(self):
choice = QtGui.QMessageBox.information \
(self, 'Please reboot to complete system changes',
"Reboot now?",
QtGui.QMessageBox.Yes | QtGui.QMessageBox.No)
if choice == QtGui.QMessageBox.Yes:
self.logger.info("system rebooted after package removals")
os.system('reboot')
else:
self.logger.info("reboot was delayed.")
def getDependencies(self):
try:
self.setCursor(QtCore.Qt.WaitCursor)
sq = '\''
col = ':'
with open("deplist", "w") as dl:
for pkg in self._cache.get_changes():
dependencies = pkg.versions[0].dependencies
for dependency in dependencies:
dependency = str(dependency).split(sq,
1)[1].split(sq,
1)[0]
if col in dependency:
dependency = dependency.split(col, 1)[0]
dl.write('{}\n'.format(dependency))
with open("keep",
"w") as output, open("deplist",
"r") as dl, open(self.file_in,
"r") as apps:
diff = set(dl).difference(apps)
for line in diff:
output.writelines(line)
self.unsetCursor()
except Exception as e:
self.unsetCursor()
self.logger.error("getting Dependencies failed: {}".format(e))
class Apply(QtGui.QDialog):
def __init__(self, file_in, response, parent=None):
super(Apply, self).__init__(parent)
self.setMinimumSize(400, 250)
self.file_in = file_in
self.response = response
self.setWindowTitle("Applying")
self.error_msg = QtGui.QMessageBox()
self.error_msg.setIcon(QtGui.QMessageBox.Critical)
self.error_msg.setWindowTitle("Error")
self.buttonCancel = QtGui.QPushButton()
self.buttonCancel.setText("Cancel")
self.buttonCancel.clicked.connect(self.cancel)
self.progress = QtGui.QProgressBar(self)
self.lbl1 = QtGui.QLabel()
gif = os.path.abspath(
"/usr/lib/resetter/data/icons/chassingarrows.gif")
self.movie = QtGui.QMovie(gif)
self.movie.setScaledSize(QtCore.QSize(20, 20))
self.pixmap = QtGui.QPixmap(
"/usr/lib/resetter/data/icons/checkmark.png")
self.pixmap2 = self.pixmap.scaled(20, 20)
verticalLayout = QtGui.QVBoxLayout(self)
verticalLayout.addWidget(self.lbl1)
verticalLayout.addWidget(self.progress)
gridLayout = QtGui.QGridLayout()
self.labels = {}
for i in range(1, 7):
for j in range(1, 3):
self.labels[(i, j)] = QtGui.QLabel()
self.labels[(i, j)].setMinimumHeight(20)
gridLayout.addWidget(self.labels[(i, j)], i, j)
gridLayout.setAlignment(QtCore.Qt.AlignCenter)
self.labels[(1, 2)].setText("Loading packages")
self.labels[(2, 2)].setText("Removing packages")
self.labels[(3, 2)].setText("Cleaning Up")
self.labels[(4, 2)].setText("Installing packages")
if self.response:
self.labels[(5, 2)].setText("Removing old kernels")
self.labels[(6, 2)].setText("Deleting Users")
else:
self.labels[(5, 2)].setText("Deleting Users")
verticalLayout.addSpacing(20)
verticalLayout.addLayout(gridLayout)
verticalLayout.addWidget(self.buttonCancel, 0, QtCore.Qt.AlignRight)
self.logger = logging.getLogger(__name__)
self.logger.setLevel(logging.DEBUG)
handler = logging.FileHandler('/var/log/resetter/resetter.log')
handler.setLevel(logging.DEBUG)
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(funcName)s - %(levelname)s - %(message)s'
)
handler.setFormatter(formatter)
self.logger.addHandler(handler)
self.progressView = ProgressThread(self.file_in, False)
self.account = AccountDialog()
self.connect(self.progressView,
QtCore.SIGNAL("updateProgressBar(int, bool)"),
self.updateProgressBar)
self._cache = self.progressView._cache
self.aprogress = UIAcquireProgress(self.progress, self.lbl1, False)
self.iprogress = UIInstallProgress(self.progress, self.lbl1)
self.addUser()
def addUser(self):
choice = QtGui.QMessageBox.question \
(self, 'Would you like set your new account?',
"Set your own account? Click 'No' so that I can create a default account for you instead",
QtGui.QMessageBox.Yes | QtGui.QMessageBox.No)
if choice == QtGui.QMessageBox.Yes:
self.show()
self.account.exec_()
self.start()
self.showMinimized()
print "Adding custom user"
if choice == QtGui.QMessageBox.No:
print "Adding default user"
try:
p = subprocess.Popen(
['bash', '/usr/lib/resetter/data/scripts/new-user.sh'],
stderr=subprocess.STDOUT,
stdout=subprocess.PIPE)
self.logger.info("Default user added")
p.wait()
self.start()
except subprocess.CalledProcessError as e:
self.logger.error("unable to add default user [{}]".format(
e.output),
exc_info=True)
print e.output
def start(self):
self.progressView.start()
def updateProgressBar(self, percent, isdone):
self.lbl1.setText("Loading Package List")
self.progress.setValue(percent)
self.labels[(1, 1)].setMovie(self.movie)
self.movie.start()
if isdone:
self.labels[(1, 1)].setPixmap(self.pixmap2)
self.movie.stop()
self.labels[(2, 1)].setMovie(self.movie)
self.movie.start()
self.removePackages()
def cancel(self):
self.logger.warning("Progress thread was cancelled")
self.progressView.terminate()
self.close()
def removePackages(self):
self.logger.info("Removing Programs")
try:
self.getDependencies()
self.logger.info("Keep Count before commit: {}".format(
self._cache.keep_count))
self.logger.info("Delete Count before commit: {}".format(
self._cache.delete_count))
self.logger.info("Broken Count before commit: {}".format(
self._cache.broken_count))
if len(self.progressView.broken_list) > 0:
self.showMessage()
self._cache.commit(self.aprogress, self.iprogress)
self.logger.info("Broken Count after commit: {}".format(
self._cache.broken_count))
self.movie.stop()
self.labels[(2, 1)].setPixmap(self.pixmap2)
self.progress.setValue(100)
self.fixBroken()
except Exception as arg:
self.movie.stop()
self.logger.error("Sorry, package removal failed [{}]".format(
str(arg)))
self.error_msg.setText(
"Something went wrong... please check details")
self.error_msg.setDetailedText(
"Package removal failed [{}]".format(str(arg)))
self.error_msg.exec_()
def fixBroken(self):
try:
self.lbl1.setText("Cleaning up...")
self.logger.info("Cleaning up...")
self.labels[(3, 1)].setMovie(self.movie)
self.progress.setRange(0, 0)
self.movie.start()
self.setCursor(QtCore.Qt.BusyCursor)
self.process = QtCore.QProcess()
self.process.finished.connect(self.onFinished)
self.process.start(
'bash', ['/usr/lib/resetter/data/scripts/fix-broken.sh'])
except Exception as e:
self.logger.error("error occured during fix-broken [{}]".format(
str(e)))
self.error_msg.setText(
"Please make sure other package managers aren't running")
self.error_msg.exec_()
def onFinished(self, exit_code, exit_status):
if exit_code or exit_status != 0:
self.progress.setRange(0, 1)
self.logger.error(
"fixBroken() finished with exit code: {} and exit_status {}.".
format(exit_code, exit_status))
else:
self.progress.setRange(0, 1)
self.progress.setValue(1)
self.logger.debug(
"Cleanup finished with exit code: {} and exit_status {}.".
format(exit_code, exit_status))
self.movie.stop()
self.labels[(3, 1)].setPixmap(self.pixmap2)
self.unsetCursor()
self.lbl1.setText("Done Cleanup")
self.installPackages()
self.removeOldKernels(self.response)
def installPackages(self):
self.logger.info("Starting kernel removal...")
self.labels[(4, 1)].setMovie(self.movie)
self.movie.start()
self.install = Install("custom-install", "Installing packages", True)
self.install.show()
self.install.exec_()
self.labels[(4, 1)].setPixmap(self.pixmap2)
def removeOldKernels(self, response):
if response:
self.logger.info("Starting kernel removal...")
self.labels[(5, 1)].setMovie(self.movie)
self.movie.start()
self.setCursor(QtCore.Qt.BusyCursor)
self._cache.clear()
self.progress.setValue(0)
try:
self.logger.info("Removing old kernels...")
self.install = Install("Kernels", "removing old kernels",
False)
self.install.show()
self.install.exec_()
self.labels[(5, 1)].setPixmap(self.pixmap2)
self.unsetCursor()
self.lbl1.setText("Finished")
except Exception as arg:
self.logger.error("Kernel removal failed [{}]".format(
str(arg)))
print "Sorry, kernel removal failed [{}]".format(str(arg))
self.removeUsers(response)
self.showUserInfo()
self.progress.setValue(1)
else:
self.lbl1.setText("Finished")
self.removeUsers(response)
self.progress.setValue(1)
self.showUserInfo()
self.logger.info("Old kernel removal option not chosen")
def removeUsers(self, response):
if response:
self.logger.info("Starting user removal")
self.labels[(6, 1)].setMovie(self.movie)
self.movie.start()
try:
subprocess.Popen(['bash', 'custom-users-to-delete.sh'],
stderr=subprocess.STDOUT,
stdout=subprocess.PIPE)
self.logger.debug(
"user removal completed successfully: [{}]".format(
subprocess.STDOUT))
except subprocess.CalledProcessError, e:
self.logger.error("unable removing user [{}]".format(e.output))
self.movie.stop()
self.labels[(6, 1)].setPixmap(self.pixmap2)
else:
class EasyInstaller(QtGui.QDialog):
def __init__(self, parent=None):
super(EasyInstaller, self).__init__(parent)
self.setWindowTitle("Easy install")
self.list_view = QtGui.QListView(self)
self.list_view.setFixedWidth(380)
self.EditText = QtGui.QLineEdit()
self.EditText.setPlaceholderText("Search for applications")
self.model = QtGui.QStandardItemModel(self.list_view)
self.setFixedSize(600, 350)
self.font = QtGui.QFont()
self.font.setBold(True)
self.font2 = QtGui.QFont()
self.font2.setBold(False)
self.EditText = QtGui.QLineEdit()
self.EditText.setPlaceholderText("Add apps to install")
self.btnRemove = QtGui.QPushButton()
self.btnInstall = QtGui.QPushButton()
self.btnBrowse = QtGui.QPushButton()
self.btnBrowse.setFixedWidth(100)
self.btnBrowse.clicked.connect(self.openBackup)
self.btnBrowse.setText("Open Backup")
self.btnRemove.setText("Remove From List")
self.btnRemove.clicked.connect(self.removeItems)
self.btnInstall.setText("Install Apps")
self.btnInstall.clicked.connect(self.installPackages)
self.btnadd = QtGui.QPushButton(self)
self.btnadd.setText("Add App")
self.btnClose = QtGui.QPushButton()
self.btnClose.setText("Close")
self.btnClose.clicked.connect(self.closeview)
self.btnadd.clicked.connect(self.addItems)
self.btnselect = QtGui.QPushButton()
self.btnselect.setText("Select All")
self.btnselect.clicked.connect(self.selectAll)
self.comboBox = QtGui.QComboBox()
self.comboBox.setVisible(False)
self.comboBox.currentIndexChanged.connect(self.setText)
miniLayout = QtGui.QVBoxLayout()
miniLayout.addWidget(self.EditText)
miniLayout.addWidget(self.comboBox)
horizontalLayout = QtGui.QHBoxLayout()
horizontalLayout.addLayout(miniLayout)
horizontalLayout.addWidget(self.btnadd)
horizontalLayout.addWidget(self.btnBrowse)
horizontalLayout.setAlignment(QtCore.Qt.AlignRight)
horizontalLayout2 = QtGui.QHBoxLayout()
horizontalLayout2.addWidget(self.btnRemove)
horizontalLayout2.addWidget(self.btnselect)
horizontalLayout2.addWidget(self.btnInstall)
horizontalLayout2.addWidget(self.btnClose)
verticalLayout = QtGui.QVBoxLayout(self)
verticalLayout.addLayout(horizontalLayout)
verticalLayout.addWidget(self.list_view)
verticalLayout.addLayout(horizontalLayout2)
self.cache = apt.Cache()
self.isWritten = False
def addItems(self):
package = str(self.EditText.text())
try:
pkg = self.cache[package.strip()]
n = pkg.shortname
v = pkg.versions[0].version
desc = pkg.versions[0].description
name = "{}: {}".format(n, v)
if len(package) > 0 and pkg.is_installed is False:
item = QtGui.QStandardItem(name)
item.setCheckable(True)
item.setSelectable(True)
item.setToolTip((textwrap.fill(desc, 70)))
item.setCheckState(QtCore.Qt.Unchecked)
self.model.appendRow(item)
self.list_view.setModel(self.model)
else:
self.alreadyInstalled(name)
self.EditText.clear()
except KeyError:
self.showMessage(package)
def setText(self):
if self.comboBox.count() > 1:
self.EditText.setText(self.comboBox.currentText())
else:
self.comboBox.setVisible(False)
def removeItems(self):
for row in xrange(self.model.rowCount()):
item = self.model.item(row)
if item and item.checkState() == QtCore.Qt.Checked:
self.model.removeRow(row)
self.removeItems()
def selectAll(self):
model = self.model
for index in xrange(model.rowCount()):
item = model.item(index)
if item.isCheckable() and item.checkState() == QtCore.Qt.Unchecked:
item.setCheckState(QtCore.Qt.Checked)
self.btnselect.setText("Deselect all")
else:
item.setCheckState(QtCore.Qt.Unchecked)
self.btnselect.setText("Select all")
def openBackup(self):
try:
dpath = os.path.abspath(os.path.join('Backup', '../../../'))
backup = QtGui.QFileDialog.getOpenFileName(self, 'Choose Backup', dpath, '(*.rbf)')
if os.path.isfile(backup):
with open(backup, 'r') as bk:
for line in bk:
try:
pkg = self.cache[line.strip()]
n = pkg.shortname
v = pkg.versions[0].version
desc = pkg.versions[0].raw_description
name = "{}: {}".format(n, v)
if len(line) > 0 and pkg.is_installed is False:
item = QtGui.QStandardItem(name)
item.setCheckable(True)
item.setSelectable(True)
item.setToolTip((textwrap.fill(desc, 70)))
item.setCheckState(QtCore.Qt.Unchecked)
self.model.appendRow(item)
self.list_view.setModel(self.model)
self.EditText.clear()
except KeyError:
continue
except IOError:
pass
def installPackages(self):
self.btnInstall.setEnabled(False)
model = self.model
for index in xrange(model.rowCount()):
item = model.item(index)
if self.isWritten:
mode = 'a'
else:
mode = 'w'
self.isWritten = True
if item.isCheckable() and item.checkState() == QtCore.Qt.Checked:
with open('install', mode) as f_out:
to_install = str(item.text()).split(':')[0]
f_out.write('{}\n'.format(to_install))
self.install = Install('install', 'Installing packages', True)
self.install.show()
self.install.exec_()
self.cache.close()
self.btnInstall.setEnabled(True)
self.removeItems()
def closeview(self):
self.cache.close()
self.close()
def alreadyInstalled(self, package):
UsefulTools().showMessage("Package already installed ", "{} is already on your system".format(package),
QtGui.QMessageBox.Information)
def showMessage(self, package):
self.comboBox.clear()
self.comboBox.addItem("Did you mean?")
i = 0
for p in self.cache:
if p.shortname.startswith(package) and len(package) > 0 and i < 12:
i += 1
self.comboBox.addItem(p.shortname)
if self.comboBox.count() > 1:
self.comboBox.setVisible(True)
msg = "The package that you've tried to add is not found in the cache"
msgd = "If you've recently added a ppa containing this package, "\
"please use [EasyPPA - refresh sources] feature, "\
"then try adding the package again."
UsefulTools().showMessage("Package not found", msg, QtGui.QMessageBox.Information, msgd)
class EasyInstaller(QtGui.QDialog):
def __init__(self, parent=None):
super(EasyInstaller, self).__init__(parent)
self.setWindowTitle("Easy install")
self.list_view = QtGui.QListView(self)
self.list_view.setFixedWidth(380)
self.EditText = QtGui.QLineEdit()
self.EditText.setPlaceholderText("Search for applications")
#self.setWindowFlags(QtCore.Qt.Window)
self.model = QtGui.QStandardItemModel(self.list_view)
self.setFixedSize(600, 350)
self.font = QtGui.QFont()
self.font.setBold(True)
self.font2 = QtGui.QFont()
self.font2.setBold(False)
self.EditText = QtGui.QLineEdit()
self.EditText.setPlaceholderText("Add apps to install")
self.btnRemove = QtGui.QPushButton()
self.btnInstall = QtGui.QPushButton()
self.btnBrowse = QtGui.QPushButton()
self.btnBrowse.setFixedWidth(100)
self.btnBrowse.clicked.connect(self.openBackup)
self.btnBrowse.setText("Open Backup")
self.btnRemove.setText("Remove From List")
self.btnRemove.clicked.connect(self.removeItems)
self.btnInstall.setText("Install Apps")
self.btnInstall.clicked.connect(self.installPackages)
self.btnadd = QtGui.QPushButton(self)
self.btnadd.setText("Add App")
self.btnClose = QtGui.QPushButton()
self.btnClose.setText("Close")
self.btnClose.clicked.connect(self.closeview)
self.btnadd.clicked.connect(self.addItems)
self.btnselect = QtGui.QPushButton()
self.btnselect.setText("Select All")
self.btnselect.clicked.connect(self.selectAll)
self.comboBox = QtGui.QComboBox()
self.comboBox.setVisible(False)
self.comboBox.currentIndexChanged.connect(self.setText)
miniLayout = QtGui.QVBoxLayout()
miniLayout.addWidget(self.EditText)
miniLayout.addWidget(self.comboBox)
horizontalLayout = QtGui.QHBoxLayout()
horizontalLayout.addLayout(miniLayout)
horizontalLayout.addWidget(self.btnadd)
horizontalLayout.addWidget(self.btnBrowse)
horizontalLayout.setAlignment(QtCore.Qt.AlignRight)
horizontalLayout2 = QtGui.QHBoxLayout()
horizontalLayout2.addWidget(self.btnRemove)
horizontalLayout2.addWidget(self.btnselect)
horizontalLayout2.addWidget(self.btnInstall)
horizontalLayout2.addWidget(self.btnClose)
verticalLayout = QtGui.QVBoxLayout(self)
verticalLayout.addLayout(horizontalLayout)
verticalLayout.addWidget(self.list_view)
verticalLayout.addLayout(horizontalLayout2)
self.cache = apt.Cache()
self.isWritten = False
def addItems(self):
package = str(self.EditText.text())
try:
pkg = self.cache[package.strip()]
n = pkg.shortname
v = pkg.versions[0].version
desc = pkg.versions[0].description
name = "{}: {}".format(n, v)
if len(package) > 0 and pkg.is_installed is False:
item = QtGui.QStandardItem(name)
item.setCheckable(True)
item.setSelectable(True)
item.setToolTip((textwrap.fill(desc, 70)))
item.setCheckState(QtCore.Qt.Unchecked)
self.model.appendRow(item)
self.list_view.setModel(self.model)
else:
self.alreadyInstalled(name)
self.EditText.clear()
except KeyError:
self.showMessage(package)
def setText(self):
if self.comboBox.count() > 1:
self.EditText.setText(self.comboBox.currentText())
else:
self.comboBox.setVisible(False)
def removeItems(self):
for row in xrange(self.model.rowCount()):
item = self.model.item(row)
if item and item.checkState() == QtCore.Qt.Checked:
self.model.removeRow(row)
self.removeItems()
def selectAll(self):
model = self.model
for index in xrange(model.rowCount()):
item = model.item(index)
if item.isCheckable() and item.checkState() == QtCore.Qt.Unchecked:
item.setCheckState(QtCore.Qt.Checked)
self.btnselect.setText("Deselect all")
else:
item.setCheckState(QtCore.Qt.Unchecked)
self.btnselect.setText("Select all")
def openBackup(self):
try:
dpath = os.path.abspath(os.path.join('Backup', '../../../'))
backup = QtGui.QFileDialog.getOpenFileName(self, 'Choose Backup',
dpath, '(*.rbf)')
if os.path.isfile(backup):
with open(backup, 'r') as bk:
for line in bk:
try:
pkg = self.cache[line.strip()]
n = pkg.shortname
v = pkg.versions[0].version
desc = pkg.versions[0].raw_description
name = "{}: {}".format(n, v)
if len(line) > 0 and pkg.is_installed is False:
item = QtGui.QStandardItem(name)
item.setCheckable(True)
item.setSelectable(True)
item.setToolTip((textwrap.fill(desc, 70)))
item.setCheckState(QtCore.Qt.Unchecked)
self.model.appendRow(item)
self.list_view.setModel(self.model)
self.EditText.clear()
except KeyError:
continue
except IOError:
pass
def installPackages(self):
self.btnInstall.setEnabled(False)
model = self.model
for index in xrange(model.rowCount()):
item = model.item(index)
if self.isWritten:
mode = 'a'
else:
mode = 'w'
self.isWritten = True
if item.isCheckable() and item.checkState() == QtCore.Qt.Checked:
with open('install', mode) as f_out:
to_install = str(item.text()).split(':')[0]
f_out.write('{}\n'.format(to_install))
self.install = Install('install', 'Installing packages', True)
self.install.show()
self.install.exec_()
self.cache.close()
self.btnInstall.setEnabled(True)
self.removeItems()
def closeview(self):
self.cache.close()
self.close()
def alreadyInstalled(self, package):
msg = QtGui.QMessageBox(self)
msg.setWindowTitle("Package already installed")
msg.setIcon(QtGui.QMessageBox.Information)
msg.setText("{} is already on your system".format(package))
msg.exec_()
def showMessage(self, package):
self.comboBox.clear()
self.comboBox.addItem("Did you mean?")
i = 0
for p in self.cache:
if p.shortname.startswith(package) and len(package) > 0 and i < 12:
i += 1
self.comboBox.addItem(p.shortname)
if self.comboBox.count() > 1:
self.comboBox.setVisible(True)
msg = QtGui.QMessageBox(self)
msg.setWindowTitle("Package not found")
msg.setIcon(QtGui.QMessageBox.Information)
msg.setText(
"The package that you've tried to add is not found in the cache")
msg.setDetailedText(
"If you've recently added a ppa containing this package, "
"please use [EasyPPA - refresh sources] feature, "
"then try adding the package again.")
msg.exec_()
class Apply(QDialog):
def __init__(self, file_in, response, rsu, parent=None):
super(Apply, self).__init__(parent)
self.setMinimumSize(400, 250)
self.file_in = file_in
self.response = response
self.rsu = rsu
self.custom_user = bool
self.remaining = 0
self.no_show = False
self.setWindowTitle("Applying")
self.buttonCancel = QPushButton()
self.buttonCancel.setText("Cancel")
self.buttonCancel.clicked.connect(self.finished)
self.progress = QProgressBar()
self.progress2 = QProgressBar()
self.progress2.setVisible(False)
self.lbl1 = QLabel()
gif = os.path.abspath("/usr/lib/resetter/data/icons/chassingarrows.gif")
self.movie = QtGui.QMovie(gif)
self.movie.setScaledSize(QtCore.QSize(20, 20))
self.pixmap = QtGui.QPixmap("/usr/lib/resetter/data/icons/checkmark.png")
self.cuser = '******'
self.pixmap2 = self.pixmap.scaled(20, 20)
verticalLayout = QVBoxLayout(self)
verticalLayout.addWidget(self.lbl1)
verticalLayout.addWidget(self.progress)
verticalLayout.addWidget(self.progress2)
gridLayout = QGridLayout()
self.labels = {}
for i in range(1, 7):
for j in range(1, 3):
self.labels[(i, j)] = QLabel()
self.labels[(i, j)].setMinimumHeight(20)
gridLayout.addWidget(self.labels[(i, j)], i, j)
gridLayout.setAlignment(QtCore.Qt.AlignCenter)
self.labels[(1, 2)].setText("Loading packages")
self.labels[(2, 2)].setText("Removing packages")
self.labels[(3, 2)].setText("Cleaning Up")
self.labels[(4, 2)].setText("Installing packages")
if self.response:
self.labels[(5, 2)].setText("Removing old kernels")
self.labels[(6, 2)].setText("Deleting Users")
else:
self.labels[(5, 2)].setText("Deleting Users")
verticalLayout.addSpacing(20)
verticalLayout.addLayout(gridLayout)
verticalLayout.addWidget(self.buttonCancel, 0, QtCore.Qt.AlignRight)
self.logger = logging.getLogger(__name__)
self.logger.setLevel(logging.DEBUG)
handler = logging.FileHandler('/var/log/resetter/resetter.log')
handler.setLevel(logging.DEBUG)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(funcName)s - %(levelname)s - %(message)s')
handler.setFormatter(formatter)
self.logger.addHandler(handler)
self.progressView = ProgressThread(self.file_in, False)
self.account = AccountDialog()
self.progressView.start_op1.connect(self.updateProgressBar)
self.progressView.aprogress.run_op.connect(self.updateProgressBar2)
self.progressView.iprogress.run_op.connect(self.updateProgressBar2)
self.progressView.start_op.connect(self.showError)
self.addUser1()
@QtCore.pyqtSlot(int, bool)
def updateProgressBar(self, percent, isdone):
self.lbl1.setText("Loading Package List")
self.progress.setValue(percent)
self.labels[(1, 1)].setMovie(self.movie)
self.movie.start()
if isdone:
self.movie.stop()
@QtCore.pyqtSlot(int, bool, str)
def updateProgressBar2(self, percent, isdone, status):
self.progress.setVisible(False)
self.progress2.setVisible(True)
self.labels[(1, 1)].setPixmap(self.pixmap2)
self.lbl1.setText(status)
self.labels[(2, 1)].setMovie(self.movie)
self.movie.start()
self.progress2.setValue(percent)
if isdone:
self.progressView.conclude_op.connect(self.finished)
self.labels[(2, 1)].setPixmap(self.pixmap2)
self.movie.stop()
self.fixBroken()
@QtCore.pyqtSlot(str, str)
def showError(self, error, m_type):
self.movie.stop()
UsefulTools().showMessage(m_type, "Something went wront, please check details.", QMessageBox.Critical,
error)
@QtCore.pyqtSlot()
def finished(self):
self.logger.warning("finished apt operation")
self.progressView.thread1.finished.connect(self.progressView.thread1.exit)
self.progressView.thread2.finished.connect(self.progressView.thread2.exit)
self.progressView.conclude_op.connect(self.progressView.exit)
self.close()
def fixBroken(self):
self.labels[(3, 1)].setMovie(self.movie)
self.movie.start()
self.lbl1.setText("Cleaning up...")
self.logger.info("Cleaning up...")
self.progress2.setRange(0, 0)
self.setCursor(QtCore.Qt.BusyCursor)
process = QtCore.QProcess()
process.finished.connect(lambda: self.onFinished(process.exitCode(), process.exitStatus()))
process.start('bash', ['/usr/lib/resetter/data/scripts/fix-broken.sh'])
def onFinished(self, exit_code, exit_status):
if exit_code or exit_status != 0:
self.progress2.setRange(0, 1)
self.logger.error("fixBroken() finished with exit code: {} and exit_status {}."
.format(exit_code, exit_status))
choice_err = QMessageBox.warning \
(self, "Problems encountered while attempting to clean up. ",
"Please run: <strong>sudo bash /usr/lib/resetter/data/scripts/fix-broken.sh</strong> on a terminal. "
"Once the task is completed, choose 'Yes' to continue.", QMessageBox.Yes | QMessageBox.No)
if choice_err == QMessageBox.Yes:
self.progress2.setRange(0, 1)
self.progress2.setValue(1)
self.unsetCursor()
self.lbl1.setText("Done Cleanup")
self.installPackages()
self.removeOldKernels(self.response)
else:
UsefulTools().showMessage("You have chosen to abort", QMessageBox.Cancel)
QApplication.restoreOverrideCursor()
self.finished()
else:
self.progress2.setRange(0, 1)
self.progress2.setValue(1)
self.logger.debug("Cleanup finished with exit code: {} and exit_status {}.".format(exit_code, exit_status))
self.labels[(3, 1)].setPixmap(self.pixmap2)
self.unsetCursor()
self.lbl1.setText("Done Cleanup")
self.installPackages()
self.removeOldKernels(self.response)
def addUser1(self):
choice = QMessageBox.question \
(self, 'Would you like set your new account?',
"Set your own account?",
QMessageBox.Yes | QMessageBox.No)
if choice == QMessageBox.Yes:
self.custom_user = True
self.show()
self.account.exec_()
self.start()
self.showMinimized()
elif choice == QMessageBox.No:
self.custom_user = False
self.no_show = True
with open('users') as u, open('custom-users-to-delete.sh') as du:
converted_du = []
for line in du:
line = line.split(' ')[-1]
converted_du.append(line)
if len(converted_du) > 0:
diff = set(u).difference(converted_du)
i = len(diff)
else:
i = len(u.read().strip().splitlines())
self.remaining = i
self.start()
def addUser2(self): # determine to add a backup user if all normal users are marked for deletion.
if self.custom_user:
p = subprocess.check_output(['bash', self.cuser])
print(p)
else:
if self.remaining == 0:
self.no_show = False
p = subprocess.check_output(['bash', '/usr/lib/resetter/data/scripts/new-user.sh'])
print(p)
def removeOldKernels(self, response):
if response:
if self.progressView.lineCount('Kernels') > 0:
self.logger.info("Starting kernel removal...")
self.labels[(5, 1)].setMovie(self.movie)
self.setCursor(QtCore.Qt.BusyCursor)
self.progress.setValue(0)
try:
self.logger.info("Removing old kernels...")
self.install = Install("Kernels", "removing old kernels", False)
self.install.show()
self.install.exec_()
self.labels[(5, 1)].setPixmap(self.pixmap2)
self.unsetCursor()
self.lbl1.setText("Finished")
except Exception as arg:
self.logger.error("Kernel removal failed [{}]".format(str(arg)))
print("Sorry, kernel removal failed [{}]".format(str(arg)))
self.removeUsers(response)
self.addUser2()
self.showUserInfo()
self.progress.setValue(1)
else:
self.labels[(5, 1)].setPixmap(self.pixmap2)
self.removeUsers(response)
self.addUser2()
self.showUserInfo()
self.progress.setValue(1)
else:
self.lbl1.setText("Finished")
self.removeUsers(response)
self.addUser2()
self.progress.setValue(1)
self.showUserInfo()
self.logger.info("Old kernel removal option not chosen")
def installPackages(self):
self.logger.info("Starting installations...")
self.labels[(4, 1)].setMovie(self.movie)
if self.progressView.lineCount('custom-install') > 0:
self.install = Install("custom-install", "Installing packages", True)
self.install.show()
self.install.exec_()
self.labels[(4, 1)].setPixmap(self.pixmap2)
else:
self.labels[(4, 1)].setPixmap(self.pixmap2)
def start(self):
self.progressView.start()
def removeSystemUsers(self, rsu):
if rsu:
self.logger.info("Starting user removal")
with open("non-default-users") as f_in, open('custom-users-to-delete.sh', 'a') as output:
for line in f_in:
line = ("userdel -rf ", line)
output.writelines(line)
else:
pass
def removeUsers(self, response):
self.removeSystemUsers(self.rsu)
if response:
self.logger.info("Starting user removal")
self.labels[(6, 1)].setMovie(self.movie)
try:
p = subprocess.check_output(['bash', 'custom-users-to-delete.sh'])
print (p)
except (subprocess.CalledProcessError) as e:
self.logger.error("unable removing user [{}]".format(e.output))
else:
self.logger.debug("user removal completed successfully")
self.labels[(6, 1)].setPixmap(self.pixmap2)
else:
self.logger.info("Starting user removal")
self.labels[(5, 1)].setMovie(self.movie)
try:
p = subprocess.check_output(['bash', 'custom-users-to-delete.sh'])
print(p)
except subprocess.CalledProcessError as e:
self.logger.error("unable removing user [{}]".format(e.output))
else:
self.logger.debug("user removal completed successfully")
self.labels[(5, 1)].setPixmap(self.pixmap2)
def rebootMessage(self):
if os.path.exists(self.cuser):
os.remove(self.cuser)
choice = QMessageBox.information \
(self, 'Please reboot to complete system changes',
"Reboot now?",
QMessageBox.Yes | QMessageBox.No)
if choice == QMessageBox.Yes:
self.logger.info("system rebooted after package removals")
os.system('reboot')
else:
self.logger.info("reboot was delayed.")
def showUserInfo(self):
if not self.no_show:
msg = QMessageBox(self)
msg.setWindowTitle("User Credentials")
msg.setIcon(QMessageBox.Information)
msg.setText("Please use these credentials the next time you log-in")
msg.setInformativeText(
"USERNAME: <b>{}</b><br/> PASSWORD: <b>{}</b>".format(self.account.getUser(),
self.account.getPassword()))
msg.setDetailedText("If you deleted your old user account, "
"this account will be the only local user on your system")
msg.exec_()
self.logger.info("Credential message shown")
self.rebootMessage()
else:
self.rebootMessage()