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