update globalsettingsdlg for new API

This commit is contained in:
Marta Marczykowska-Górecka 2017-06-25 16:39:30 +02:00
parent 812c71f855
commit 5daa35a301
No known key found for this signature in database
GPG Key ID: 9A752C30B26FD04B
2 changed files with 52 additions and 106 deletions

View File

@ -7,7 +7,7 @@
<x>0</x> <x>0</x>
<y>0</y> <y>0</y>
<width>678</width> <width>678</width>
<height>318</height> <height>331</height>
</rect> </rect>
</property> </property>
<property name="windowTitle"> <property name="windowTitle">
@ -161,13 +161,6 @@
</property> </property>
</widget> </widget>
</item> </item>
<item row="2" column="0" colspan="2">
<widget class="QCheckBox" name="dispvm_in_memory">
<property name="text">
<string>Keep dispVM in memory</string>
</property>
</widget>
</item>
</layout> </layout>
</widget> </widget>
</item> </item>

View File

@ -1,4 +1,4 @@
#!/usr/bin/python2 #!/usr/bin/python3
# #
# The Qubes OS Project, http://www.qubes-os.org # The Qubes OS Project, http://www.qubes-os.org
# #
@ -26,29 +26,14 @@ import os
from PyQt4.QtCore import * from PyQt4.QtCore import *
from PyQt4.QtGui import * from PyQt4.QtGui import *
from qubes.qubes import QubesVmCollection from qubesadmin import Qubes
from qubes.qubes import QubesException
from qubes.qubes import QubesDaemonPidfile
from qubes.qubes import QubesHost
from qubes.qubes import system_path
import qubesmanager.resources_rc from qubesmanager.ui_globalsettingsdlg import *
from pyinotify import WatchManager, Notifier, ThreadedNotifier, EventsCodes, ProcessEvent from configparser import ConfigParser
from qubesadmin.utils import parse_size, updates_vms_status
from qubesadmin.vm import TemplateVM
import time
import threading
from operator import itemgetter
from ui_globalsettingsdlg import *
from ConfigParser import SafeConfigParser
from qubes.qubesutils import parse_size
from qubes.qubesutils import updates_dom0_toggle,updates_vms_toggle,\
updates_dom0_status,updates_vms_status
from qubes import qmemman_algo
dont_keep_dvm_in_memory_path = '/var/lib/qubes/dvmdata/dont-use-shm'
qmemman_config_path = '/etc/qubes/qmemman.conf' qmemman_config_path = '/etc/qubes/qmemman.conf'
@ -70,14 +55,14 @@ class GlobalSettingsWindow(Ui_GlobalSettings, QDialog):
self.__init_mem_defaults__() self.__init_mem_defaults__()
self.__init_updates__() self.__init_updates__()
def __init_system_defaults__(self): def __init_system_defaults__(self):
#updatevm and clockvm # updatevm and clockvm
all_vms = [vm for vm in self.qvm_collection.values() if not all_vms = [vm for vm in self.qvm_collection.domains
vm.internal and vm.qid != 0] if (not vm.features.get('internal', False)) and vm.qid != 0]
self.updatevm_idx = -1 self.updatevm_idx = -1
current_update_vm = self.qvm_collection.get_updatevm_vm() current_update_vm = self.qvm_collection.updatevm
for (i, vm) in enumerate(all_vms): for (i, vm) in enumerate(all_vms):
text = vm.name text = vm.name
if vm is current_update_vm: if vm is current_update_vm:
@ -89,10 +74,10 @@ class GlobalSettingsWindow(Ui_GlobalSettings, QDialog):
self.updatevm_idx = len(all_vms) self.updatevm_idx = len(all_vms)
self.update_vm_combo.setCurrentIndex(self.updatevm_idx) self.update_vm_combo.setCurrentIndex(self.updatevm_idx)
#clockvm # clockvm
self.clockvm_idx = -1 self.clockvm_idx = -1
current_clock_vm = self.qvm_collection.get_clockvm_vm() current_clock_vm = self.qvm_collection.clockvm
for (i, vm) in enumerate(all_vms): for (i, vm) in enumerate(all_vms):
text = vm.name text = vm.name
if vm is current_clock_vm: if vm is current_clock_vm:
@ -104,11 +89,11 @@ class GlobalSettingsWindow(Ui_GlobalSettings, QDialog):
self.clockvm_idx = len(all_vms) self.clockvm_idx = len(all_vms)
self.clock_vm_combo.setCurrentIndex(self.clockvm_idx) self.clock_vm_combo.setCurrentIndex(self.clockvm_idx)
#default netvm # default netvm
netvms = [vm for vm in all_vms if vm.is_netvm()] netvms = [vm for vm in all_vms if getattr(vm, 'provides_network', False)]
self.netvm_idx = -1 self.netvm_idx = -1
current_netvm = self.qvm_collection.get_default_netvm() current_netvm = self.qvm_collection.default_netvm
for (i, vm) in enumerate(netvms): for (i, vm) in enumerate(netvms):
text = vm.name text = vm.name
if vm is current_netvm: if vm is current_netvm:
@ -119,10 +104,10 @@ class GlobalSettingsWindow(Ui_GlobalSettings, QDialog):
self.default_netvm_combo.setCurrentIndex(self.netvm_idx) self.default_netvm_combo.setCurrentIndex(self.netvm_idx)
#default template #default template
templates = [vm for vm in all_vms if vm.is_template()] templates = [vm for vm in all_vms if isinstance(vm, TemplateVM)]
self.template_idx = -1 self.template_idx = -1
current_template = self.qvm_collection.get_default_template() current_template = self.qvm_collection.default_template
for (i, vm) in enumerate(templates): for (i, vm) in enumerate(templates):
text = vm.name text = vm.name
if vm is current_template: if vm is current_template:
@ -137,49 +122,46 @@ class GlobalSettingsWindow(Ui_GlobalSettings, QDialog):
if self.update_vm_combo.currentIndex() != self.updatevm_idx: if self.update_vm_combo.currentIndex() != self.updatevm_idx:
updatevm_name = str(self.update_vm_combo.currentText()) updatevm_name = str(self.update_vm_combo.currentText())
updatevm_name = updatevm_name.split(' ')[0] updatevm_name = updatevm_name.split(' ')[0]
updatevm = self.qvm_collection.get_vm_by_name(updatevm_name) updatevm = self.qvm_collection.domains[updatevm_name]
self.qvm_collection.set_updatevm_vm(updatevm) self.qvm_collection.updatevm = updatevm
self.anything_changed = True
#clockvm #clockvm
if self.clock_vm_combo.currentIndex() != self.clockvm_idx: if self.clock_vm_combo.currentIndex() != self.clockvm_idx:
clockvm_name = str(self.clock_vm_combo.currentText()) clockvm_name = str(self.clock_vm_combo.currentText())
clockvm_name = clockvm_name.split(' ')[0] clockvm_name = clockvm_name.split(' ')[0]
clockvm = self.qvm_collection.get_vm_by_name(clockvm_name) clockvm = self.qvm_collection.domains[clockvm_name]
self.qvm_collection.set_clockvm_vm(clockvm) self.qvm_collection.clockvm = clockvm
self.anything_changed = True
#default netvm #default netvm
if self.default_netvm_combo.currentIndex() != self.netvm_idx: if self.default_netvm_combo.currentIndex() != self.netvm_idx:
name = str(self.default_netvm_combo.currentText()) name = str(self.default_netvm_combo.currentText())
name = name.split(' ')[0] name = name.split(' ')[0]
vm = self.qvm_collection.get_vm_by_name(name) vm = self.qvm_collection.domains[name]
self.qvm_collection.set_default_netvm(vm) self.qvm_collection.default_netvm = vm
self.anything_changed = True
#default template #default template
if self.default_template_combo.currentIndex() != self.template_idx: if self.default_template_combo.currentIndex() != self.template_idx:
name = str(self.default_template_combo.currentText()) name = str(self.default_template_combo.currentText())
name = name.split(' ')[0] name = name.split(' ')[0]
vm = self.qvm_collection.get_vm_by_name(name) vm = self.qvm_collection.domains[name]
self.qvm_collection.set_default_template(vm) self.qvm_collection.default_template = vm
self.anything_changed = True
def __init_kernel_defaults__(self): def __init_kernel_defaults__(self):
kernel_list = [] kernel_list = []
for k in os.listdir(system_path["qubes_kernels_base_dir"]): # TODO system_path["qubes_kernels_base_dir"] idea: qubes.pulls['linux-kernel'].volumes
for k in os.listdir('/var/lib/qubes/vm-kernels'):
kernel_list.append(k) kernel_list.append(k)
self.kernel_idx = 0 self.kernel_idx = 0
for (i, k) in enumerate(kernel_list): for (i, k) in enumerate(kernel_list):
text = k text = k
if k == self.qvm_collection.get_default_kernel(): if k == self.qvm_collection.default_kernel:
text += self.tr(" (current)") text += self.tr(" (current)")
self.kernel_idx = i self.kernel_idx = i
self.default_kernel_combo.insertItem(i, text) self.default_kernel_combo.insertItem(i, text)
@ -190,16 +172,15 @@ class GlobalSettingsWindow(Ui_GlobalSettings, QDialog):
kernel = str(self.default_kernel_combo.currentText()) kernel = str(self.default_kernel_combo.currentText())
kernel = kernel.split(' ')[0] kernel = kernel.split(' ')[0]
self.qvm_collection.set_default_kernel(kernel) self.qvm_collection.default_kernel = kernel
self.anything_changed = True
def __init_mem_defaults__(self): def __init_mem_defaults__(self):
#qmemman settings #qmemman settings
self.qmemman_config = SafeConfigParser() self.qmemman_config = ConfigParser()
self.vm_min_mem_val = str(qmemman_algo.MIN_PREFMEM) self.vm_min_mem_val = '200MiB' #str(qmemman_algo.MIN_PREFMEM)
self.dom0_mem_boost_val = str(qmemman_algo.DOM0_MEM_BOOST) self.dom0_mem_boost_val = '350MiB' #str(qmemman_algo.DOM0_MEM_BOOST)
self.qmemman_config.read(qmemman_config_path) self.qmemman_config.read(qmemman_config_path)
if self.qmemman_config.has_section('global'): if self.qmemman_config.has_section('global'):
@ -212,10 +193,6 @@ class GlobalSettingsWindow(Ui_GlobalSettings, QDialog):
self.min_vm_mem.setValue(self.vm_min_mem_val/1024/1024) self.min_vm_mem.setValue(self.vm_min_mem_val/1024/1024)
self.dom0_mem_boost.setValue(self.dom0_mem_boost_val/1024/1024) self.dom0_mem_boost.setValue(self.dom0_mem_boost_val/1024/1024)
#keep dispvm in memory
exists = os.path.exists(dont_keep_dvm_in_memory_path)
self.dispvm_in_memory.setChecked( not exists)
def __apply_mem_defaults__(self): def __apply_mem_defaults__(self):
@ -233,7 +210,7 @@ class GlobalSettingsWindow(Ui_GlobalSettings, QDialog):
self.qmemman_config.add_section('global') self.qmemman_config.add_section('global')
self.qmemman_config.set('global', 'vm-min-mem', current_min_vm_mem) self.qmemman_config.set('global', 'vm-min-mem', current_min_vm_mem)
self.qmemman_config.set('global', 'dom0-mem-boost', current_dom0_mem_boost) self.qmemman_config.set('global', 'dom0-mem-boost', current_dom0_mem_boost)
self.qmemman_config.set('global', 'cache-margin-factor', str(qmemman_algo.CACHE_FACTOR)) self.qmemman_config.set('global', 'cache-margin-factor', str(1.3)) # removed qmemman_algo.CACHE_FACTOR
qmemman_config_file = open(qmemman_config_path, 'a') qmemman_config_file = open(qmemman_config_path, 'a')
self.qmemman_config.write(qmemman_config_file) self.qmemman_config.write(qmemman_config_file)
@ -268,22 +245,11 @@ class GlobalSettingsWindow(Ui_GlobalSettings, QDialog):
qmemman_config_file.writelines(config_lines) qmemman_config_file.writelines(config_lines)
qmemman_config_file.close() qmemman_config_file.close()
self.anything_changed = True
#keep dispvm in memory
was_checked = not os.path.exists(dont_keep_dvm_in_memory_path)
if was_checked != self.dispvm_in_memory.isChecked():
if was_checked:
#touch file
open(dont_keep_dvm_in_memory_path, 'w').close()
else:
#rm file
os.remove(dont_keep_dvm_in_memory_path)
self.anything_changed = True
def __init_updates__(self): def __init_updates__(self):
self.updates_val = False self.updates_val = False
self.updates_dom0_val = updates_dom0_status(self.qvm_collection) # TODO updates_dom0_status(self.qvm_collection)
self.updates_dom0_val = True
self.updates_dom0.setChecked(self.updates_dom0_val) self.updates_dom0.setChecked(self.updates_dom0_val)
updates_vms = updates_vms_status(self.qvm_collection) updates_vms = updates_vms_status(self.qvm_collection)
if updates_vms is None: if updates_vms is None:
@ -293,33 +259,28 @@ class GlobalSettingsWindow(Ui_GlobalSettings, QDialog):
def __apply_updates__(self): def __apply_updates__(self):
if self.updates_dom0.isChecked() != self.updates_dom0_val: if self.updates_dom0.isChecked() != self.updates_dom0_val:
updates_dom0_toggle(self.qvm_collection, self.updates_dom0.isChecked()) # TODO updates_dom0_toggle(self.qvm_collection, self.updates_dom0.isChecked())
raise NotImplementedError('Toggle dom0 updates not implemented')
if self.updates_vm.checkState() != Qt.PartiallyChecked: if self.updates_vm.checkState() != Qt.PartiallyChecked:
self.anything_changed = True for vm in self.qvm_collection.domains:
updates_vms_toggle(self.qvm_collection, self.updates_vm vm.features['check-updates'] = bool(self.updates_vm.checkState())
.isChecked())
def reject(self): def reject(self):
self.done(0) self.done(0)
def save_and_apply(self): def save_and_apply(self):
self.qvm_collection.lock_db_for_writing()
self.anything_changed = False
self.__apply_system_defaults__() self.__apply_system_defaults__()
self.__apply_kernel_defaults__() self.__apply_kernel_defaults__()
self.__apply_mem_defaults__() self.__apply_mem_defaults__()
self.__apply_updates__() self.__apply_updates__()
if self.anything_changed == True:
self.qvm_collection.save()
self.qvm_collection.unlock_db()
# Bases on the original code by: # Bases on the original code by:
# Copyright (c) 2002-2007 Pascal Varet <p.varet@gmail.com> # Copyright (c) 2002-2007 Pascal Varet <p.varet@gmail.com>
def handle_exception( exc_type, exc_value, exc_traceback ): def handle_exception( exc_type, exc_value, exc_traceback ):
import sys
import os.path import os.path
import traceback import traceback
@ -337,31 +298,23 @@ def handle_exception( exc_type, exc_value, exc_traceback ):
def main(): def main():
global qubes_host global qtapp
qubes_host = QubesHost() qtapp = QApplication(sys.argv)
qtapp.setOrganizationName("The Qubes Project")
global app qtapp.setOrganizationDomain("http://qubes-os.org")
app = QApplication(sys.argv) qtapp.setApplicationName("Qubes Global Settings")
app.setOrganizationName("The Qubes Project")
app.setOrganizationDomain("http://qubes-os.org")
app.setApplicationName("Qubes Global Settings")
sys.excepthook = handle_exception sys.excepthook = handle_exception
qvm_collection = QubesVmCollection() app = Qubes()
qvm_collection.lock_db_for_reading()
qvm_collection.load()
qvm_collection.unlock_db()
global global_window global global_window
global_window = GlobalSettingsWindow() global_window = GlobalSettingsWindow(qtapp, app)
global_window.show() global_window.show()
app.exec_() qtapp.exec_()
app.exit() qtapp.exit()
if __name__ == "__main__": if __name__ == "__main__":
main() main()