2 from enigma import getPrevAsciiCode
3 from Tools.NumericalTextInput import NumericalTextInput
4 from Tools.Directories import resolveFilename, SCOPE_CONFIG
5 from Components.Harddisk import harddiskmanager
9 # ConfigElement, the base class of all ConfigElements.
12 # value the current value, usefully encoded.
13 # usually a property which retrieves _value,
14 # and maybe does some reformatting
15 # _value the value as it's going to be saved in the configfile,
16 # though still in non-string form.
17 # this is the object which is actually worked on.
18 # default the initial value. If _value is equal to default,
19 # it will not be stored in the config file
20 # saved_value is a text representation of _value, stored in the config file
22 # and has (at least) the following methods:
23 # save() stores _value into saved_value,
24 # (or stores 'None' if it should not be stored)
25 # load() loads _value from saved_value, or loads
26 # the default if saved_value is 'None' (default)
29 class ConfigElement(object):
33 self.saved_value = None
34 self.last_value = None
35 self.save_disabled = False
37 self.notifiers_final = []
39 self.callNotifiersOnSaveAndCancel = False
41 # you need to override this to do input validation
42 def setValue(self, value):
49 value = property(getValue, setValue)
51 # you need to override this if self.value is not a string
52 def fromstring(self, value):
55 # you can overide this for fancy default handling
59 self.value = self.default
61 self.value = self.fromstring(sv)
63 def tostring(self, value):
66 # you need to override this if str(self.value) doesn't work
68 if self.save_disabled or self.value == self.default:
69 self.saved_value = None
71 self.saved_value = self.tostring(self.value)
72 if self.callNotifiersOnSaveAndCancel:
77 if self.callNotifiersOnSaveAndCancel:
82 if sv is None and self.value == self.default:
84 return self.tostring(self.value) != sv
87 for x in self.notifiers:
90 def changedFinal(self):
91 for x in self.notifiers_final:
94 def addNotifier(self, notifier, initial_call = True, immediate_feedback = True):
95 assert callable(notifier), "notifiers must be callable"
96 if immediate_feedback:
97 self.notifiers.append(notifier)
99 self.notifiers_final.append(notifier)
101 # do we want to call the notifier
102 # - at all when adding it? (yes, though optional)
103 # - when the default is active? (yes)
104 # - when no value *yet* has been set,
105 # because no config has ever been read (currently yes)
106 # (though that's not so easy to detect.
107 # the entry could just be new.)
111 def disableSave(self):
112 self.save_disabled = True
114 def __call__(self, selected):
115 return self.getMulti(selected)
117 def onSelect(self, session):
120 def onDeselect(self, session):
121 if not self.last_value == self.value:
123 self.last_value = self.value
135 KEY_NUMBERS = range(12, 12+10)
139 def getKeyNumber(key):
140 assert key in KEY_NUMBERS
144 # ConfigSelection is a "one of.."-type.
145 # it has the "choices", usually a list, which contains
146 # (id, desc)-tuples (or just only the ids, in case the id
147 # will be used as description)
149 # all ids MUST be plain strings.
151 class ConfigSelection(ConfigElement):
152 def __init__(self, choices, default = None):
153 ConfigElement.__init__(self)
155 self.setChoices(choices, default)
156 self.last_value = self._value
158 def setChoices(self, choices, default = None):
160 self.description = {}
162 if isinstance(choices, list):
164 if isinstance(x, tuple):
165 self.choices.append(x[0])
166 self.description[x[0]] = x[1]
168 self.choices.append(x)
169 self.description[x] = x
170 elif isinstance(choices, dict):
171 for (key, val) in choices.items():
172 self.choices.append(key)
173 self.description[key] = val
175 assert False, "ConfigSelection choices must be dict or list!"
177 #assert len(self.choices), "you can't have an empty configselection"
178 if len(self.choices) == 0:
180 self.description[""] = ""
183 default = self.choices[0]
185 assert default in self.choices, "default must be in choice list, but " + repr(default) + " is not!"
186 # for x in self.choices:
187 # assert isinstance(x, str), "ConfigSelection choices must be strings"
188 self.default = default
190 if self.value == None or not self.value in self.choices:
193 def setValue(self, value):
194 if value in self.choices:
197 self._value = self.default
201 def tostring(self, val):
207 def setCurrentText(self, text):
208 i = self.choices.index(self.value)
209 del self.description[self.choices[i]]
210 self.choices[i] = text
211 self.description[text] = text
214 value = property(getValue, setValue)
217 return self.choices.index(self.value)
219 index = property(getIndex)
222 def handleKey(self, key):
223 nchoices = len(self.choices)
224 i = self.choices.index(self.value)
226 self.value = self.choices[(i + nchoices - 1) % nchoices]
227 elif key == KEY_RIGHT:
228 self.value = self.choices[(i + 1) % nchoices]
229 elif key == KEY_HOME:
230 self.value = self.choices[0]
232 self.value = self.choices[nchoices - 1]
234 def selectNext(self):
235 nchoices = len(self.choices)
236 i = self.choices.index(self.value)
237 self.value = self.choices[(i + 1) % nchoices]
240 descr = self.description[self.value]
245 def getMulti(self, selected):
246 descr = self.description[self.value]
248 return ("text", _(descr))
249 return ("text", descr)
252 def getHTML(self, id):
254 for v in self.choices:
256 checked = 'checked="checked" '
259 res += '<input type="radio" name="' + id + '" ' + checked + 'value="' + v + '">' + self.description[v] + "</input></br>\n"
262 def unsafeAssign(self, value):
263 # setValue does check if value is in choices. This is safe enough.
268 # several customized versions exist for different
271 class ConfigBoolean(ConfigElement):
272 def __init__(self, default = False, descriptions = {False: "false", True: "true"}):
273 ConfigElement.__init__(self)
274 self.descriptions = descriptions
275 self.value = self.last_value = self.default = default
277 def handleKey(self, key):
278 if key in [KEY_LEFT, KEY_RIGHT]:
279 self.value = not self.value
280 elif key == KEY_HOME:
286 descr = self.descriptions[self.value]
291 def getMulti(self, selected):
292 descr = self.descriptions[self.value]
294 return ("text", _(descr))
295 return ("text", descr)
297 def tostring(self, value):
303 def fromstring(self, val):
309 def getHTML(self, id):
311 checked = ' checked="checked"'
314 return '<input type="checkbox" name="' + id + '" value="1" ' + checked + " />"
316 # this is FLAWED. and must be fixed.
317 def unsafeAssign(self, value):
323 def onDeselect(self, session):
324 if not self.last_value == self.value:
326 self.last_value = self.value
328 class ConfigYesNo(ConfigBoolean):
329 def __init__(self, default = False):
330 ConfigBoolean.__init__(self, default = default, descriptions = {False: _("no"), True: _("yes")})
332 class ConfigOnOff(ConfigBoolean):
333 def __init__(self, default = False):
334 ConfigBoolean.__init__(self, default = default, descriptions = {False: _("off"), True: _("on")})
336 class ConfigEnableDisable(ConfigBoolean):
337 def __init__(self, default = False):
338 ConfigBoolean.__init__(self, default = default, descriptions = {False: _("disable"), True: _("enable")})
340 class ConfigDateTime(ConfigElement):
341 def __init__(self, default, formatstring, increment = 86400):
342 ConfigElement.__init__(self)
343 self.increment = increment
344 self.formatstring = formatstring
345 self.value = self.last_value = self.default = int(default)
347 def handleKey(self, key):
349 self.value = self.value - self.increment
350 elif key == KEY_RIGHT:
351 self.value = self.value + self.increment
352 elif key == KEY_HOME or key == KEY_END:
353 self.value = self.default
356 return time.strftime(self.formatstring, time.localtime(self.value))
358 def getMulti(self, selected):
359 return ("text", time.strftime(self.formatstring, time.localtime(self.value)))
361 def fromstring(self, val):
364 # *THE* mighty config element class
366 # allows you to store/edit a sequence of values.
367 # can be used for IP-addresses, dates, plain integers, ...
368 # several helper exist to ease this up a bit.
370 class ConfigSequence(ConfigElement):
371 def __init__(self, seperator, limits, censor_char = "", default = None):
372 ConfigElement.__init__(self)
373 assert isinstance(limits, list) and len(limits[0]) == 2, "limits must be [(min, max),...]-tuple-list"
374 assert censor_char == "" or len(censor_char) == 1, "censor char must be a single char (or \"\")"
375 #assert isinstance(default, list), "default must be a list"
376 #assert isinstance(default[0], int), "list must contain numbers"
377 #assert len(default) == len(limits), "length must match"
380 self.seperator = seperator
382 self.censor_char = censor_char
384 self.default = default
385 self.value = copy.copy(default)
386 self.last_value = copy.copy(default)
388 self.endNotifier = []
393 for i in self._value:
394 max_pos += len(str(self.limits[num][1]))
396 if self._value[num] < self.limits[num][0]:
397 self._value[num] = self.limits[num][0]
399 if self._value[num] > self.limits[num][1]:
400 self._value[num] = self.limits[num][1]
404 if self.marked_pos >= max_pos:
405 for x in self.endNotifier:
407 self.marked_pos = max_pos - 1
409 if self.marked_pos < 0:
412 def validatePos(self):
413 if self.marked_pos < 0:
416 total_len = sum([len(str(x[1])) for x in self.limits])
418 if self.marked_pos >= total_len:
419 self.marked_pos = total_len - 1
421 def addEndNotifier(self, notifier):
422 self.endNotifier.append(notifier)
424 def handleKey(self, key):
440 for i in self._value:
441 max_pos += len(str(self.limits[num][1]))
443 self.marked_pos = max_pos - 1
446 if key in KEY_NUMBERS or key == KEY_ASCII:
448 code = getPrevAsciiCode()
449 if code < 48 or code > 57:
453 number = getKeyNumber(key)
456 for x in self.limits:
457 block_len.append(len(str(x[1])))
459 total_len = sum(block_len)
463 block_len_total = [0, ]
465 pos += block_len[blocknumber]
466 block_len_total.append(pos)
467 if pos - 1 >= self.marked_pos:
472 # length of numberblock
473 number_len = len(str(self.limits[blocknumber][1]))
475 # position in the block
476 posinblock = self.marked_pos - block_len_total[blocknumber]
478 oldvalue = self._value[blocknumber]
479 olddec = oldvalue % 10 ** (number_len - posinblock) - (oldvalue % 10 ** (number_len - posinblock - 1))
480 newvalue = oldvalue - olddec + (10 ** (number_len - posinblock - 1) * number)
482 self._value[blocknumber] = newvalue
490 mPos = self.marked_pos
492 for i in self._value:
493 if len(value): #fixme no heading separator possible
494 value += self.seperator
495 if mPos >= len(value) - 1:
498 if self.censor_char == "":
499 value += ("%0" + str(len(str(self.limits[num][1]))) + "d") % i
501 value += (self.censor_char * len(str(self.limits[num][1])))
506 (value, mPos) = self.genText()
509 def getMulti(self, selected):
510 (value, mPos) = self.genText()
511 # only mark cursor when we are selected
512 # (this code is heavily ink optimized!)
514 return ("mtext"[1-selected:], value, [mPos])
516 return ("text", value)
518 def tostring(self, val):
519 return self.seperator.join([self.saveSingle(x) for x in val])
521 def saveSingle(self, v):
524 def fromstring(self, value):
525 return [int(x) for x in value.split(self.seperator)]
527 def onDeselect(self, session):
528 if not self.last_value == self._value:
530 self.last_value = copy.copy(self._value)
532 class ConfigIP(ConfigSequence):
533 def __init__(self, default, auto_jump = False):
534 ConfigSequence.__init__(self, seperator = ".", limits = [(0,255),(0,255),(0,255),(0,255)], default = default)
536 for x in self.limits:
537 self.block_len.append(len(str(x[1])))
538 self.marked_block = 0
539 self.overwrite = True
540 self.auto_jump = auto_jump
542 def handleKey(self, key):
545 if self.marked_block > 0:
546 self.marked_block -= 1
547 self.overwrite = True
550 if self.marked_block < len(self.limits)-1:
551 self.marked_block += 1
552 self.overwrite = True
555 self.marked_block = 0
556 self.overwrite = True
559 self.marked_block = len(self.limits)-1
560 self.overwrite = True
562 if key in KEY_NUMBERS or key == KEY_ASCII:
564 code = getPrevAsciiCode()
565 if code < 48 or code > 57:
569 number = getKeyNumber(key)
570 oldvalue = self._value[self.marked_block]
573 self._value[self.marked_block] = number
574 self.overwrite = False
577 newvalue = oldvalue + number
578 if self.auto_jump and newvalue > self.limits[self.marked_block][1] and self.marked_block < len(self.limits)-1:
579 self.handleKey(KEY_RIGHT)
583 self._value[self.marked_block] = newvalue
585 if len(str(self._value[self.marked_block])) >= self.block_len[self.marked_block]:
586 self.handleKey(KEY_RIGHT)
594 for i in self._value:
595 block_strlen.append(len(str(i)))
597 value += self.seperator
599 leftPos = sum(block_strlen[:(self.marked_block)])+self.marked_block
600 rightPos = sum(block_strlen[:(self.marked_block+1)])+self.marked_block
601 mBlock = range(leftPos, rightPos)
602 return (value, mBlock)
604 def getMulti(self, selected):
605 (value, mBlock) = self.genText()
607 return ("mtext"[1-selected:], value, mBlock)
609 return ("text", value)
611 def getHTML(self, id):
612 # we definitely don't want leading zeros
613 return '.'.join(["%d" % d for d in self.value])
615 class ConfigMAC(ConfigSequence):
616 def __init__(self, default):
617 ConfigSequence.__init__(self, seperator = ":", limits = [(1,255),(1,255),(1,255),(1,255),(1,255),(1,255)], default = default)
619 class ConfigPosition(ConfigSequence):
620 def __init__(self, default, args):
621 ConfigSequence.__init__(self, seperator = ",", limits = [(0,args[0]),(0,args[1]),(0,args[2]),(0,args[3])], default = default)
623 class ConfigClock(ConfigSequence):
624 def __init__(self, default):
626 t = time.localtime(default)
627 ConfigSequence.__init__(self, seperator = ":", limits = [(0,23),(0,59)], default = [t.tm_hour, t.tm_min])
630 # Check if Minutes maxed out
631 if self._value[1] == 59:
632 # Check if Hours not maxed out
633 if self._value[0] < 23:
634 # Increment Hour, reset Minutes to 0
644 # Check if Minutes is minimum
645 if self._value[1] == 0:
646 # Check if Hour is greater than 0
647 if self._value[0] > 0:
648 # Decrement Hour, set Minutes to 59
657 class ConfigInteger(ConfigSequence):
658 def __init__(self, default, limits = (0, 9999999999)):
659 ConfigSequence.__init__(self, seperator = ":", limits = [limits], default = default)
661 # you need to override this to do input validation
662 def setValue(self, value):
663 self._value = [value]
667 return self._value[0]
669 value = property(getValue, setValue)
671 def fromstring(self, value):
674 def tostring(self, value):
677 class ConfigPIN(ConfigInteger):
678 def __init__(self, default, len = 4, censor = ""):
679 assert isinstance(default, int), "ConfigPIN default must be an integer"
682 ConfigSequence.__init__(self, seperator = ":", limits = [(0, (10**len)-1)], censor_char = censor, default = default)
688 class ConfigFloat(ConfigSequence):
689 def __init__(self, default, limits):
690 ConfigSequence.__init__(self, seperator = ".", limits = limits, default = default)
693 return float(self.value[1] / float(self.limits[1][1] + 1) + self.value[0])
695 float = property(getFloat)
697 # an editable text...
698 class ConfigText(ConfigElement, NumericalTextInput):
699 def __init__(self, default = "", fixed_size = True, visible_width = False):
700 ConfigElement.__init__(self)
701 NumericalTextInput.__init__(self, nextFunc = self.nextFunc, handleTimeout = False)
704 self.allmarked = (default != "")
705 self.fixed_size = fixed_size
706 self.visible_width = visible_width
708 self.overwrite = fixed_size
709 self.help_window = None
710 self.value = self.last_value = self.default = default
712 def validateMarker(self):
714 if self.marked_pos > len(self.text)-1:
715 self.marked_pos = len(self.text)-1
717 if self.marked_pos > len(self.text):
718 self.marked_pos = len(self.text)
719 if self.marked_pos < 0:
721 if self.visible_width:
722 if self.marked_pos < self.offset:
723 self.offset = self.marked_pos
724 if self.marked_pos >= self.offset + self.visible_width:
725 if self.marked_pos == len(self.text):
726 self.offset = self.marked_pos - self.visible_width
728 self.offset = self.marked_pos - self.visible_width + 1
729 if self.offset > 0 and self.offset + self.visible_width > len(self.text):
730 self.offset = max(0, len(self.text) - self.visible_width)
732 def insertChar(self, ch, pos, owr):
733 if owr or self.overwrite:
734 self.text = self.text[0:pos] + ch + self.text[pos + 1:]
735 elif self.fixed_size:
736 self.text = self.text[0:pos] + ch + self.text[pos:-1]
738 self.text = self.text[0:pos] + ch + self.text[pos:]
740 def deleteChar(self, pos):
741 if not self.fixed_size:
742 self.text = self.text[0:pos] + self.text[pos + 1:]
744 self.text = self.text[0:pos] + " " + self.text[pos + 1:]
746 self.text = self.text[0:pos] + self.text[pos + 1:] + " "
748 def deleteAllChars(self):
750 self.text = " " * len(self.text)
755 def handleKey(self, key):
756 # this will no change anything on the value itself
757 # so we can handle it here in gui element
758 if key == KEY_DELETE:
761 self.deleteAllChars()
762 self.allmarked = False
764 self.deleteChar(self.marked_pos)
765 if self.fixed_size and self.overwrite:
767 elif key == KEY_BACKSPACE:
770 self.deleteAllChars()
771 self.allmarked = False
772 elif self.marked_pos > 0:
773 self.deleteChar(self.marked_pos-1)
774 if not self.fixed_size and self.offset > 0:
777 elif key == KEY_LEFT:
780 self.marked_pos = len(self.text)
781 self.allmarked = False
784 elif key == KEY_RIGHT:
788 self.allmarked = False
791 elif key == KEY_HOME:
793 self.allmarked = False
797 self.allmarked = False
798 self.marked_pos = len(self.text)
799 elif key == KEY_TOGGLEOW:
801 self.overwrite = not self.overwrite
802 elif key == KEY_ASCII:
804 newChar = unichr(getPrevAsciiCode())
806 self.deleteAllChars()
807 self.allmarked = False
808 self.insertChar(newChar, self.marked_pos, False)
810 elif key in KEY_NUMBERS:
811 owr = self.lastKey == getKeyNumber(key)
812 newChar = self.getKey(getKeyNumber(key))
814 self.deleteAllChars()
815 self.allmarked = False
816 self.insertChar(newChar, self.marked_pos, owr)
817 elif key == KEY_TIMEOUT:
820 self.help_window.update(self)
824 self.help_window.update(self)
825 self.validateMarker()
830 self.validateMarker()
834 return self.text.encode("utf-8")
836 def setValue(self, val):
838 self.text = val.decode("utf-8")
839 except UnicodeDecodeError:
840 self.text = val.decode("utf-8", "ignore")
843 value = property(getValue, setValue)
844 _value = property(getValue, setValue)
847 return self.text.encode("utf-8")
849 def getMulti(self, selected):
850 if self.visible_width:
852 mark = range(0, min(self.visible_width, len(self.text)))
854 mark = [self.marked_pos-self.offset]
855 return ("mtext"[1-selected:], self.text[self.offset:self.offset+self.visible_width].encode("utf-8")+" ", mark)
858 mark = range(0, len(self.text))
860 mark = [self.marked_pos]
861 return ("mtext"[1-selected:], self.text.encode("utf-8")+" ", mark)
863 def onSelect(self, session):
864 self.allmarked = (self.value != "")
865 if session is not None:
866 from Screens.NumericalTextInputHelpDialog import NumericalTextInputHelpDialog
867 self.help_window = session.instantiateDialog(NumericalTextInputHelpDialog, self)
868 self.help_window.show()
870 def onDeselect(self, session):
874 session.deleteDialog(self.help_window)
875 self.help_window = None
876 if not self.last_value == self.value:
878 self.last_value = self.value
880 def getHTML(self, id):
881 return '<input type="text" name="' + id + '" value="' + self.value + '" /><br>\n'
883 def unsafeAssign(self, value):
884 self.value = str(value)
886 class ConfigPassword(ConfigText):
887 def __init__(self, default = "", fixed_size = False, visible_width = False, censor = "*"):
888 ConfigText.__init__(self, default = default, fixed_size = fixed_size, visible_width = visible_width)
889 self.censor_char = censor
892 def getMulti(self, selected):
893 mtext, text, mark = ConfigText.getMulti(self, selected)
895 text = len(text) * self.censor_char
896 return (mtext, text, mark)
898 def onSelect(self, session):
899 ConfigText.onSelect(self, session)
902 def onDeselect(self, session):
903 ConfigText.onDeselect(self, session)
906 class ConfigNumber(ConfigText):
907 def __init__(self, default = 0):
908 ConfigText.__init__(self, str(default), fixed_size = False)
911 return int(self.text)
913 def setValue(self, val):
916 value = property(getValue, setValue)
917 _value = property(getValue, setValue)
920 pos = len(self.text) - self.marked_pos
921 self.text = self.text.lstrip("0")
924 if pos > len(self.text):
927 self.marked_pos = len(self.text) - pos
929 def handleKey(self, key):
930 if key in KEY_NUMBERS or key == KEY_ASCII:
932 ascii = getPrevAsciiCode()
933 if not (48 <= ascii <= 57):
936 ascii = getKeyNumber(key) + 48
937 newChar = unichr(ascii)
939 self.deleteAllChars()
940 self.allmarked = False
941 self.insertChar(newChar, self.marked_pos, False)
944 ConfigText.handleKey(self, key)
947 def onSelect(self, session):
948 self.allmarked = (self.value != "")
950 def onDeselect(self, session):
953 if not self.last_value == self.value:
955 self.last_value = self.value
957 class ConfigSearchText(ConfigText):
958 def __init__(self, default = "", fixed_size = False, visible_width = False):
959 ConfigText.__init__(self, default = default, fixed_size = fixed_size, visible_width = visible_width)
960 NumericalTextInput.__init__(self, nextFunc = self.nextFunc, handleTimeout = False, search = True)
962 class ConfigDirectory(ConfigText):
963 def __init__(self, default="", visible_width=60):
964 ConfigText.__init__(self, default, fixed_size = True, visible_width = visible_width)
965 def handleKey(self, key):
971 return ConfigText.getValue(self)
972 def setValue(self, val):
975 ConfigText.setValue(self, val)
976 def getMulti(self, selected):
978 return ("mtext"[1-selected:], _("List of Storage Devices"), range(0))
980 return ConfigText.getMulti(self, selected)
983 class ConfigSlider(ConfigElement):
984 def __init__(self, default = 0, increment = 1, limits = (0, 100)):
985 ConfigElement.__init__(self)
986 self.value = self.last_value = self.default = default
989 self.increment = increment
991 def checkValues(self):
992 if self.value < self.min:
993 self.value = self.min
995 if self.value > self.max:
996 self.value = self.max
998 def handleKey(self, key):
1000 self.value -= self.increment
1001 elif key == KEY_RIGHT:
1002 self.value += self.increment
1003 elif key == KEY_HOME:
1004 self.value = self.min
1005 elif key == KEY_END:
1006 self.value = self.max
1012 return "%d / %d" % (self.value, self.max)
1014 def getMulti(self, selected):
1016 return ("slider", self.value, self.max)
1018 def fromstring(self, value):
1021 # a satlist. in fact, it's a ConfigSelection.
1022 class ConfigSatlist(ConfigSelection):
1023 def __init__(self, list, default = None):
1024 if default is not None:
1025 default = str(default)
1026 ConfigSelection.__init__(self, choices = [(str(orbpos), desc) for (orbpos, desc, flags) in list], default = default)
1028 def getOrbitalPosition(self):
1029 if self.value == "":
1031 return int(self.value)
1033 orbital_position = property(getOrbitalPosition)
1035 class ConfigSet(ConfigElement):
1036 def __init__(self, choices, default = []):
1037 ConfigElement.__init__(self)
1039 self.description = {}
1040 if isinstance(choices, list):
1043 if isinstance(x, tuple):
1044 self.choices.append(x[0])
1045 self.description[x[0]] = str(x[1])
1047 self.choices.append(x)
1048 self.description[x] = str(x)
1050 assert False, "ConfigSet choices must be a list!"
1051 if len(self.choices) == 0:
1053 self.description[""] = ""
1058 self.last_value = self.default = default
1059 self.value = default+[]
1061 def toggleChoice(self, choice):
1062 if choice in self.value:
1063 self.value.remove(choice)
1065 self.value.append(choice)
1069 def handleKey(self, key):
1070 if key in KEY_NUMBERS + [KEY_DELETE, KEY_BACKSPACE]:
1072 self.toggleChoice(self.choices[self.pos])
1073 elif key == KEY_LEFT:
1076 self.pos = len(self.choices)-1
1077 elif key == KEY_RIGHT:
1079 if self.pos >= len(self.choices):
1081 elif key in [KEY_HOME, KEY_END]:
1084 def genString(self, lst):
1087 res += self.description[x]+" "
1091 return self.genString(self.value)
1093 def getMulti(self, selected):
1094 if not selected or self.pos == -1:
1095 return ("text", self.genString(self.value))
1098 ch = self.choices[self.pos]
1099 mem = ch in self.value
1104 val1 = self.genString(tmp[:ind])
1105 val2 = " "+self.genString(tmp[ind+1:])
1107 chstr = " "+self.description[ch]+" "
1109 chstr = "("+self.description[ch]+")"
1110 return ("mtext", val1+chstr+val2, range(len(val1),len(val1)+len(chstr)))
1112 def onDeselect(self, session):
1114 if not self.last_value == self.value:
1116 self.last_value = self.value+[]
1118 def tostring(self, value):
1121 def fromstring(self, val):
1124 class ConfigLocations(ConfigElement):
1125 def __init__(self, default = [], visible_width = False):
1126 ConfigElement.__init__(self)
1127 self.visible_width = visible_width
1129 self.default = default
1131 self.mountpoints = []
1132 harddiskmanager.on_partition_list_change.append(self.mountpointsChanged)
1133 self.value = default+[]
1135 def setValue(self, value):
1136 loc = [x[0] for x in self.locations if x[3]]
1137 add = [x for x in value if not x in loc]
1138 diff = add + [x for x in loc if not x in value]
1139 self.locations = [x for x in self.locations if not x[0] in diff] + [[x, self.getMountpoint(x), True, True] for x in add]
1140 self.locations.sort(key = lambda x: x[0])
1144 self.checkChangedMountpoints()
1145 for x in self.locations:
1147 return [x[0] for x in self.locations if x[3]]
1149 value = property(getValue, setValue)
1151 def tostring(self, value):
1154 def fromstring(self, val):
1158 sv = self.saved_value
1162 tmp = self.fromstring(sv)
1163 self.locations = [[x, None, False, False] for x in tmp]
1164 self.refreshMountpoints()
1165 for x in self.locations:
1166 if os.path.exists(x[0]):
1167 x[1] = self.getMountpoint(x[0])
1171 if self.save_disabled or self.locations == []:
1172 self.saved_value = None
1174 self.saved_value = self.tostring([x[0] for x in self.locations])
1176 def isChanged(self):
1177 sv = self.saved_value
1178 if val is None and self.locations == []:
1180 return self.tostring([x[0] for x in self.locations]) != sv
1182 def mountpointsChanged(self, action, dev):
1183 print "Mounts changed: ", action, dev
1184 mp = dev.mountpoint+"/"
1187 elif action == "remove":
1188 self.removedMount(mp)
1189 self.refreshMountpoints()
1191 def addedMount(self, mp):
1192 for x in self.locations:
1195 elif x[1] == None and os.path.exists(x[0]):
1196 x[1] = self.getMountpoint(x[0])
1199 def removedMount(self, mp):
1200 for x in self.locations:
1204 def refreshMountpoints(self):
1205 self.mountpoints = [p.mountpoint + "/" for p in harddiskmanager.getMountedPartitions() if p.mountpoint != "/"]
1206 self.mountpoints.sort(key = lambda x: -len(x))
1208 def checkChangedMountpoints(self):
1209 oldmounts = self.mountpoints
1210 self.refreshMountpoints()
1211 if oldmounts == self.mountpoints:
1214 if not x in self.mountpoints:
1215 self.removedMount(x)
1216 for x in self.mountpoints:
1217 if not x in oldmounts:
1220 def getMountpoint(self, file):
1221 file = os.path.realpath(file)+"/"
1222 for m in self.mountpoints:
1223 if file.startswith(m):
1227 def handleKey(self, key):
1231 self.pos = len(self.value)-1
1232 elif key == KEY_RIGHT:
1234 if self.pos >= len(self.value):
1236 elif key in [KEY_HOME, KEY_END]:
1240 return " ".join(self.value)
1242 def getMulti(self, selected):
1244 valstr = " ".join(self.value)
1245 if self.visible_width and len(valstr) > self.visible_width:
1246 return ("text", valstr[0:self.visible_width])
1248 return ("text", valstr)
1254 for val in self.value:
1257 valstr += str(val)+" "
1261 if self.visible_width and len(valstr) > self.visible_width:
1262 if ind1+1 < self.visible_width/2:
1265 off = min(ind1+1-self.visible_width/2, len(valstr)-self.visible_width)
1266 return ("mtext", valstr[off:off+self.visible_width], range(ind1-off,ind2-off))
1268 return ("mtext", valstr, range(ind1,ind2))
1270 def onDeselect(self, session):
1274 class ConfigNothing(ConfigSelection):
1276 ConfigSelection.__init__(self, choices = [""])
1278 # until here, 'saved_value' always had to be a *string*.
1279 # now, in ConfigSubsection, and only there, saved_value
1280 # is a dict, essentially forming a tree.
1282 # config.foo.bar=True
1283 # config.foobar=False
1286 # config.saved_value == {"foo": {"bar": "True"}, "foobar": "False"}
1290 class ConfigSubsectionContent(object):
1293 # we store a backup of the loaded configuration
1294 # data in self.stored_values, to be able to deploy
1295 # them when a new config element will be added,
1296 # so non-default values are instantly available
1298 # A list, for example:
1299 # config.dipswitches = ConfigSubList()
1300 # config.dipswitches.append(ConfigYesNo())
1301 # config.dipswitches.append(ConfigYesNo())
1302 # config.dipswitches.append(ConfigYesNo())
1303 class ConfigSubList(list, object):
1305 object.__init__(self)
1307 self.stored_values = {}
1317 def getSavedValue(self):
1319 for i in range(len(self)):
1320 sv = self[i].saved_value
1325 def setSavedValue(self, values):
1326 self.stored_values = dict(values)
1327 for (key, val) in self.stored_values.items():
1328 if int(key) < len(self):
1329 self[int(key)].saved_value = val
1331 saved_value = property(getSavedValue, setSavedValue)
1333 def append(self, item):
1335 list.append(self, item)
1336 if i in self.stored_values:
1337 item.saved_value = self.stored_values[i]
1342 for index in range(len(self)):
1343 res[str(index)] = self[index]
1346 # same as ConfigSubList, just as a dictionary.
1347 # care must be taken that the 'key' has a proper
1348 # str() method, because it will be used in the config
1350 class ConfigSubDict(dict, object):
1352 object.__init__(self)
1354 self.stored_values = {}
1357 for x in self.values():
1361 for x in self.values():
1364 def getSavedValue(self):
1366 for (key, val) in self.items():
1367 sv = val.saved_value
1372 def setSavedValue(self, values):
1373 self.stored_values = dict(values)
1374 for (key, val) in self.items():
1375 if str(key) in self.stored_values:
1376 val = self.stored_values[str(key)]
1378 saved_value = property(getSavedValue, setSavedValue)
1380 def __setitem__(self, key, item):
1381 dict.__setitem__(self, key, item)
1382 if str(key) in self.stored_values:
1383 item.saved_value = self.stored_values[str(key)]
1389 # Like the classes above, just with a more "native"
1392 # some evil stuff must be done to allow instant
1393 # loading of added elements. this is why this class
1396 # we need the 'content' because we overwrite
1398 # If you don't understand this, try adding
1399 # __setattr__ to a usual exisiting class and you will.
1400 class ConfigSubsection(object):
1402 object.__init__(self)
1403 self.__dict__["content"] = ConfigSubsectionContent()
1404 self.content.items = { }
1405 self.content.stored_values = { }
1407 def __setattr__(self, name, value):
1408 if name == "saved_value":
1409 return self.setSavedValue(value)
1410 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"
1411 self.content.items[name] = value
1412 if name in self.content.stored_values:
1413 #print "ok, now we have a new item,", name, "and have the following value for it:", self.content.stored_values[name]
1414 value.saved_value = self.content.stored_values[name]
1417 def __getattr__(self, name):
1418 return self.content.items[name]
1420 def getSavedValue(self):
1421 res = self.content.stored_values
1422 for (key, val) in self.content.items.items():
1423 sv = val.saved_value
1430 def setSavedValue(self, values):
1431 values = dict(values)
1433 self.content.stored_values = values
1435 for (key, val) in self.content.items.items():
1437 val.saved_value = values[key]
1439 saved_value = property(getSavedValue, setSavedValue)
1442 for x in self.content.items.values():
1446 for x in self.content.items.values():
1450 return self.content.items
1452 # the root config object, which also can "pickle" (=serialize)
1453 # down the whole config tree.
1455 # we try to keep non-existing config entries, to apply them whenever
1456 # a new config entry is added to a subsection
1457 # also, non-existing config entries will be saved, so they won't be
1458 # lost when a config entry disappears.
1459 class Config(ConfigSubsection):
1461 ConfigSubsection.__init__(self)
1463 def pickle_this(self, prefix, topickle, result):
1464 for (key, val) in topickle.items():
1465 name = prefix + "." + key
1467 if isinstance(val, dict):
1468 self.pickle_this(name, val, result)
1469 elif isinstance(val, tuple):
1470 result.append(name + "=" + val[0]) # + " ; " + val[1])
1472 result.append(name + "=" + val)
1476 self.pickle_this("config", self.saved_value, result)
1477 return '\n'.join(result) + "\n"
1479 def unpickle(self, lines):
1482 if not len(l) or l[0] == '#':
1486 val = l[n+1:].strip()
1488 names = l[:n].split('.')
1489 # if val.find(' ') != -1:
1490 # val = val[:val.find(' ')]
1494 for n in names[:-1]:
1495 base = base.setdefault(n, {})
1497 base[names[-1]] = val
1499 # we inherit from ConfigSubsection, so ...
1500 #object.__setattr__(self, "saved_value", tree["config"])
1501 if "config" in tree:
1502 self.setSavedValue(tree["config"])
1504 def saveToFile(self, filename):
1505 f = open(filename, "w")
1506 f.write(self.pickle())
1509 def loadFromFile(self, filename):
1510 f = open(filename, "r")
1511 self.unpickle(f.readlines())
1515 config.misc = ConfigSubsection()
1518 CONFIG_FILE = resolveFilename(SCOPE_CONFIG, "settings")
1522 config.loadFromFile(self.CONFIG_FILE)
1524 print "unable to load config (%s), assuming defaults..." % str(e)
1528 config.saveToFile(self.CONFIG_FILE)
1530 def __resolveValue(self, pickles, cmap):
1531 if cmap.has_key(pickles[0]):
1532 if len(pickles) > 1:
1533 return self.__resolveValue(pickles[1:], cmap[pickles[0]].dict())
1535 return str(cmap[pickles[0]].value)
1538 def getResolvedKey(self, key):
1539 names = key.split('.')
1541 if names[0] == "config":
1542 ret=self.__resolveValue(names[1:], config.content.items)
1543 if ret and len(ret):
1545 print "getResolvedKey", key, "failed !! (Typo??)"
1548 def NoSave(element):
1549 element.disableSave()
1552 configfile = ConfigFile()
1556 def getConfigListEntry(*args):
1557 assert len(args) > 1, "getConfigListEntry needs a minimum of two arguments (descr, configElement)"
1560 def updateConfigElement(element, newelement):
1561 newelement.value = element.value
1567 #config.bla = ConfigSubsection()
1568 #config.bla.test = ConfigYesNo()
1569 #config.nim = ConfigSubList()
1570 #config.nim.append(ConfigSubsection())
1571 #config.nim[0].bla = ConfigYesNo()
1572 #config.nim.append(ConfigSubsection())
1573 #config.nim[1].bla = ConfigYesNo()
1574 #config.nim[1].blub = ConfigYesNo()
1575 #config.arg = ConfigSubDict()
1576 #config.arg["Hello"] = ConfigYesNo()
1578 #config.arg["Hello"].handleKey(KEY_RIGHT)
1579 #config.arg["Hello"].handleKey(KEY_RIGHT)
1581 ##config.saved_value
1585 #print config.pickle()