sface/screens/configscreen.py

   1
   2 from PyQt4.QtCore import *
   3 from PyQt4.QtGui import *
   4
   5 from sface.config import config
   6 from sface.screens.sfascreen import SfaScreen
   7
   8 from sfa.util.version import version_core
   9 from sface.version import version_dict
  10
  11 static_labels = {
  12     'slice' :  [
  13         "Sface : %s (%s)" % (version_dict()['code_tag'], version_dict()['code_url']),
  14         "with (local) SFA libs : %s (%s)" % (version_core()['code_tag'],version_core()['code_url']),
  15         ] ,
  16     'registry': "usual port for registry: 12345",
  17     'slicemgr': ["usual port for slice manager: 12347","usual port for aggregate: 12346"],
  18 }
  19
  20 class ConfigWidget(QWidget):
  21     def __init__(self, parent):
  22         QWidget.__init__(self, parent)
  23         # init can be called several times for when the config dir is changed
  24         self.inited=False
  25         self.init ()
  26
  27     def store_local (self, name, value):
  28         setattr (self, 'widget_'+name, value)
  29     def retrieve_local (self, name):
  30         return getattr (self, 'widget_'+name, None)
  31
  32     def init (self):
  33         # if already inited we just need to set the values
  34         if self.inited:
  35             for (field,msg) in config.field_labels():
  36                 edit = self.retrieve_local(field)
  37                 if isinstance (edit,QCheckBox):
  38                     if config.is_true(config.get(field)): edit.setCheckState (Qt.Checked)
  39                     else: edit.setCheckState (Qt.Unchecked)
  40                 else:
  41                     edit.setText (config.get(field) or "")
  42             return
  43
  44         # otherwise we need to build the whole thing up
  45         glayout = QGridLayout()
  46         row = 0
  47         for (field,msg) in config.field_labels():
  48
  49             if static_labels.has_key(field):
  50                 labels=static_labels[field]
  51                 if not isinstance(labels,list): labels = [ labels, ]
  52                 for label in labels:
  53                     glayout.addWidget(QLabel(label),row,1)
  54                     row+=1
  55             # edit : text or checkbox
  56             default=config.field_default(field)
  57             if isinstance(default,bool):
  58                 edit=QCheckBox(msg)
  59                 self.store_local (field, edit)
  60                 if config.is_true(config.get(field)):
  61                     edit.setCheckState(Qt.Checked)
  62             else:
  63                 edit=QLineEdit(config.get(field) or "", self)
  64                 self.store_local (field, edit)
  65                 edit.setAttribute(Qt.WA_MacShowFocusRect, 0)
  66             setattr(self,field,edit)
  67
  68             glayout.addWidget(QLabel(msg+":",self), row, 0)
  69             glayout.addWidget(edit, row, 1)
  70
  71             row += 1
  72
  73         hlayout = QHBoxLayout()
  74         hlayout.addStretch()
  75         def conf_button (action,label):
  76             button=QPushButton(label, self)
  77             button.setSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum)
  78             hlayout.addWidget(button)
  79             hlayout.addSpacing(10)
  80             self.connect(button, SIGNAL('clicked()'), getattr(self,action))
  81
  82         conf_button ('load','Load Conf. Dir')
  83
  84         # the config dir edit dialog
  85         edit=QLineEdit (config.get_dirname(),self)
  86         self.store_local('config_dirname',edit)
  87         edit.setAttribute(Qt.WA_MacShowFocusRect, 0)
  88         edit.setSizePolicy(QSizePolicy.Expanding,QSizePolicy.Maximum)
  89         self.connect(edit,SIGNAL ('returnPressed()'), self.load)
  90         hlayout.addWidget (edit)
  91         hlayout.addSpacing(10)
  92
  93         conf_button ('apply','Apply Only'),
  94         conf_button ('save','Apply && Save')
  95
  96         layout = QVBoxLayout()
  97         layout.addLayout(glayout)
  98         layout.addLayout(hlayout)
  99         layout.addStretch()
 100         self.setLayout(layout)
 101         self.inited=True
 102
 103
 104     def apply(self):
 105         print 'applying'
 106         for field in config.fields():
 107             widget=getattr(self,field)
 108             if isinstance(widget,QCheckBox):
 109                 if widget.checkState() == Qt.Checked:
 110                     config.set(field, True)
 111                 else:
 112                     config.set(field, False)
 113             else:
 114                 config.set(field, str(widget.text()))
 115
 116         self.parent().setStatus("<font color='green'>Settings loaded for current session</font>", timeout=5000)
 117         config.display("after apply")
 118
 119         self.parent().signalAll('configurationChanged')
 120
 121
 122     def save(self):
 123         self.apply()
 124         config.save_config()
 125         self.parent().setStatus("<font color='green'>Configuration saved in %s !</font>"%config.config_file(), timeout=5000)
 126
 127     # switch to another config dir
 128     def load(self):
 129         # obtain new dor somehow
 130
 131         edit=self.retrieve_local('config_dirname')
 132         newdir=str(edit.text())
 133         newdir+='/'
 134         print 'installing',newdir
 135         config.set_dirname (newdir)
 136         self.init()
 137         self.parent().signalAll('configurationChanged')
 138
 139
 140 class ConfigScreen(SfaScreen):
 141     def __init__(self, parent):
 142         SfaScreen.__init__(self, parent)
 143
 144         widget = ConfigWidget(self)
 145         self.init(widget, "Configure", "Configure the OneLab SFA crawler")
 146