2 from Tools.NumericalTextInput import NumericalTextInput
3 from Tools.Directories import resolveFilename, SCOPE_CONFIG
7 # ConfigElement, the base class of all ConfigElements.
10 # value the current value, usefully encoded.
11 # usually a property which retrieves _value,
12 # and maybe does some reformatting
13 # _value the value as it's going to be saved in the configfile,
14 # though still in non-string form.
15 # this is the object which is actually worked on.
16 # default the initial value. If _value is equal to default,
17 # it will not be stored in the config file
18 # saved_value is a text representation of _value, stored in the config file
20 # and has (at least) the following methods:
21 # save() stores _value into saved_value,
22 # (or stores 'None' if it should not be stored)
23 # load() loads _value from saved_value, or loads
24 # the default if saved_value is 'None' (default)
27 class ConfigElement(object):
31 self.saved_value = None
32 self.save_disabled = False
36 # you need to override this to do input validation
37 def setValue(self, value):
44 value = property(getValue, setValue)
46 # you need to override this if self.value is not a string
47 def fromstring(self, value):
50 # you can overide this for fancy default handling
52 if self.saved_value is None:
53 self.value = self.default
55 self.value = self.fromstring(self.saved_value)
57 def tostring(self, value):
60 # you need to override this if str(self.value) doesn't work
62 if self.save_disabled or self.value == self.default:
63 self.saved_value = None
65 self.saved_value = self.tostring(self.value)
71 if self.saved_value is None and self.value == self.default:
73 return self.tostring(self.value) != self.saved_value
76 for x in self.notifiers:
79 def addNotifier(self, notifier, initial_call = True):
80 assert callable(notifier), "notifiers must be callable"
81 self.notifiers.append(notifier)
84 # do we want to call the notifier
85 # - at all when adding it? (yes, though optional)
86 # - when the default is active? (yes)
87 # - when no value *yet* has been set,
88 # because no config has ever been read (currently yes)
89 # (though that's not so easy to detect.
90 # the entry could just be new.)
94 def disableSave(self):
95 self.save_disabled = True
97 def __call__(self, selected):
98 return self.getMulti(selected)
100 def helpWindow(self):
108 KEY_NUMBERS = range(12, 12+10)
112 def getKeyNumber(key):
113 assert key in KEY_NUMBERS
117 # ConfigSelection is a "one of.."-type.
118 # it has the "choices", usually a list, which contains
119 # (id, desc)-tuples (or just only the ids, in case the id
120 # will be used as description)
122 # all ids MUST be plain strings.
124 class ConfigSelection(ConfigElement):
125 def __init__(self, choices, default = None):
126 ConfigElement.__init__(self)
128 self.description = {}
130 if isinstance(choices, list):
132 if isinstance(x, tuple):
133 self.choices.append(x[0])
134 self.description[x[0]] = x[1]
136 self.choices.append(x)
137 self.description[x] = x
138 elif isinstance(choices, dict):
139 for (key, val) in choices.items():
140 self.choices.append(key)
141 self.description[key] = val
143 assert False, "ConfigSelection choices must be dict or list!"
145 #assert len(self.choices), "you can't have an empty configselection"
146 if len(self.choices) == 0:
148 self.description[""] = ""
151 default = self.choices[0]
153 assert default in self.choices, "default must be in choice list, but " + repr(default) + " is not!"
154 for x in self.choices:
155 assert isinstance(x, str), "ConfigSelection choices must be strings"
157 self.value = self.default = default
159 def setValue(self, value):
160 if value in self.choices:
163 self._value = self.default
167 def tostring(self, val):
173 def setCurrentText(self, text):
174 i = self.choices.index(self.value)
175 del self.description[self.choices[i]]
176 self.choices[i] = text
177 self.description[text] = text
180 value = property(getValue, setValue)
183 return self.choices.index(self.value)
185 index = property(getIndex)
188 def handleKey(self, key):
189 nchoices = len(self.choices)
190 i = self.choices.index(self.value)
192 self.value = self.choices[(i + nchoices - 1) % nchoices]
193 elif key == KEY_RIGHT:
194 self.value = self.choices[(i + 1) % nchoices]
197 descr = self.description[self.value]
202 def getMulti(self, selected):
203 descr = self.description[self.value]
205 return ("text", _(descr))
206 return ("text", descr)
209 def getHTML(self, id):
211 for v in self.choices:
213 checked = 'checked="checked" '
216 res += '<input type="radio" name="' + id + '" ' + checked + 'value="' + v + '">' + self.description[v] + "</input></br>\n"
219 def unsafeAssign(self, value):
220 # setValue does check if value is in choices. This is safe enough.
225 # several customized versions exist for different
228 class ConfigBoolean(ConfigElement):
229 def __init__(self, default = False, descriptions = {False: "false", True: "true"}):
230 ConfigElement.__init__(self)
231 self.descriptions = descriptions
232 self.value = self.default = default
233 def handleKey(self, key):
234 if key in [KEY_LEFT, KEY_RIGHT]:
235 self.value = not self.value
238 descr = self.descriptions[self.value]
243 def getMulti(self, selected):
244 descr = self.descriptions[self.value]
246 return ("text", _(descr))
247 return ("text", descr)
249 def tostring(self, value):
255 def fromstring(self, val):
261 def getHTML(self, id):
263 checked = ' checked="checked"'
266 return '<input type="checkbox" name="' + id + '" value="1" ' + checked + " />"
268 # this is FLAWED. and must be fixed.
269 def unsafeAssign(self, value):
275 class ConfigYesNo(ConfigBoolean):
276 def __init__(self, default = False):
277 ConfigBoolean.__init__(self, default = default, descriptions = {False: _("no"), True: _("yes")})
279 class ConfigOnOff(ConfigBoolean):
280 def __init__(self, default = False):
281 ConfigBoolean.__init__(self, default = default, descriptions = {False: _("off"), True: _("on")})
283 class ConfigEnableDisable(ConfigBoolean):
284 def __init__(self, default = False):
285 ConfigBoolean.__init__(self, default = default, descriptions = {False: _("disable"), True: _("enable")})
287 class ConfigDateTime(ConfigElement):
288 def __init__(self, default, formatstring, increment = 86400):
289 ConfigElement.__init__(self)
290 self.increment = increment
291 self.formatstring = formatstring
292 self.value = self.default = int(default)
294 def handleKey(self, key):
296 self.value = self.value - self.increment
298 self.value = self.value + self.increment
301 return time.strftime(self.formatstring, time.localtime(self.value))
303 def getMulti(self, selected):
304 return ("text", time.strftime(self.formatstring, time.localtime(self.value)))
306 def fromstring(self, val):
309 # *THE* mighty config element class
311 # allows you to store/edit a sequence of values.
312 # can be used for IP-addresses, dates, plain integers, ...
313 # several helper exist to ease this up a bit.
315 class ConfigSequence(ConfigElement):
316 def __init__(self, seperator, limits, censor_char = "", default = None):
317 ConfigElement.__init__(self)
318 assert isinstance(limits, list) and len(limits[0]) == 2, "limits must be [(min, max),...]-tuple-list"
319 assert censor_char == "" or len(censor_char) == 1, "censor char must be a single char (or \"\")"
320 #assert isinstance(default, list), "default must be a list"
321 #assert isinstance(default[0], int), "list must contain numbers"
322 #assert len(default) == len(limits), "length must match"
325 self.seperator = seperator
327 self.censor_char = censor_char
329 self.default = default
330 self.value = copy.copy(default)
332 self.endNotifier = []
337 for i in self._value:
338 max_pos += len(str(self.limits[num][1]))
340 while self._value[num] < self.limits[num][0]:
341 self._value[num] += 1
343 while self._value[num] > self.limits[num][1]:
344 self._value[num] -= 1
348 if self.marked_pos >= max_pos:
349 for x in self.endNotifier:
351 self.marked_pos = max_pos - 1
353 if self.marked_pos < 0:
356 def validatePos(self):
357 if self.marked_pos < 0:
360 total_len = sum([len(str(x[1])) for x in self.limits])
362 if self.marked_pos >= total_len:
363 self.marked_pos = total_len - 1
365 def addEndNotifier(self, notifier):
366 self.endNotifier.append(notifier)
368 def handleKey(self, key):
377 if key in KEY_NUMBERS:
379 for x in self.limits:
380 block_len.append(len(str(x[1])))
382 total_len = sum(block_len)
386 block_len_total = [0, ]
388 pos += block_len[blocknumber]
389 block_len_total.append(pos)
390 if pos - 1 >= self.marked_pos:
395 number = getKeyNumber(key)
397 # length of numberblock
398 number_len = len(str(self.limits[blocknumber][1]))
400 # position in the block
401 posinblock = self.marked_pos - block_len_total[blocknumber]
403 oldvalue = self._value[blocknumber]
404 olddec = oldvalue % 10 ** (number_len - posinblock) - (oldvalue % 10 ** (number_len - posinblock - 1))
405 newvalue = oldvalue - olddec + (10 ** (number_len - posinblock - 1) * number)
407 self._value[blocknumber] = newvalue
415 mPos = self.marked_pos
417 for i in self._value:
418 if len(value): #fixme no heading separator possible
419 value += self.seperator
420 if mPos >= len(value) - 1:
423 if self.censor_char == "":
424 value += ("%0" + str(len(str(self.limits[num][1]))) + "d") % i
426 value += (self.censor_char * len(str(self.limits[num][1])))
431 (value, mPos) = self.genText()
434 def getMulti(self, selected):
435 (value, mPos) = self.genText()
436 # only mark cursor when we are selected
437 # (this code is heavily ink optimized!)
439 return ("mtext"[1-selected:], value, [mPos])
441 return ("text", value)
443 def tostring(self, val):
444 return self.seperator.join([self.saveSingle(x) for x in val])
446 def saveSingle(self, v):
449 def fromstring(self, value):
450 return [int(x) for x in value.split(self.seperator)]
452 class ConfigIP(ConfigSequence):
453 def __init__(self, default):
454 ConfigSequence.__init__(self, seperator = ".", limits = [(0,255),(0,255),(0,255),(0,255)], default = default)
456 def getHTML(self, id):
457 # we definitely don't want leading zeros
458 return '.'.join(["%d" % d for d in self.value])
460 class ConfigMAC(ConfigSequence):
461 def __init__(self, default):
462 ConfigSequence.__init__(self, seperator = ":", limits = [(1,255),(1,255),(1,255),(1,255),(1,255),(1,255)], default = default)
464 class ConfigPosition(ConfigSequence):
465 def __init__(self, default, args):
466 ConfigSequence.__init__(self, seperator = ",", limits = [(0,args[0]),(0,args[1]),(0,args[2]),(0,args[3])], default = default)
468 class ConfigClock(ConfigSequence):
469 def __init__(self, default):
471 t = time.localtime(default)
472 ConfigSequence.__init__(self, seperator = ":", limits = [(0,23),(0,59)], default = [t.tm_hour, t.tm_min])
474 class ConfigInteger(ConfigSequence):
475 def __init__(self, default, limits = (0, 10000000000)):
476 ConfigSequence.__init__(self, seperator = ":", limits = [limits], default = default)
478 # you need to override this to do input validation
479 def setValue(self, value):
480 self._value = [value]
484 return self._value[0]
486 value = property(getValue, setValue)
488 def fromstring(self, value):
491 def tostring(self, value):
494 class ConfigPIN(ConfigInteger):
495 def __init__(self, default, len = 4, censor = ""):
496 assert isinstance(default, int), "ConfigPIN default must be an integer"
499 ConfigSequence.__init__(self, seperator = ":", limits = [(0, (10**len)-1)], censor_char = censor, default = default)
505 class ConfigFloat(ConfigSequence):
506 def __init__(self, default, limits):
507 ConfigSequence.__init__(self, seperator = ".", limits = limits, default = default)
510 return float(self.value[1] / float(self.limits[1][1] + 1) + self.value[0])
512 float = property(getFloat)
514 # an editable text...
515 class ConfigText(ConfigElement, NumericalTextInput):
516 def __init__(self, default = "", fixed_size = True):
517 ConfigElement.__init__(self)
518 NumericalTextInput.__init__(self, nextFunc = self.nextFunc, handleTimeout = False)
521 self.fixed_size = fixed_size
523 self.value = self.default = default
525 def validateMarker(self):
526 if self.marked_pos >= len(self.text):
527 self.marked_pos = len(self.text) - 1
528 if self.marked_pos < 0:
531 #def nextEntry(self):
532 # self.vals[1](self.getConfigPath())
534 def handleKey(self, key):
535 # this will no change anything on the value itself
536 # so we can handle it here in gui element
537 if key == KEY_DELETE:
538 self.text = self.text[0:self.marked_pos] + self.text[self.marked_pos + 1:]
539 elif key == KEY_LEFT:
541 elif key == KEY_RIGHT:
544 elif key in KEY_NUMBERS:
545 number = self.getKey(getKeyNumber(key))
546 self.text = self.text[0:self.marked_pos] + unicode(number) + self.text[self.marked_pos + 1:]
547 elif key == KEY_TIMEOUT:
551 self.validateMarker()
554 def maybeExpand(self):
555 if not self.fixed_size:
556 if self.marked_pos >= len(self.text):
557 self.text = self.text.ljust(len(self.text) + 1)
562 self.validateMarker()
566 return self.text.encode("utf-8")
568 def setValue(self, val):
570 self.text = val.decode("utf-8")
571 except UnicodeDecodeError:
575 value = property(getValue, setValue)
576 _value = property(getValue, setValue)
581 def getMulti(self, selected):
582 return ("mtext"[1-selected:], self.value, [self.marked_pos])
584 def helpWindow(self):
585 from Screens.NumericalTextInputHelpDialog import NumericalTextInputHelpDialog
586 return (NumericalTextInputHelpDialog,self)
588 def getHTML(self, id):
589 return '<input type="text" name="' + id + '" value="' + self.value + '" /><br>\n'
591 def unsafeAssign(self, value):
592 self.value = str(value)
595 class ConfigSlider(ConfigElement):
596 def __init__(self, default = 0, increment = 1, limits = (0, 100)):
597 ConfigElement.__init__(self)
598 self.value = self.default = default
601 self.increment = increment
603 def checkValues(self):
604 if self.value < self.min:
605 self.value = self.min
607 if self.value > self.max:
608 self.value = self.max
610 def handleKey(self, key):
612 self.value -= self.increment
613 elif key == KEY_RIGHT:
614 self.value += self.increment
622 return "%d / %d" % (self.value, self.max)
624 def getMulti(self, selected):
626 return ("slider", self.value, self.max)
628 def fromstring(self, value):
631 # a satlist. in fact, it's a ConfigSelection.
632 class ConfigSatlist(ConfigSelection):
633 def __init__(self, list, default = None):
634 if default is not None:
635 default = str(default)
636 ConfigSelection.__init__(self, choices = [(str(orbpos), desc) for (orbpos, desc, flags) in list], default = default)
638 def getOrbitalPosition(self):
641 return int(self.value)
643 orbital_position = property(getOrbitalPosition)
646 class ConfigNothing(ConfigSelection):
648 ConfigSelection.__init__(self, choices = [""])
650 # until here, 'saved_value' always had to be a *string*.
651 # now, in ConfigSubsection, and only there, saved_value
652 # is a dict, essentially forming a tree.
654 # config.foo.bar=True
655 # config.foobar=False
658 # config.saved_value == {"foo": {"bar": "True"}, "foobar": "False"}
662 class ConfigSubsectionContent(object):
665 # we store a backup of the loaded configuration
666 # data in self.stored_values, to be able to deploy
667 # them when a new config element will be added,
668 # so non-default values are instantly available
670 # A list, for example:
671 # config.dipswitches = ConfigSubList()
672 # config.dipswitches.append(ConfigYesNo())
673 # config.dipswitches.append(ConfigYesNo())
674 # config.dipswitches.append(ConfigYesNo())
675 class ConfigSubList(list, object):
677 object.__init__(self)
679 self.stored_values = {}
689 def getSavedValue(self):
691 for i in range(len(self)):
692 sv = self[i].saved_value
697 def setSavedValue(self, values):
698 self.stored_values = dict(values)
699 for (key, val) in self.stored_values.items():
700 if int(key) < len(self):
701 self[int(key)].saved_value = val
703 saved_value = property(getSavedValue, setSavedValue)
705 def append(self, item):
707 list.append(self, item)
708 if i in self.stored_values:
709 item.saved_value = self.stored_values[i]
714 for index in range(len(self)):
715 res[str(index)] = self[index]
718 # same as ConfigSubList, just as a dictionary.
719 # care must be taken that the 'key' has a proper
720 # str() method, because it will be used in the config
722 class ConfigSubDict(dict, object):
724 object.__init__(self)
726 self.stored_values = {}
729 for x in self.values():
733 for x in self.values():
736 def getSavedValue(self):
738 for (key, val) in self.items():
739 if val.saved_value is not None:
740 res[str(key)] = val.saved_value
743 def setSavedValue(self, values):
744 self.stored_values = dict(values)
745 for (key, val) in self.items():
746 if str(key) in self.stored_values:
747 val = self.stored_values[str(key)]
749 saved_value = property(getSavedValue, setSavedValue)
751 def __setitem__(self, key, item):
752 dict.__setitem__(self, key, item)
753 if str(key) in self.stored_values:
754 item.saved_value = self.stored_values[str(key)]
760 # Like the classes above, just with a more "native"
763 # some evil stuff must be done to allow instant
764 # loading of added elements. this is why this class
767 # we need the 'content' because we overwrite
769 # If you don't understand this, try adding
770 # __setattr__ to a usual exisiting class and you will.
771 class ConfigSubsection(object):
773 object.__init__(self)
774 self.__dict__["content"] = ConfigSubsectionContent()
775 self.content.items = { }
776 self.content.stored_values = { }
778 def __setattr__(self, name, value):
779 if name == "saved_value":
780 return self.setSavedValue(value)
781 assert isinstance(value, ConfigSubsection) or isinstance(value, ConfigElement) or isinstance(value, ConfigSubList) or isinstance(value, ConfigSubDict), "ConfigSubsections can only store ConfigSubsections, ConfigSubLists, ConfigSubDicts or ConfigElements"
782 self.content.items[name] = value
783 if name in self.content.stored_values:
784 #print "ok, now we have a new item,", name, "and have the following value for it:", self.content.stored_values[name]
785 value.saved_value = self.content.stored_values[name]
788 def __getattr__(self, name):
789 return self.content.items[name]
791 def getSavedValue(self):
792 res = self.content.stored_values
793 for (key, val) in self.content.items.items():
794 if val.saved_value is not None:
795 res[key] = val.saved_value
801 def setSavedValue(self, values):
802 values = dict(values)
804 self.content.stored_values = values
806 for (key, val) in self.content.items.items():
808 val.setSavedValue(values[key])
810 saved_value = property(getSavedValue, setSavedValue)
813 for x in self.content.items.values():
817 for x in self.content.items.values():
821 return self.content.items
823 # the root config object, which also can "pickle" (=serialize)
824 # down the whole config tree.
826 # we try to keep non-existing config entries, to apply them whenever
827 # a new config entry is added to a subsection
828 # also, non-existing config entries will be saved, so they won't be
829 # lost when a config entry disappears.
830 class Config(ConfigSubsection):
832 ConfigSubsection.__init__(self)
834 def pickle_this(self, prefix, topickle, result):
835 for (key, val) in topickle.items():
836 name = prefix + "." + key
838 if isinstance(val, dict):
839 self.pickle_this(name, val, result)
840 elif isinstance(val, tuple):
841 result.append(name + "=" + val[0]) # + " ; " + val[1])
843 result.append(name + "=" + val)
847 self.pickle_this("config", self.saved_value, result)
848 return '\n'.join(result) + "\n"
850 def unpickle(self, lines):
853 if not len(l) or l[0] == '#':
857 val = l[n+1:].strip()
859 names = l[:n].split('.')
860 # if val.find(' ') != -1:
861 # val = val[:val.find(' ')]
866 base = base.setdefault(n, {})
868 base[names[-1]] = val
870 # we inherit from ConfigSubsection, so ...
871 #object.__setattr__(self, "saved_value", tree["config"])
873 self.setSavedValue(tree["config"])
875 def saveToFile(self, filename):
876 f = open(filename, "w")
877 f.write(self.pickle())
880 def loadFromFile(self, filename):
881 f = open(filename, "r")
882 self.unpickle(f.readlines())
886 config.misc = ConfigSubsection()
889 CONFIG_FILE = resolveFilename(SCOPE_CONFIG, "settings")
893 config.loadFromFile(self.CONFIG_FILE)
895 print "unable to load config (%s), assuming defaults..." % str(e)
899 config.saveToFile(self.CONFIG_FILE)
901 def __resolveValue(self, pickles, cmap):
902 if cmap.has_key(pickles[0]):
904 return self.__resolveValue(pickles[1:], cmap[pickles[0]].dict())
906 return str(cmap[pickles[0]].value)
909 def getResolvedKey(self, key):
910 names = key.split('.')
912 if names[0] == "config":
913 ret=self.__resolveValue(names[1:], config.content.items)
916 print "getResolvedKey", key, "failed !! (Typo??)"
920 element.disableSave()
923 configfile = ConfigFile()
927 def getConfigListEntry(*args):
928 assert len(args) > 1, "getConfigListEntry needs a minimum of two arguments (descr, configElement)"
934 #config.bla = ConfigSubsection()
935 #config.bla.test = ConfigYesNo()
936 #config.nim = ConfigSubList()
937 #config.nim.append(ConfigSubsection())
938 #config.nim[0].bla = ConfigYesNo()
939 #config.nim.append(ConfigSubsection())
940 #config.nim[1].bla = ConfigYesNo()
941 #config.nim[1].blub = ConfigYesNo()
942 #config.arg = ConfigSubDict()
943 #config.arg["Hello"] = ConfigYesNo()
945 #config.arg["Hello"].handleKey(KEY_RIGHT)
946 #config.arg["Hello"].handleKey(KEY_RIGHT)
952 #print config.pickle()