code.vuplus.com Git - vuplus_dvbapp/blob - lib/python/Components/config.py

   1 from enigma import getPrevAsciiCode
   2 from Tools.NumericalTextInput import NumericalTextInput
   3 from Tools.Directories import resolveFilename, SCOPE_CONFIG, fileExists
   4 from Components.Harddisk import harddiskmanager
   5 from copy import copy as copy_copy
   6 from os import path as os_path
   7 from time import localtime, strftime
   8
   9 # ConfigElement, the base class of all ConfigElements.
  10
  11 # it stores:
  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
  21 #
  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)
  27 #            or invalid.
  28 #
  29 class ConfigElement(object):
  30         def __init__(self):
  31                 self.saved_value = None
  32                 self.save_forced = False
  33                 self.last_value = None
  34                 self.save_disabled = False
  35                 self.__notifiers = None
  36                 self.__notifiers_final = None
  37                 self.enabled = True
  38                 self.callNotifiersOnSaveAndCancel = False
  39
  40         def getNotifiers(self):
  41                 if self.__notifiers is None:
  42                         self.__notifiers = [ ]
  43                 return self.__notifiers
  44
  45         def setNotifiers(self, val):
  46                 self.__notifiers = val
  47
  48         notifiers = property(getNotifiers, setNotifiers)
  49
  50         def getNotifiersFinal(self):
  51                 if self.__notifiers_final is None:
  52                         self.__notifiers_final = [ ]
  53                 return self.__notifiers_final
  54
  55         def setNotifiersFinal(self, val):
  56                 self.__notifiers_final = val
  57
  58         notifiers_final = property(getNotifiersFinal, setNotifiersFinal)
  59
  60         # you need to override this to do input validation
  61         def setValue(self, value):
  62                 self._value = value
  63                 self.changed()
  64
  65         def getValue(self):
  66                 return self._value
  67
  68         value = property(getValue, setValue)
  69
  70         # you need to override this if self.value is not a string
  71         def fromstring(self, value):
  72                 return value
  73
  74         # you can overide this for fancy default handling
  75         def load(self):
  76                 sv = self.saved_value
  77                 if sv is None:
  78                         self.value = self.default
  79                 else:
  80                         self.value = self.fromstring(sv)
  81
  82         def tostring(self, value):
  83                 return str(value)
  84
  85         # you need to override this if str(self.value) doesn't work
  86         def save(self):
  87                 if self.save_disabled or (self.value == self.default and not self.save_forced):
  88                         self.saved_value = None
  89                 else:
  90                         self.saved_value = self.tostring(self.value)
  91                 if self.callNotifiersOnSaveAndCancel:
  92                         self.changed()
  93
  94         def cancel(self):
  95                 self.load()
  96                 if self.callNotifiersOnSaveAndCancel:
  97                         self.changed()
  98
  99         def isChanged(self):
 100                 sv = self.saved_value
 101                 if sv is None and self.value == self.default:
 102                         return False
 103                 return self.tostring(self.value) != sv
 104
 105         def changed(self):
 106                 if self.__notifiers:
 107                         for x in self.notifiers:
 108                                 x(self)
 109
 110         def changedFinal(self):
 111                 if self.__notifiers_final:
 112                         for x in self.notifiers_final:
 113                                 x(self)
 114
 115         def addNotifier(self, notifier, initial_call = True, immediate_feedback = True):
 116                 assert callable(notifier), "notifiers must be callable"
 117                 if immediate_feedback:
 118                         self.notifiers.append(notifier)
 119                 else:
 120                         self.notifiers_final.append(notifier)
 121                 # CHECKME:
 122                 # do we want to call the notifier
 123                 #  - at all when adding it? (yes, though optional)
 124                 #  - when the default is active? (yes)
 125                 #  - when no value *yet* has been set,
 126                 #    because no config has ever been read (currently yes)
 127                 #    (though that's not so easy to detect.
 128                 #     the entry could just be new.)
 129                 if initial_call:
 130                         notifier(self)
 131
 132         def disableSave(self):
 133                 self.save_disabled = True
 134
 135         def __call__(self, selected):
 136                 return self.getMulti(selected)
 137
 138         def onSelect(self, session):
 139                 pass
 140
 141         def onDeselect(self, session):
 142                 if not self.last_value == self.value:
 143                         self.changedFinal()
 144                         self.last_value = self.value
 145
 146 KEY_LEFT = 0
 147 KEY_RIGHT = 1
 148 KEY_OK = 2
 149 KEY_DELETE = 3
 150 KEY_BACKSPACE = 4
 151 KEY_HOME = 5
 152 KEY_END = 6
 153 KEY_TOGGLEOW = 7
 154 KEY_ASCII = 8
 155 KEY_TIMEOUT = 9
 156 KEY_NUMBERS = range(12, 12+10)
 157 KEY_0 = 12
 158 KEY_9 = 12+9
 159
 160 def getKeyNumber(key):
 161         assert key in KEY_NUMBERS
 162         return key - KEY_0
 163
 164 class choicesList(object): # XXX: we might want a better name for this
 165         LIST_TYPE_LIST = 1
 166         LIST_TYPE_DICT = 2
 167
 168         def __init__(self, choices, type = None):
 169                 self.choices = choices
 170                 if type is None:
 171                         if isinstance(choices, list):
 172                                 self.type = choicesList.LIST_TYPE_LIST
 173                         elif isinstance(choices, dict):
 174                                 self.type = choicesList.LIST_TYPE_DICT
 175                         else:
 176                                 assert False, "choices must be dict or list!"
 177                 else:
 178                         self.type = type
 179
 180         def __list__(self):
 181                 if self.type == choicesList.LIST_TYPE_LIST:
 182                         ret = [not isinstance(x, tuple) and x or x[0] for x in self.choices]
 183                 else:
 184                         ret = self.choices.keys()
 185                 return ret or [""]
 186
 187         def __iter__(self):
 188                 if self.type == choicesList.LIST_TYPE_LIST:
 189                         ret = [not isinstance(x, tuple) and x or x[0] for x in self.choices]
 190                 else:
 191                         ret = self.choices
 192                 return iter(ret or [""])
 193
 194         def __len__(self):
 195                 return len(self.choices) or 1
 196
 197         def __getitem__(self, index):
 198                 if self.type == choicesList.LIST_TYPE_LIST:
 199                         ret = self.choices[index]
 200                         if isinstance(ret, tuple):
 201                                 ret = ret[0]
 202                         return ret
 203                 return self.choices.keys()[index]
 204
 205         def index(self, value):
 206                 return self.__list__().index(value)
 207
 208         def __setitem__(self, index, value):
 209                 if self.type == choicesList.LIST_TYPE_LIST:
 210                         orig = self.choices[index]
 211                         if isinstance(orig, tuple):
 212                                 self.choices[index] = (value, orig[1])
 213                         else:
 214                                 self.choices[index] = value
 215                 else:
 216                         key = self.choices.keys()[index]
 217                         orig = self.choices[key]
 218                         del self.choices[key]
 219                         self.choices[value] = orig
 220
 221         def default(self):
 222                 choices = self.choices
 223                 if not choices:
 224                         return ""
 225                 if self.type is choicesList.LIST_TYPE_LIST:
 226                         default = choices[0]
 227                         if isinstance(default, tuple):
 228                                 default = default[0]
 229                 else:
 230                         default = choices.keys()[0]
 231                 return default
 232
 233 class descriptionList(choicesList): # XXX: we might want a better name for this
 234         def __list__(self):
 235                 if self.type == choicesList.LIST_TYPE_LIST:
 236                         ret = [not isinstance(x, tuple) and x or x[1] for x in self.choices]
 237                 else:
 238                         ret = self.choices.values()
 239                 return ret or [""]
 240
 241         def __iter__(self):
 242                 return iter(self.__list__())
 243
 244         def __getitem__(self, index):
 245                 if self.type == choicesList.LIST_TYPE_LIST:
 246                         for x in self.choices:
 247                                 if isinstance(x, tuple):
 248                                         if x[0] == index:
 249                                                 return str(x[1])
 250                                 elif x == index:
 251                                         return str(x)
 252                         return str(index) # Fallback!
 253                 else:
 254                         return str(self.choices.get(index, ""))
 255
 256         def __setitem__(self, index, value):
 257                 if self.type == choicesList.LIST_TYPE_LIST:
 258                         i = self.index(index)
 259                         orig = self.choices[i]
 260                         if isinstance(orig, tuple):
 261                                 self.choices[i] = (orig[0], value)
 262                         else:
 263                                 self.choices[i] = value
 264                 else:
 265                         self.choices[index] = value
 266
 267 #
 268 # ConfigSelection is a "one of.."-type.
 269 # it has the "choices", usually a list, which contains
 270 # (id, desc)-tuples (or just only the ids, in case the id
 271 # will be used as description)
 272 #
 273 # all ids MUST be plain strings.
 274 #
 275 class ConfigSelection(ConfigElement):
 276         def __init__(self, choices, default = None):
 277                 ConfigElement.__init__(self)
 278                 self.choices = choicesList(choices)
 279
 280                 if default is None:
 281                         default = self.choices.default()
 282
 283                 self._descr = None
 284                 self.default = self._value = self.last_value = default
 285
 286         def setChoices(self, choices, default = None):
 287                 self.choices = choicesList(choices)
 288
 289                 if default is None:
 290                         default = self.choices.default()
 291                 self.default = default
 292
 293                 if self.value not in self.choices:
 294                         self.value = default
 295
 296         def setValue(self, value):
 297                 if value in self.choices:
 298                         self._value = value
 299                 else:
 300                         self._value = self.default
 301                 self._descr = None
 302                 self.changed()
 303
 304         def tostring(self, val):
 305                 return val
 306
 307         def getValue(self):
 308                 return self._value
 309
 310         def setCurrentText(self, text):
 311                 i = self.choices.index(self.value)
 312                 self.choices[i] = text
 313                 self._descr = self.description[text] = text
 314                 self._value = text
 315
 316         value = property(getValue, setValue)
 317
 318         def getIndex(self):
 319                 return self.choices.index(self.value)
 320
 321         index = property(getIndex)
 322
 323         # GUI
 324         def handleKey(self, key):
 325                 nchoices = len(self.choices)
 326                 i = self.choices.index(self.value)
 327                 if key == KEY_LEFT:
 328                         self.value = self.choices[(i + nchoices - 1) % nchoices]
 329                 elif key == KEY_RIGHT:
 330                         self.value = self.choices[(i + 1) % nchoices]
 331                 elif key == KEY_HOME:
 332                         self.value = self.choices[0]
 333                 elif key == KEY_END:
 334                         self.value = self.choices[nchoices - 1]
 335
 336         def selectNext(self):
 337                 nchoices = len(self.choices)
 338                 i = self.choices.index(self.value)
 339                 self.value = self.choices[(i + 1) % nchoices]
 340
 341         def getText(self):
 342                 if self._descr is not None:
 343                         return self._descr
 344                 descr = self._descr = self.description[self.value]
 345                 if descr:
 346                         return _(descr)
 347                 return descr
 348
 349         def getMulti(self, selected):
 350                 if self._descr is not None:
 351                         descr = self._descr
 352                 else:
 353                         descr = self._descr = self.description[self.value]
 354                 if descr:
 355                         return ("text", _(descr))
 356                 return ("text", descr)
 357
 358         # HTML
 359         def getHTML(self, id):
 360                 res = ""
 361                 for v in self.choices:
 362                         descr = self.description[v]
 363                         if self.value == v:
 364                                 checked = 'checked="checked" '
 365                         else:
 366                                 checked = ''
 367                         res += '<input type="radio" name="' + id + '" ' + checked + 'value="' + v + '">' + descr + "</input></br>\n"
 368                 return res;
 369
 370         def unsafeAssign(self, value):
 371                 # setValue does check if value is in choices. This is safe enough.
 372                 self.value = value
 373
 374         description = property(lambda self: descriptionList(self.choices.choices, self.choices.type))
 375
 376 # a binary decision.
 377 #
 378 # several customized versions exist for different
 379 # descriptions.
 380 #
 381 boolean_descriptions = {False: "false", True: "true"}
 382 class ConfigBoolean(ConfigElement):
 383         def __init__(self, default = False, descriptions = boolean_descriptions):
 384                 ConfigElement.__init__(self)
 385                 self.descriptions = descriptions
 386                 self.value = self.last_value = self.default = default
 387
 388         def handleKey(self, key):
 389                 if key in (KEY_LEFT, KEY_RIGHT):
 390                         self.value = not self.value
 391                 elif key == KEY_HOME:
 392                         self.value = False
 393                 elif key == KEY_END:
 394                         self.value = True
 395
 396         def getText(self):
 397                 descr = self.descriptions[self.value]
 398                 if descr:
 399                         return _(descr)
 400                 return descr
 401
 402         def getMulti(self, selected):
 403                 descr = self.descriptions[self.value]
 404                 if descr:
 405                         return ("text", _(descr))
 406                 return ("text", descr)
 407
 408         def tostring(self, value):
 409                 if not value:
 410                         return "false"
 411                 else:
 412                         return "true"
 413
 414         def fromstring(self, val):
 415                 if val == "true":
 416                         return True
 417                 else:
 418                         return False
 419
 420         def getHTML(self, id):
 421                 if self.value:
 422                         checked = ' checked="checked"'
 423                 else:
 424                         checked = ''
 425                 return '<input type="checkbox" name="' + id + '" value="1" ' + checked + " />"
 426
 427         # this is FLAWED. and must be fixed.
 428         def unsafeAssign(self, value):
 429                 if value == "1":
 430                         self.value = True
 431                 else:
 432                         self.value = False
 433
 434         def onDeselect(self, session):
 435                 if not self.last_value == self.value:
 436                         self.changedFinal()
 437                         self.last_value = self.value
 438
 439 yes_no_descriptions = {False: _("no"), True: _("yes")}
 440 class ConfigYesNo(ConfigBoolean):
 441         def __init__(self, default = False):
 442                 ConfigBoolean.__init__(self, default = default, descriptions = yes_no_descriptions)
 443
 444 on_off_descriptions = {False: _("off"), True: _("on")}
 445 class ConfigOnOff(ConfigBoolean):
 446         def __init__(self, default = False):
 447                 ConfigBoolean.__init__(self, default = default, descriptions = on_off_descriptions)
 448
 449 enable_disable_descriptions = {False: _("disable"), True: _("enable")}
 450 class ConfigEnableDisable(ConfigBoolean):
 451         def __init__(self, default = False):
 452                 ConfigBoolean.__init__(self, default = default, descriptions = enable_disable_descriptions)
 453
 454 class ConfigDateTime(ConfigElement):
 455         def __init__(self, default, formatstring, increment = 86400):
 456                 ConfigElement.__init__(self)
 457                 self.increment = increment
 458                 self.formatstring = formatstring
 459                 self.value = self.last_value = self.default = int(default)
 460
 461         def handleKey(self, key):
 462                 if key == KEY_LEFT:
 463                         self.value = self.value - self.increment
 464                 elif key == KEY_RIGHT:
 465                         self.value = self.value + self.increment
 466                 elif key == KEY_HOME or key == KEY_END:
 467                         self.value = self.default
 468
 469         def getText(self):
 470                 return strftime(self.formatstring, localtime(self.value))
 471
 472         def getMulti(self, selected):
 473                 return ("text", strftime(self.formatstring, localtime(self.value)))
 474
 475         def fromstring(self, val):
 476                 return int(val)
 477
 478 # *THE* mighty config element class
 479 #
 480 # allows you to store/edit a sequence of values.
 481 # can be used for IP-addresses, dates, plain integers, ...
 482 # several helper exist to ease this up a bit.
 483 #
 484 class ConfigSequence(ConfigElement):
 485         def __init__(self, seperator, limits, default, censor_char = ""):
 486                 ConfigElement.__init__(self)
 487                 assert isinstance(limits, list) and len(limits[0]) == 2, "limits must be [(min, max),...]-tuple-list"
 488                 assert censor_char == "" or len(censor_char) == 1, "censor char must be a single char (or \"\")"
 489                 #assert isinstance(default, list), "default must be a list"
 490                 #assert isinstance(default[0], int), "list must contain numbers"
 491                 #assert len(default) == len(limits), "length must match"
 492
 493                 self.marked_pos = 0
 494                 self.seperator = seperator
 495                 self.limits = limits
 496                 self.censor_char = censor_char
 497
 498                 self.last_value = self.default = default
 499                 self.value = copy_copy(default)
 500                 self.endNotifier = None
 501
 502         def validate(self):
 503                 max_pos = 0
 504                 num = 0
 505                 for i in self._value:
 506                         max_pos += len(str(self.limits[num][1]))
 507
 508                         if self._value[num] < self.limits[num][0]:
 509                                 self._value[num] = self.limits[num][0]
 510
 511                         if self._value[num] > self.limits[num][1]:
 512                                 self._value[num] = self.limits[num][1]
 513
 514                         num += 1
 515
 516                 if self.marked_pos >= max_pos:
 517                         if self.endNotifier:
 518                                 for x in self.endNotifier:
 519                                         x(self)
 520                         self.marked_pos = max_pos - 1
 521
 522                 if self.marked_pos < 0:
 523                         self.marked_pos = 0
 524
 525         def validatePos(self):
 526                 if self.marked_pos < 0:
 527                         self.marked_pos = 0
 528
 529                 total_len = sum([len(str(x[1])) for x in self.limits])
 530
 531                 if self.marked_pos >= total_len:
 532                         self.marked_pos = total_len - 1
 533
 534         def addEndNotifier(self, notifier):
 535                 if self.endNotifier is None:
 536                         self.endNotifier = []
 537                 self.endNotifier.append(notifier)
 538
 539         def handleKey(self, key):
 540                 if key == KEY_LEFT:
 541                         self.marked_pos -= 1
 542                         self.validatePos()
 543
 544                 elif key == KEY_RIGHT:
 545                         self.marked_pos += 1
 546                         self.validatePos()
 547
 548                 elif key == KEY_HOME:
 549                         self.marked_pos = 0
 550                         self.validatePos()
 551
 552                 elif key == KEY_END:
 553                         max_pos = 0
 554                         num = 0
 555                         for i in self._value:
 556                                 max_pos += len(str(self.limits[num][1]))
 557                                 num += 1
 558                         self.marked_pos = max_pos - 1
 559                         self.validatePos()
 560
 561                 elif key in KEY_NUMBERS or key == KEY_ASCII:
 562                         if key == KEY_ASCII:
 563                                 code = getPrevAsciiCode()
 564                                 if code < 48 or code > 57:
 565                                         return
 566                                 number = code - 48
 567                         else:
 568                                 number = getKeyNumber(key)
 569
 570                         block_len = [len(str(x[1])) for x in self.limits]
 571                         total_len = sum(block_len)
 572
 573                         pos = 0
 574                         blocknumber = 0
 575                         block_len_total = [0, ]
 576                         for x in block_len:
 577                                 pos += block_len[blocknumber]
 578                                 block_len_total.append(pos)
 579                                 if pos - 1 >= self.marked_pos:
 580                                         pass
 581                                 else:
 582                                         blocknumber += 1
 583
 584                         # length of numberblock
 585                         number_len = len(str(self.limits[blocknumber][1]))
 586
 587                         # position in the block
 588                         posinblock = self.marked_pos - block_len_total[blocknumber]
 589
 590                         oldvalue = self._value[blocknumber]
 591                         olddec = oldvalue % 10 ** (number_len - posinblock) - (oldvalue % 10 ** (number_len - posinblock - 1))
 592                         newvalue = oldvalue - olddec + (10 ** (number_len - posinblock - 1) * number)
 593
 594                         self._value[blocknumber] = newvalue
 595                         self.marked_pos += 1
 596
 597                         self.validate()
 598                         self.changed()
 599
 600         def genText(self):
 601                 value = ""
 602                 mPos = self.marked_pos
 603                 num = 0;
 604                 for i in self._value:
 605                         if value:       #fixme no heading separator possible
 606                                 value += self.seperator
 607                                 if mPos >= len(value) - 1:
 608                                         mPos += 1
 609                         if self.censor_char == "":
 610                                 value += ("%0" + str(len(str(self.limits[num][1]))) + "d") % i
 611                         else:
 612                                 value += (self.censor_char * len(str(self.limits[num][1])))
 613                         num += 1
 614                 return (value, mPos)
 615
 616         def getText(self):
 617                 (value, mPos) = self.genText()
 618                 return value
 619
 620         def getMulti(self, selected):
 621                 (value, mPos) = self.genText()
 622                         # only mark cursor when we are selected
 623                         # (this code is heavily ink optimized!)
 624                 if self.enabled:
 625                         return ("mtext"[1-selected:], value, [mPos])
 626                 else:
 627                         return ("text", value)
 628
 629         def tostring(self, val):
 630                 return self.seperator.join([self.saveSingle(x) for x in val])
 631
 632         def saveSingle(self, v):
 633                 return str(v)
 634
 635         def fromstring(self, value):
 636                 return [int(x) for x in value.split(self.seperator)]
 637
 638         def onDeselect(self, session):
 639                 if self.last_value != self._value:
 640                         self.changedFinal()
 641                         self.last_value = copy_copy(self._value)
 642
 643 ip_limits = [(0,255),(0,255),(0,255),(0,255)]
 644 class ConfigIP(ConfigSequence):
 645         def __init__(self, default, auto_jump = False):
 646                 ConfigSequence.__init__(self, seperator = ".", limits = ip_limits, default = default)
 647                 self.block_len = [len(str(x[1])) for x in self.limits]
 648                 self.marked_block = 0
 649                 self.overwrite = True
 650                 self.auto_jump = auto_jump
 651
 652         def handleKey(self, key):
 653                 if key == KEY_LEFT:
 654                         if self.marked_block > 0:
 655                                 self.marked_block -= 1
 656                         self.overwrite = True
 657
 658                 elif key == KEY_RIGHT:
 659                         if self.marked_block < len(self.limits)-1:
 660                                 self.marked_block += 1
 661                         self.overwrite = True
 662
 663                 elif key == KEY_HOME:
 664                         self.marked_block = 0
 665                         self.overwrite = True
 666
 667                 elif key == KEY_END:
 668                         self.marked_block = len(self.limits)-1
 669                         self.overwrite = True
 670
 671                 elif key in KEY_NUMBERS or key == KEY_ASCII:
 672                         if key == KEY_ASCII:
 673                                 code = getPrevAsciiCode()
 674                                 if code < 48 or code > 57:
 675                                         return
 676                                 number = code - 48
 677                         else:
 678                                 number = getKeyNumber(key)
 679                         oldvalue = self._value[self.marked_block]
 680
 681                         if self.overwrite:
 682                                 self._value[self.marked_block] = number
 683                                 self.overwrite = False
 684                         else:
 685                                 oldvalue *= 10
 686                                 newvalue = oldvalue + number
 687                                 if self.auto_jump and newvalue > self.limits[self.marked_block][1] and self.marked_block < len(self.limits)-1:
 688                                         self.handleKey(KEY_RIGHT)
 689                                         self.handleKey(key)
 690                                         return
 691                                 else:
 692                                         self._value[self.marked_block] = newvalue
 693
 694                         if len(str(self._value[self.marked_block])) >= self.block_len[self.marked_block]:
 695                                 self.handleKey(KEY_RIGHT)
 696
 697                         self.validate()
 698                         self.changed()
 699
 700         def genText(self):
 701                 value = ""
 702                 block_strlen = []
 703                 for i in self._value:
 704                         block_strlen.append(len(str(i)))
 705                         if value:
 706                                 value += self.seperator
 707                         value += str(i)
 708                 leftPos = sum(block_strlen[:(self.marked_block)])+self.marked_block
 709                 rightPos = sum(block_strlen[:(self.marked_block+1)])+self.marked_block
 710                 mBlock = range(leftPos, rightPos)
 711                 return (value, mBlock)
 712
 713         def getMulti(self, selected):
 714                 (value, mBlock) = self.genText()
 715                 if self.enabled:
 716                         return ("mtext"[1-selected:], value, mBlock)
 717                 else:
 718                         return ("text", value)
 719
 720         def getHTML(self, id):
 721                 # we definitely don't want leading zeros
 722                 return '.'.join(["%d" % d for d in self.value])
 723
 724 mac_limits = [(1,255),(1,255),(1,255),(1,255),(1,255),(1,255)]
 725 class ConfigMAC(ConfigSequence):
 726         def __init__(self, default):
 727                 ConfigSequence.__init__(self, seperator = ":", limits = mac_limits, default = default)
 728
 729 class ConfigPosition(ConfigSequence):
 730         def __init__(self, default, args):
 731                 ConfigSequence.__init__(self, seperator = ",", limits = [(0,args[0]),(0,args[1]),(0,args[2]),(0,args[3])], default = default)
 732
 733 clock_limits = [(0,23),(0,59)]
 734 class ConfigClock(ConfigSequence):
 735         def __init__(self, default):
 736                 t = localtime(default)
 737                 ConfigSequence.__init__(self, seperator = ":", limits = clock_limits, default = [t.tm_hour, t.tm_min])
 738
 739         def increment(self):
 740                 # Check if Minutes maxed out
 741                 if self._value[1] == 59:
 742                         # Increment Hour, reset Minutes
 743                         if self._value[0] < 23:
 744                                 self._value[0] += 1
 745                         else:
 746                                 self._value[0] = 0
 747                         self._value[1] = 0
 748                 else:
 749                         # Increment Minutes
 750                         self._value[1] += 1
 751                 # Trigger change
 752                 self.changed()
 753
 754         def decrement(self):
 755                 # Check if Minutes is minimum
 756                 if self._value[1] == 0:
 757                         # Decrement Hour, set Minutes to 59
 758                         if self._value[0] > 0:
 759                                 self._value[0] -= 1
 760                         else:
 761                                 self._value[0] = 23
 762                         self._value[1] = 59
 763                 else:
 764                         # Decrement Minutes
 765                         self._value[1] -= 1
 766                 # Trigger change
 767                 self.changed()
 768
 769 integer_limits = (0, 9999999999)
 770 class ConfigInteger(ConfigSequence):
 771         def __init__(self, default, limits = integer_limits):
 772                 ConfigSequence.__init__(self, seperator = ":", limits = [limits], default = default)
 773
 774         # you need to override this to do input validation
 775         def setValue(self, value):
 776                 self._value = [value]
 777                 self.changed()
 778
 779         def getValue(self):
 780                 return self._value[0]
 781
 782         value = property(getValue, setValue)
 783
 784         def fromstring(self, value):
 785                 return int(value)
 786
 787         def tostring(self, value):
 788                 return str(value)
 789
 790 class ConfigPIN(ConfigInteger):
 791         def __init__(self, default, len = 4, censor = ""):
 792                 assert isinstance(default, int), "ConfigPIN default must be an integer"
 793                 if default == -1:
 794                         default = "aaaa"
 795                 ConfigSequence.__init__(self, seperator = ":", limits = [(0, (10**len)-1)], censor_char = censor, default = default)
 796                 self.len = len
 797
 798         def getLength(self):
 799                 return self.len
 800
 801 class ConfigFloat(ConfigSequence):
 802         def __init__(self, default, limits):
 803                 ConfigSequence.__init__(self, seperator = ".", limits = limits, default = default)
 804
 805         def getFloat(self):
 806                 return float(self.value[1] / float(self.limits[1][1] + 1) + self.value[0])
 807
 808         float = property(getFloat)
 809
 810 #### vuplus
 811 #Normal, LShift(42), RAlt(100), LShift+RAlt(100+42)/LArt(56)
 812 vukeymap_us_de = {
 813           2:[u"1", u"!", None, None]
 814         , 3:[u"2", u"@", None, None]
 815         , 4:[u"3", u"#", None, '\xc2\xa3']
 816         , 5:[u"4", u"$", '\xc3\xa7', None]
 817         , 6:[u"5", u"%", '\xc3\xbc', '\xe2\x82\xac']
 818         , 7:[u"6", u"^", '\xc3\xb6', None]
 819         , 8:[u"7", u"&", '\xc3\xa4', None]
 820         , 9:[u"8", u"*", '\xc3\xa0', None]
 821         ,10:[u"9", u"(", '\xc3\xa8', None]
 822         ,11:[u"0", u")", '\xc3\xa9', None]
 823         ,12:[u"-", u"_", None, None]
 824         ,13:[u"=", u"+", "~", None]
 825         ,16:[u"q", u"Q", None, None]
 826         ,17:[u"w", u"W", None, None]
 827         ,18:[u"e", u"E", '\xe2\x82\xac', None]
 828         ,19:[u"r", u"R", None, None]
 829         ,20:[u"t", u"T", None, None]
 830         ,21:[u"y", u"Y", None, None]
 831         ,22:[u"u", u"U", None, None]
 832         ,23:[u"i", u"I", None, None]
 833         ,24:[u"o", u"O", None, None]
 834         ,25:[u"p", u"P", None, None]
 835         ,26:[u"[", u"{", None, None]
 836         ,27:[u"]", u"}", None, None]
 837         ,30:[u"a", u"A", None, None]
 838         ,31:[u"s", u"S", '\xc3\x9f', None]
 839         ,32:[u"d", u"D", None, None]
 840         ,33:[u"f", u"F", None, None]
 841         ,34:[u"g", u"G", None, None]
 842         ,35:[u"h", u"H", None, None]
 843         ,36:[u"j", u"J", None, None]
 844         ,37:[u"k", u"K", None, None]
 845         ,38:[u"l", u"L", None, None]
 846         ,39:[u";", u":", None, None]
 847         ,40:[u"\'", u"\"", None, None]
 848         ,41:['\xc2\xa7', '\xc2\xb0', '\xc2\xac', None]
 849         ,43:[u"\\", u"|", None, None]
 850         ,44:[u"z", u"Z", None, u"<"]
 851         ,45:[u"x", u"X", None, u">"]
 852         ,46:[u"c", u"C", '\xc2\xa2', None]
 853         ,47:[u"v", u"V", None, None]
 854         ,48:[u"b", u"B", None, None]
 855         ,49:[u"n", u"N", None, None]
 856         ,50:[u"m", u"M", '\xc2\xb5', None]
 857         ,51:[u",", "<", None, None]
 858         ,52:[u".", ">", None, None]
 859         ,53:[u"/", u"?", None, None]
 860         ,57:[u" ", None, None, None]
 861 }
 862 vumapidx = 0
 863 vukeymap = vukeymap_us_de
 864 rckeyboard_enable = False
 865 if file("/proc/stb/info/vumodel").read().strip() == "ultimo":
 866         rckeyboard_enable = True
 867
 868 def getCharValue(code):
 869         global vumapidx
 870         global vukeymap
 871         global rckeyboard_enable
 872         print "got ascii code : %d [%d]"%(code, vumapidx)
 873         if rckeyboard_enable:
 874                 if code == 0:
 875                         vumapidx = 0
 876                         return None
 877                 elif code == 42:
 878                         vumapidx += 1
 879                         return None
 880                 elif code == 56:
 881                         vumapidx += 3
 882                         return None
 883                 elif code == 100:
 884                         vumapidx += 2
 885                         return None
 886                 try:
 887                         return vukeymap[code][vumapidx]
 888                 except:
 889                         return None
 890         else:
 891                 return unichr(getPrevAsciiCode())
 892 #### vuplus
 893
 894 # an editable text...
 895 class ConfigText(ConfigElement, NumericalTextInput):
 896         def __init__(self, default = "", fixed_size = True, visible_width = False):
 897                 ConfigElement.__init__(self)
 898                 NumericalTextInput.__init__(self, nextFunc = self.nextFunc, handleTimeout = False)
 899
 900                 self.marked_pos = 0
 901                 self.allmarked = (default != "")
 902                 self.fixed_size = fixed_size
 903                 self.visible_width = visible_width
 904                 self.offset = 0
 905                 self.overwrite = fixed_size
 906                 self.help_window = None
 907                 self.value = self.last_value = self.default = default
 908
 909         def validateMarker(self):
 910                 textlen = len(self.text)
 911                 if self.fixed_size:
 912                         if self.marked_pos > textlen-1:
 913                                 self.marked_pos = textlen-1
 914                 else:
 915                         if self.marked_pos > textlen:
 916                                 self.marked_pos = textlen
 917                 if self.marked_pos < 0:
 918                         self.marked_pos = 0
 919                 if self.visible_width:
 920                         if self.marked_pos < self.offset:
 921                                 self.offset = self.marked_pos
 922                         if self.marked_pos >= self.offset + self.visible_width:
 923                                 if self.marked_pos == textlen:
 924                                         self.offset = self.marked_pos - self.visible_width
 925                                 else:
 926                                         self.offset = self.marked_pos - self.visible_width + 1
 927                         if self.offset > 0 and self.offset + self.visible_width > textlen:
 928                                 self.offset = max(0, len - self.visible_width)
 929
 930         def insertChar(self, ch, pos, owr):
 931                 if owr or self.overwrite:
 932                         self.text = self.text[0:pos] + ch + self.text[pos + 1:]
 933                 elif self.fixed_size:
 934                         self.text = self.text[0:pos] + ch + self.text[pos:-1]
 935                 else:
 936                         self.text = self.text[0:pos] + ch + self.text[pos:]
 937
 938         def deleteChar(self, pos):
 939                 if not self.fixed_size:
 940                         self.text = self.text[0:pos] + self.text[pos + 1:]
 941                 elif self.overwrite:
 942                         self.text = self.text[0:pos] + " " + self.text[pos + 1:]
 943                 else:
 944                         self.text = self.text[0:pos] + self.text[pos + 1:] + " "
 945
 946         def deleteAllChars(self):
 947                 if self.fixed_size:
 948                         self.text = " " * len(self.text)
 949                 else:
 950                         self.text = ""
 951                 self.marked_pos = 0
 952
 953         def handleKey(self, key):
 954                 # this will no change anything on the value itself
 955                 # so we can handle it here in gui element
 956                 if key == KEY_DELETE:
 957                         self.timeout()
 958                         if self.allmarked:
 959                                 self.deleteAllChars()
 960                                 self.allmarked = False
 961                         else:
 962                                 self.deleteChar(self.marked_pos)
 963                                 if self.fixed_size and self.overwrite:
 964                                         self.marked_pos += 1
 965                 elif key == KEY_BACKSPACE:
 966                         self.timeout()
 967                         if self.allmarked:
 968                                 self.deleteAllChars()
 969                                 self.allmarked = False
 970                         elif self.marked_pos > 0:
 971                                 self.deleteChar(self.marked_pos-1)
 972                                 if not self.fixed_size and self.offset > 0:
 973                                         self.offset -= 1
 974                                 self.marked_pos -= 1
 975                 elif key == KEY_LEFT:
 976                         self.timeout()
 977                         if self.allmarked:
 978                                 self.marked_pos = len(self.text)
 979                                 self.allmarked = False
 980                         else:
 981                                 self.marked_pos -= 1
 982                 elif key == KEY_RIGHT:
 983                         self.timeout()
 984                         if self.allmarked:
 985                                 self.marked_pos = 0
 986                                 self.allmarked = False
 987                         else:
 988                                 self.marked_pos += 1
 989                 elif key == KEY_HOME:
 990                         self.timeout()
 991                         self.allmarked = False
 992                         self.marked_pos = 0
 993                 elif key == KEY_END:
 994                         self.timeout()
 995                         self.allmarked = False
 996                         self.marked_pos = len(self.text)
 997                 elif key == KEY_TOGGLEOW:
 998                         self.timeout()
 999                         self.overwrite = not self.overwrite
1000                 elif key == KEY_ASCII:
1001                         self.timeout()
1002                         #### vuplus
1003                         #newChar = unichr(getPrevAsciiCode())
1004                         #if not self.useableChars or newChar in self.useableChars:
1005                         newChar = getCharValue(getPrevAsciiCode())
1006                         if (newChar is not None) and (not self.useableChars or newChar in self.useableChars):
1007                                 #### vuplus
1008                                 if self.allmarked:
1009                                         self.deleteAllChars()
1010                                         self.allmarked = False
1011                                 self.insertChar(newChar, self.marked_pos, False)
1012                                 self.marked_pos += 1
1013                 elif key in KEY_NUMBERS:
1014                         owr = self.lastKey == getKeyNumber(key)
1015                         newChar = self.getKey(getKeyNumber(key))
1016                         if self.allmarked:
1017                                 self.deleteAllChars()
1018                                 self.allmarked = False
1019                         self.insertChar(newChar, self.marked_pos, owr)
1020                 elif key == KEY_TIMEOUT:
1021                         self.timeout()
1022                         if self.help_window:
1023                                 self.help_window.update(self)
1024                         return
1025
1026                 if self.help_window:
1027                         self.help_window.update(self)
1028                 self.validateMarker()
1029                 self.changed()
1030
1031         def nextFunc(self):
1032                 self.marked_pos += 1
1033                 self.validateMarker()
1034                 self.changed()
1035
1036         def getValue(self):
1037                 return self.text.encode("utf-8")
1038
1039         def setValue(self, val):
1040                 try:
1041                         self.text = val.decode("utf-8")
1042                 except UnicodeDecodeError:
1043                         self.text = val.decode("utf-8", "ignore")
1044                         print "Broken UTF8!"
1045
1046         value = property(getValue, setValue)
1047         _value = property(getValue, setValue)
1048
1049         def getText(self):
1050                 return self.text.encode("utf-8")
1051
1052         def getMulti(self, selected):
1053                 if self.visible_width:
1054                         if self.allmarked:
1055                                 mark = range(0, min(self.visible_width, len(self.text)))
1056                         else:
1057                                 mark = [self.marked_pos-self.offset]
1058                         return ("mtext"[1-selected:], self.text[self.offset:self.offset+self.visible_width].encode("utf-8")+" ", mark)
1059                 else:
1060                         if self.allmarked:
1061                                 mark = range(0, len(self.text))
1062                         else:
1063                                 mark = [self.marked_pos]
1064                         return ("mtext"[1-selected:], self.text.encode("utf-8")+" ", mark)
1065
1066         def onSelect(self, session):
1067                 self.allmarked = (self.value != "")
1068                 if session is not None:
1069                         from Screens.NumericalTextInputHelpDialog import NumericalTextInputHelpDialog
1070                         self.help_window = session.instantiateDialog(NumericalTextInputHelpDialog, self)
1071                         self.help_window.show()
1072
1073         def onDeselect(self, session):
1074                 self.marked_pos = 0
1075                 self.offset = 0
1076                 if self.help_window:
1077                         session.deleteDialog(self.help_window)
1078                         self.help_window = None
1079                 if not self.last_value == self.value:
1080                         self.changedFinal()
1081                         self.last_value = self.value
1082
1083         def getHTML(self, id):
1084                 return '<input type="text" name="' + id + '" value="' + self.value + '" /><br>\n'
1085
1086         def unsafeAssign(self, value):
1087                 self.value = str(value)
1088
1089 class ConfigPassword(ConfigText):
1090         def __init__(self, default = "", fixed_size = False, visible_width = False, censor = "*"):
1091                 ConfigText.__init__(self, default = default, fixed_size = fixed_size, visible_width = visible_width)
1092                 self.censor_char = censor
1093                 self.hidden = True
1094
1095         def getMulti(self, selected):
1096                 mtext, text, mark = ConfigText.getMulti(self, selected)
1097                 if self.hidden:
1098                         text = len(text) * self.censor_char
1099                 return (mtext, text, mark)
1100
1101         def onSelect(self, session):
1102                 ConfigText.onSelect(self, session)
1103                 self.hidden = False
1104
1105         def onDeselect(self, session):
1106                 ConfigText.onDeselect(self, session)
1107                 self.hidden = True
1108
1109 # lets the user select between [min, min+stepwidth, min+(stepwidth*2)..., maxval] with maxval <= max depending
1110 # on the stepwidth
1111 # min, max, stepwidth, default are int values
1112 # wraparound: pressing RIGHT key at max value brings you to min value and vice versa if set to True
1113 class ConfigSelectionNumber(ConfigSelection):
1114         def __init__(self, min, max, stepwidth, default = None, wraparound = False):
1115                 self.wraparound = wraparound
1116                 if default is None:
1117                         default = min
1118                 default = str(default)
1119                 choices = []
1120                 step = min
1121                 while step <= max:
1122                         choices.append(str(step))
1123                         step += stepwidth
1124
1125                 ConfigSelection.__init__(self, choices, default)
1126
1127         def getValue(self):
1128                 return int(ConfigSelection.getValue(self))
1129
1130         def setValue(self, val):
1131                 ConfigSelection.setValue(self, str(val))
1132
1133         def handleKey(self, key):
1134                 if not self.wraparound:
1135                         if key == KEY_RIGHT:
1136                                 if len(self.choices) == (self.choices.index(self.value) + 1):
1137                                         return
1138                         if key == KEY_LEFT:
1139                                 if self.choices.index(self.value) == 0:
1140                                         return
1141                 ConfigSelection.handleKey(self, key)
1142
1143 class ConfigNumber(ConfigText):
1144         def __init__(self, default = 0):
1145                 ConfigText.__init__(self, str(default), fixed_size = False)
1146
1147         def getValue(self):
1148                 return int(self.text)
1149
1150         def setValue(self, val):
1151                 self.text = str(val)
1152
1153         value = property(getValue, setValue)
1154         _value = property(getValue, setValue)
1155
1156         def isChanged(self):
1157                 sv = self.saved_value
1158                 strv = self.tostring(self.value)
1159                 if sv is None and strv == self.default:
1160                         return False
1161                 return strv != sv
1162
1163         def conform(self):
1164                 pos = len(self.text) - self.marked_pos
1165                 self.text = self.text.lstrip("0")
1166                 if self.text == "":
1167                         self.text = "0"
1168                 if pos > len(self.text):
1169                         self.marked_pos = 0
1170                 else:
1171                         self.marked_pos = len(self.text) - pos
1172
1173         def handleKey(self, key):
1174                 if key in KEY_NUMBERS or key == KEY_ASCII:
1175                         if key == KEY_ASCII:
1176                                 ascii = getPrevAsciiCode()
1177                                 if not (48 <= ascii <= 57):
1178                                         return
1179                         else:
1180                                 ascii = getKeyNumber(key) + 48
1181                         newChar = unichr(ascii)
1182                         if self.allmarked:
1183                                 self.deleteAllChars()
1184                                 self.allmarked = False
1185                         self.insertChar(newChar, self.marked_pos, False)
1186                         self.marked_pos += 1
1187                 else:
1188                         ConfigText.handleKey(self, key)
1189                 self.conform()
1190
1191         def onSelect(self, session):
1192                 self.allmarked = (self.value != "")
1193
1194         def onDeselect(self, session):
1195                 self.marked_pos = 0
1196                 self.offset = 0
1197                 if not self.last_value == self.value:
1198                         self.changedFinal()
1199                         self.last_value = self.value
1200
1201 class ConfigSearchText(ConfigText):
1202         def __init__(self, default = "", fixed_size = False, visible_width = False):
1203                 ConfigText.__init__(self, default = default, fixed_size = fixed_size, visible_width = visible_width)
1204                 NumericalTextInput.__init__(self, nextFunc = self.nextFunc, handleTimeout = False, search = True)
1205
1206 class ConfigDirectory(ConfigText):
1207         def __init__(self, default="", visible_width=60):
1208                 ConfigText.__init__(self, default, fixed_size = True, visible_width = visible_width)
1209
1210         def handleKey(self, key):
1211                 pass
1212
1213         def getValue(self):
1214                 if self.text == "":
1215                         return None
1216                 else:
1217                         return ConfigText.getValue(self)
1218
1219         def setValue(self, val):
1220                 if val == None:
1221                         val = ""
1222                 ConfigText.setValue(self, val)
1223
1224         def getMulti(self, selected):
1225                 if self.text == "":
1226                         return ("mtext"[1-selected:], _("List of Storage Devices"), range(0))
1227                 else:
1228                         return ConfigText.getMulti(self, selected)
1229
1230         def onSelect(self, session):
1231                 self.allmarked = (self.value != "")
1232
1233 # a slider.
1234 class ConfigSlider(ConfigElement):
1235         def __init__(self, default = 0, increment = 1, limits = (0, 100)):
1236                 ConfigElement.__init__(self)
1237                 self.value = self.last_value = self.default = default
1238                 self.min = limits[0]
1239                 self.max = limits[1]
1240                 self.increment = increment
1241
1242         def checkValues(self):
1243                 if self.value < self.min:
1244                         self.value = self.min
1245
1246                 if self.value > self.max:
1247                         self.value = self.max
1248
1249         def handleKey(self, key):
1250                 if key == KEY_LEFT:
1251                         self.value -= self.increment
1252                 elif key == KEY_RIGHT:
1253                         self.value += self.increment
1254                 elif key == KEY_HOME:
1255                         self.value = self.min
1256                 elif key == KEY_END:
1257                         self.value = self.max
1258                 else:
1259                         return
1260                 self.checkValues()
1261
1262         def getText(self):
1263                 return "%d / %d" % (self.value, self.max)
1264
1265         def getMulti(self, selected):
1266                 self.checkValues()
1267                 return ("slider", self.value, self.max)
1268
1269         def fromstring(self, value):
1270                 return int(value)
1271
1272 # a satlist. in fact, it's a ConfigSelection.
1273 class ConfigSatlist(ConfigSelection):
1274         def __init__(self, list, default = None):
1275                 if default is not None:
1276                         default = str(default)
1277                 ConfigSelection.__init__(self, choices = [(str(orbpos), desc) for (orbpos, desc, flags) in list], default = default)
1278
1279         def getOrbitalPosition(self):
1280                 if self.value == "":
1281                         return None
1282                 return int(self.value)
1283
1284         orbital_position = property(getOrbitalPosition)
1285
1286 class ConfigSet(ConfigElement):
1287         def __init__(self, choices, default = []):
1288                 ConfigElement.__init__(self)
1289                 if isinstance(choices, list):
1290                         choices.sort()
1291                         self.choices = choicesList(choices, choicesList.LIST_TYPE_LIST)
1292                 else:
1293                         assert False, "ConfigSet choices must be a list!"
1294                 if default is None:
1295                         default = []
1296                 self.pos = -1
1297                 default.sort()
1298                 self.last_value = self.default = default
1299                 self.value = default[:]
1300
1301         def toggleChoice(self, choice):
1302                 value = self.value
1303                 if choice in value:
1304                         value.remove(choice)
1305                 else:
1306                         value.append(choice)
1307                         value.sort()
1308                 self.changed()
1309
1310         def handleKey(self, key):
1311                 if key in KEY_NUMBERS + [KEY_DELETE, KEY_BACKSPACE]:
1312                         if self.pos != -1:
1313                                 self.toggleChoice(self.choices[self.pos])
1314                 elif key == KEY_LEFT:
1315                         if self.pos < 0:
1316                                 self.pos = len(self.choices)-1
1317                         else:
1318                                 self.pos -= 1
1319                 elif key == KEY_RIGHT:
1320                         if self.pos >= len(self.choices)-1:
1321                                 self.pos = -1
1322                         else:
1323                                 self.pos += 1
1324                 elif key in (KEY_HOME, KEY_END):
1325                         self.pos = -1
1326
1327         def genString(self, lst):
1328                 res = ""
1329                 for x in lst:
1330                         res += self.description[x]+" "
1331                 return res
1332
1333         def getText(self):
1334                 return self.genString(self.value)
1335
1336         def getMulti(self, selected):
1337                 if not selected or self.pos == -1:
1338                         return ("text", self.genString(self.value))
1339                 else:
1340                         tmp = self.value[:]
1341                         ch = self.choices[self.pos]
1342                         mem = ch in self.value
1343                         if not mem:
1344                                 tmp.append(ch)
1345                                 tmp.sort()
1346                         ind = tmp.index(ch)
1347                         val1 = self.genString(tmp[:ind])
1348                         val2 = " "+self.genString(tmp[ind+1:])
1349                         if mem:
1350                                 chstr = " "+self.description[ch]+" "
1351                         else:
1352                                 chstr = "("+self.description[ch]+")"
1353                         len_val1 = len(val1)
1354                         return ("mtext", val1+chstr+val2, range(len_val1, len_val1 + len(chstr)))
1355
1356         def onDeselect(self, session):
1357                 self.pos = -1
1358                 if not self.last_value == self.value:
1359                         self.changedFinal()
1360                         self.last_value = self.value[:]
1361
1362         def tostring(self, value):
1363                 return str(value)
1364
1365         def fromstring(self, val):
1366                 return eval(val)
1367
1368         description = property(lambda self: descriptionList(self.choices.choices, choicesList.LIST_TYPE_LIST))
1369
1370 class ConfigLocations(ConfigElement):
1371         def __init__(self, default = [], visible_width = False):
1372                 ConfigElement.__init__(self)
1373                 self.visible_width = visible_width
1374                 self.pos = -1
1375                 self.default = default
1376                 self.locations = []
1377                 self.mountpoints = []
1378                 self.value = default[:]
1379
1380         def setValue(self, value):
1381                 locations = self.locations
1382                 loc = [x[0] for x in locations if x[3]]
1383                 add = [x for x in value if not x in loc]
1384                 diff = add + [x for x in loc if not x in value]
1385                 locations = [x for x in locations if not x[0] in diff] + [[x, self.getMountpoint(x), True, True] for x in add]
1386                 locations.sort(key = lambda x: x[0])
1387                 self.locations = locations
1388                 self.changed()
1389
1390         def getValue(self):
1391                 self.checkChangedMountpoints()
1392                 locations = self.locations
1393                 for x in locations:
1394                         x[3] = x[2]
1395                 return [x[0] for x in locations if x[3]]
1396
1397         value = property(getValue, setValue)
1398
1399         def tostring(self, value):
1400                 return str(value)
1401
1402         def fromstring(self, val):
1403                 return eval(val)
1404
1405         def load(self):
1406                 sv = self.saved_value
1407                 if sv is None:
1408                         tmp = self.default
1409                 else:
1410                         tmp = self.fromstring(sv)
1411                 locations = [[x, None, False, False] for x in tmp]
1412                 self.refreshMountpoints()
1413                 for x in locations:
1414                         if fileExists(x[0]):
1415                                 x[1] = self.getMountpoint(x[0])
1416                                 x[2] = True
1417                 self.locations = locations
1418
1419         def save(self):
1420                 locations = self.locations
1421                 if self.save_disabled or not locations:
1422                         self.saved_value = None
1423                 else:
1424                         self.saved_value = self.tostring([x[0] for x in locations])
1425
1426         def isChanged(self):
1427                 sv = self.saved_value
1428                 locations = self.locations
1429                 if val is None and not locations:
1430                         return False
1431                 return self.tostring([x[0] for x in locations]) != sv
1432
1433         def addedMount(self, mp):
1434                 for x in self.locations:
1435                         if x[1] == mp:
1436                                 x[2] = True
1437                         elif x[1] == None and fileExists(x[0]):
1438                                 x[1] = self.getMountpoint(x[0])
1439                                 x[2] = True
1440
1441         def removedMount(self, mp):
1442                 for x in self.locations:
1443                         if x[1] == mp:
1444                                 x[2] = False
1445
1446         def refreshMountpoints(self):
1447                 self.mountpoints = [p.mountpoint for p in harddiskmanager.getMountedPartitions() if p.mountpoint != "/"]
1448                 self.mountpoints.sort(key = lambda x: -len(x))
1449
1450         def checkChangedMountpoints(self):
1451                 oldmounts = self.mountpoints
1452                 self.refreshMountpoints()
1453                 newmounts = self.mountpoints
1454                 if oldmounts == newmounts:
1455                         return
1456                 for x in oldmounts:
1457                         if not x in newmounts:
1458                                 self.removedMount(x)
1459                 for x in newmounts:
1460                         if not x in oldmounts:
1461                                 self.addedMount(x)
1462
1463         def getMountpoint(self, file):
1464                 file = os_path.realpath(file)+"/"
1465                 for m in self.mountpoints:
1466                         if file.startswith(m):
1467                                 return m
1468                 return None
1469
1470         def handleKey(self, key):
1471                 if key == KEY_LEFT:
1472                         self.pos -= 1
1473                         if self.pos < -1:
1474                                 self.pos = len(self.value)-1
1475                 elif key == KEY_RIGHT:
1476                         self.pos += 1
1477                         if self.pos >= len(self.value):
1478                                 self.pos = -1
1479                 elif key in (KEY_HOME, KEY_END):
1480                         self.pos = -1
1481
1482         def getText(self):
1483                 return " ".join(self.value)
1484
1485         def getMulti(self, selected):
1486                 if not selected:
1487                         valstr = " ".join(self.value)
1488                         if self.visible_width and len(valstr) > self.visible_width:
1489                                 return ("text", valstr[0:self.visible_width])
1490                         else:
1491                                 return ("text", valstr)
1492                 else:
1493                         i = 0
1494                         valstr = ""
1495                         ind1 = 0
1496                         ind2 = 0
1497                         for val in self.value:
1498                                 if i == self.pos:
1499                                         ind1 = len(valstr)
1500                                 valstr += str(val)+" "
1501                                 if i == self.pos:
1502                                         ind2 = len(valstr)
1503                                 i += 1
1504                         if self.visible_width and len(valstr) > self.visible_width:
1505                                 if ind1+1 < self.visible_width/2:
1506                                         off = 0
1507                                 else:
1508                                         off = min(ind1+1-self.visible_width/2, len(valstr)-self.visible_width)
1509                                 return ("mtext", valstr[off:off+self.visible_width], range(ind1-off,ind2-off))
1510                         else:
1511                                 return ("mtext", valstr, range(ind1,ind2))
1512
1513         def onDeselect(self, session):
1514                 self.pos = -1
1515
1516 # nothing.
1517 class ConfigNothing(ConfigSelection):
1518         def __init__(self):
1519                 ConfigSelection.__init__(self, choices = [("","")])
1520
1521 # until here, 'saved_value' always had to be a *string*.
1522 # now, in ConfigSubsection, and only there, saved_value
1523 # is a dict, essentially forming a tree.
1524 #
1525 # config.foo.bar=True
1526 # config.foobar=False
1527 #
1528 # turns into:
1529 # config.saved_value == {"foo": {"bar": "True"}, "foobar": "False"}
1530 #
1531
1532 class ConfigSubsectionContent(object):
1533         pass
1534
1535 # we store a backup of the loaded configuration
1536 # data in self.stored_values, to be able to deploy
1537 # them when a new config element will be added,
1538 # so non-default values are instantly available
1539
1540 # A list, for example:
1541 # config.dipswitches = ConfigSubList()
1542 # config.dipswitches.append(ConfigYesNo())
1543 # config.dipswitches.append(ConfigYesNo())
1544 # config.dipswitches.append(ConfigYesNo())
1545 class ConfigSubList(list, object):
1546         def __init__(self):
1547                 list.__init__(self)
1548                 self.stored_values = {}
1549
1550         def save(self):
1551                 for x in self:
1552                         x.save()
1553
1554         def load(self):
1555                 for x in self:
1556                         x.load()
1557
1558         def getSavedValue(self):
1559                 res = { }
1560                 for i, val in enumerate(self):
1561                         sv = val.saved_value
1562                         if sv is not None:
1563                                 res[str(i)] = sv
1564                 return res
1565
1566         def setSavedValue(self, values):
1567                 self.stored_values = dict(values)
1568                 for (key, val) in self.stored_values.items():
1569                         if int(key) < len(self):
1570                                 self[int(key)].saved_value = val
1571
1572         saved_value = property(getSavedValue, setSavedValue)
1573
1574         def append(self, item):
1575                 i = str(len(self))
1576                 list.append(self, item)
1577                 if i in self.stored_values:
1578                         item.saved_value = self.stored_values[i]
1579                         item.load()
1580
1581         def dict(self):
1582                 return dict([(str(index), value) for index, value in enumerate(self)])
1583
1584 # same as ConfigSubList, just as a dictionary.
1585 # care must be taken that the 'key' has a proper
1586 # str() method, because it will be used in the config
1587 # file.
1588 class ConfigSubDict(dict, object):
1589         def __init__(self):
1590                 dict.__init__(self)
1591                 self.stored_values = {}
1592
1593         def save(self):
1594                 for x in self.values():
1595                         x.save()
1596
1597         def load(self):
1598                 for x in self.values():
1599                         x.load()
1600
1601         def getSavedValue(self):
1602                 res = {}
1603                 for (key, val) in self.items():
1604                         sv = val.saved_value
1605                         if sv is not None:
1606                                 res[str(key)] = sv
1607                 return res
1608
1609         def setSavedValue(self, values):
1610                 self.stored_values = dict(values)
1611                 for (key, val) in self.items():
1612                         if str(key) in self.stored_values:
1613                                 val.saved_value = self.stored_values[str(key)]
1614
1615         saved_value = property(getSavedValue, setSavedValue)
1616
1617         def __setitem__(self, key, item):
1618                 dict.__setitem__(self, key, item)
1619                 if str(key) in self.stored_values:
1620                         item.saved_value = self.stored_values[str(key)]
1621                         item.load()
1622
1623         def dict(self):
1624                 return self
1625
1626 # Like the classes above, just with a more "native"
1627 # syntax.
1628 #
1629 # some evil stuff must be done to allow instant
1630 # loading of added elements. this is why this class
1631 # is so complex.
1632 #
1633 # we need the 'content' because we overwrite
1634 # __setattr__.
1635 # If you don't understand this, try adding
1636 # __setattr__ to a usual exisiting class and you will.
1637 class ConfigSubsection(object):
1638         def __init__(self):
1639                 self.__dict__["content"] = ConfigSubsectionContent()
1640                 self.content.items = { }
1641                 self.content.stored_values = { }
1642
1643         def __setattr__(self, name, value):
1644                 if name == "saved_value":
1645                         return self.setSavedValue(value)
1646                 assert isinstance(value, (ConfigSubsection, ConfigElement, ConfigSubList, ConfigSubDict)), "ConfigSubsections can only store ConfigSubsections, ConfigSubLists, ConfigSubDicts or ConfigElements"
1647                 content = self.content
1648                 content.items[name] = value
1649                 x = content.stored_values.get(name, None)
1650                 if x is not None:
1651                         #print "ok, now we have a new item,", name, "and have the following value for it:", x
1652                         value.saved_value = x
1653                         value.load()
1654
1655         def __getattr__(self, name):
1656                 return self.content.items[name]
1657
1658         def getSavedValue(self):
1659                 res = self.content.stored_values
1660                 for (key, val) in self.content.items.items():
1661                         sv = val.saved_value
1662                         if sv is not None:
1663                                 res[key] = sv
1664                         elif key in res:
1665                                 del res[key]
1666                 return res
1667
1668         def setSavedValue(self, values):
1669                 values = dict(values)
1670                 self.content.stored_values = values
1671                 for (key, val) in self.content.items.items():
1672                         value = values.get(key, None)
1673                         if value is not None:
1674                                 val.saved_value = value
1675
1676         saved_value = property(getSavedValue, setSavedValue)
1677
1678         def save(self):
1679                 for x in self.content.items.values():
1680                         x.save()
1681
1682         def load(self):
1683                 for x in self.content.items.values():
1684                         x.load()
1685
1686         def dict(self):
1687                 return self.content.items
1688
1689 # the root config object, which also can "pickle" (=serialize)
1690 # down the whole config tree.
1691 #
1692 # we try to keep non-existing config entries, to apply them whenever
1693 # a new config entry is added to a subsection
1694 # also, non-existing config entries will be saved, so they won't be
1695 # lost when a config entry disappears.
1696 class Config(ConfigSubsection):
1697         def __init__(self):
1698                 ConfigSubsection.__init__(self)
1699
1700         def pickle_this(self, prefix, topickle, result):
1701                 for (key, val) in topickle.items():
1702                         name = '.'.join((prefix, key))
1703                         if isinstance(val, dict):
1704                                 self.pickle_this(name, val, result)
1705                         elif isinstance(val, tuple):
1706                                 result += [name, '=', val[0], '\n']
1707                         else:
1708                                 result += [name, '=', val, '\n']
1709
1710         def pickle(self):
1711                 result = []
1712                 self.pickle_this("config", self.saved_value, result)
1713                 return ''.join(result)
1714
1715         def unpickle(self, lines, base_file=True):
1716                 tree = { }
1717                 for l in lines:
1718                         if not l or l[0] == '#':
1719                                 continue
1720
1721                         n = l.find('=')
1722                         name = l[:n]
1723                         val = l[n+1:].strip()
1724
1725                         names = name.split('.')
1726 #                       if val.find(' ') != -1:
1727 #                               val = val[:val.find(' ')]
1728
1729                         base = tree
1730
1731                         for n in names[:-1]:
1732                                 base = base.setdefault(n, {})
1733
1734                         base[names[-1]] = val
1735
1736                         if not base_file: # not the initial config file..
1737                                 #update config.x.y.value when exist
1738                                 try:
1739                                         configEntry = eval(name)
1740                                         if configEntry is not None:
1741                                                 configEntry.value = val
1742                                 except (SyntaxError, KeyError):
1743                                         pass
1744
1745                 # we inherit from ConfigSubsection, so ...
1746                 #object.__setattr__(self, "saved_value", tree["config"])
1747                 if "config" in tree:
1748                         self.setSavedValue(tree["config"])
1749
1750         def saveToFile(self, filename):
1751                 text = self.pickle()
1752                 try:
1753                         f = open(filename, "w")
1754                         f.write(text)
1755                         f.close()
1756                 except IOError:
1757                         print "Config: Couldn't write %s" % filename
1758
1759         def loadFromFile(self, filename, base_file=False):
1760                 f = open(filename, "r")
1761                 self.unpickle(f.readlines(), base_file)
1762                 f.close()
1763
1764 config = Config()
1765 config.misc = ConfigSubsection()
1766
1767 class ConfigFile:
1768         CONFIG_FILE = resolveFilename(SCOPE_CONFIG, "settings")
1769
1770         def load(self):
1771                 try:
1772                         config.loadFromFile(self.CONFIG_FILE, True)
1773                 except IOError, e:
1774                         print "unable to load config (%s), assuming defaults..." % str(e)
1775
1776         def save(self):
1777 #               config.save()
1778                 config.saveToFile(self.CONFIG_FILE)
1779
1780         def __resolveValue(self, pickles, cmap):
1781                 key = pickles[0]
1782                 if cmap.has_key(key):
1783                         if len(pickles) > 1:
1784                                 return self.__resolveValue(pickles[1:], cmap[key].dict())
1785                         else:
1786                                 return str(cmap[key].value)
1787                 return None
1788
1789         def getResolvedKey(self, key):
1790                 names = key.split('.')
1791                 if len(names) > 1:
1792                         if names[0] == "config":
1793                                 ret=self.__resolveValue(names[1:], config.content.items)
1794                                 if ret and len(ret):
1795                                         return ret
1796                 print "getResolvedKey", key, "failed !! (Typo??)"
1797                 return ""
1798
1799 def NoSave(element):
1800         element.disableSave()
1801         return element
1802
1803 configfile = ConfigFile()
1804
1805 configfile.load()
1806
1807 def getConfigListEntry(*args):
1808         assert len(args) > 1, "getConfigListEntry needs a minimum of two arguments (descr, configElement)"
1809         return args
1810
1811 def updateConfigElement(element, newelement):
1812         newelement.value = element.value
1813         return newelement
1814
1815 #def _(x):
1816 #       return x
1817 #
1818 #config.bla = ConfigSubsection()
1819 #config.bla.test = ConfigYesNo()
1820 #config.nim = ConfigSubList()
1821 #config.nim.append(ConfigSubsection())
1822 #config.nim[0].bla = ConfigYesNo()
1823 #config.nim.append(ConfigSubsection())
1824 #config.nim[1].bla = ConfigYesNo()
1825 #config.nim[1].blub = ConfigYesNo()
1826 #config.arg = ConfigSubDict()
1827 #config.arg["Hello"] = ConfigYesNo()
1828 #
1829 #config.arg["Hello"].handleKey(KEY_RIGHT)
1830 #config.arg["Hello"].handleKey(KEY_RIGHT)
1831 #
1832 ##config.saved_value
1833 #
1834 ##configfile.save()
1835 #config.save()
1836 #print config.pickle()