lib/python/Components/config.py

   1 from time import *
   2 from Tools.NumericalTextInput import *
   3 from Tools.Directories import *
   4
   5 class configFile:
   6         def __init__(self):
   7                 self.changed = 0
   8                 self.configElements = { }
   9                 try:
  10                         self.file = open(resolveFilename(SCOPE_CONFIG, "config"))
  11                 except IOError:
  12                         print "cannot open config file"
  13                         return
  14
  15                 while 1:
  16                         line = self.file.readline()
  17                         if line == "":
  18                                 break
  19
  20                         if line.startswith("#"):                #skip comments
  21                                 continue
  22
  23                         self.addElement(line)
  24                 self.file.close()
  25
  26         def addElement(self, line):
  27                 x = line.find("=")
  28                 if x > -1:
  29                         self.configElements[line[:x]] = line[x + 1:-1]
  30
  31         def getKey(self, key):
  32                 return self.configElements[key]
  33
  34         def setKey(self, key, value):
  35                 self.changed = 1
  36                 self.configElements[key] = value
  37
  38         def save(self):
  39                 if self.changed == 0:           #no changes, so no write to disk needed
  40                         return
  41
  42                 fileHandle = open(resolveFilename(SCOPE_CONFIG, "config"), "w")
  43
  44                 keys = self.configElements.keys()
  45                 keys.sort()
  46                 for x in keys:
  47                         wstr = x + "=" + self.configElements[x] + "\n"
  48
  49                         fileHandle.write(wstr)
  50
  51                 fileHandle.close()
  52
  53 def currentConfigSelectionElement(element):
  54         return element.vals[element.value][0]
  55
  56 def getConfigSelectionElement(element, value):
  57         count = 0
  58         for x in element.vals:
  59                 if x[0] == value:
  60                         return count
  61                 count += 1
  62         return -1
  63
  64 class configSelection:
  65         def __init__(self, parent):
  66                 self.parent = parent
  67
  68         def checkValues(self):
  69                 if self.parent.value < 0:
  70                         self.parent.value = len(self.parent.vals) - 1
  71                 elif(self.parent.value > (len(self.parent.vals) - 1)):
  72                         self.parent.value = 0
  73
  74         def cancel(self):
  75                 self.parent.reload()
  76
  77         def save(self):
  78                 self.parent.save()
  79
  80         def handleKey(self, key):
  81                 if key == config.key["prevElement"]:
  82                         self.parent.value = self.parent.value - 1
  83                 if key == config.key["nextElement"]:
  84                         self.parent.value = self.parent.value + 1
  85
  86                 self.checkValues()
  87
  88                 self.parent.change()
  89
  90         def __call__(self, selected):                   #needed by configlist
  91                 self.checkValues()
  92                 if isinstance(self.parent.vals[self.parent.value], str):
  93                         returnValue = _(self.parent.vals[self.parent.value])
  94                 else:
  95                         returnValue = _(self.parent.vals[self.parent.value][1])
  96
  97                 return ("text", returnValue)
  98
  99 class configDateTime:
 100         def __init__(self, parent):
 101                 self.parent = parent
 102
 103         def checkValues(self):
 104                 pass
 105 #               if self.parent.value < 0:
 106                         #self.parent.value = 0
 107
 108                 #if(self.parent.value >= (len(self.parent.vals) - 1)):
 109                         #self.parent.value = len(self.parent.vals) - 1
 110
 111         def cancel(self):
 112                 self.parent.reload()
 113
 114         def save(self):
 115                 self.parent.save()
 116
 117         def handleKey(self, key):
 118                 if key == config.key["prevElement"]:
 119                         self.parent.value = self.parent.value - self.parent.vals[1]
 120                 if key == config.key["nextElement"]:
 121                         self.parent.value = self.parent.value + self.parent.vals[1]
 122
 123                 self.checkValues()
 124
 125                 self.parent.change()
 126
 127         def __call__(self, selected):                   #needed by configlist
 128                 self.checkValues()
 129                 return ("text", strftime(self.parent.vals[0], localtime(self.parent.value)))
 130
 131 class configSatlist:
 132         def __init__(self, parent):
 133                 self.parent = parent
 134
 135         def checkValues(self):
 136                 if self.parent.value < 0:
 137                         self.parent.value = 0
 138
 139                 if(self.parent.value >= (len(self.parent.vals) - 1)):
 140                         self.parent.value = len(self.parent.vals) - 1
 141
 142         def cancel(self):
 143                 self.parent.reload()
 144
 145         def save(self):
 146                 self.parent.save()
 147
 148         def handleKey(self, key):
 149                 if key == config.key["prevElement"]:
 150                         self.parent.value = self.parent.value - 1
 151                 if key == config.key["nextElement"]:
 152                         self.parent.value = self.parent.value + 1
 153
 154                 self.checkValues()
 155
 156                 self.parent.change()
 157
 158         def __call__(self, selected):                   #needed by configlist
 159                 self.checkValues()
 160                 #fixme
 161                 return ("text", str(self.parent.vals[self.parent.value][0]))
 162
 163 class configSequenceArg:
 164         def get(self, type, args = ()):
 165                 # configsequencearg.get ("IP")
 166                 if (type == "IP"):
 167                         return (("."), [(0,255),(0,255),(0,255),(0,255)], "")
 168                 # configsequencearg.get ("MAC")
 169                 if (type == "MAC"):
 170                         return ((":"), [(1,255),(1,255),(1,255),(1,255),(1,255),(1,255)], "")
 171                 # configsequencearg.get ("CLOCK")
 172                 if (type == "CLOCK"):
 173                         return ((":"), [(0,23),(0,59)], "")
 174                 # configsequencearg.get("INTEGER", (min, max)) => x with min <= x <= max
 175                 if (type == "INTEGER"):
 176                         return ((":"), [args], "")
 177                 # configsequencearg.get("PINCODE", (number, "*")) => pin with number = length of pincode and "*" as numbers shown as stars
 178                 # configsequencearg.get("PINCODE", (number, "")) => pin with number = length of pincode and numbers shown
 179                 if (type == "PINCODE"):
 180                         return ((":"), [(0, (10**args[0])-1)], args[1])
 181                 # configsequencearg.get("FLOAT", [(min,max),(min1,max1)]) => x.y with min <= x <= max and min1 <= y <= max1
 182                 if (type == "FLOAT"):
 183                         return (("."), args, "")
 184
 185 configsequencearg = configSequenceArg()
 186
 187 class configSequence:
 188         def __init__(self, parent):
 189                 self.parent = parent
 190                 self.markedPos = 0
 191                 self.seperator = self.parent.vals[0]
 192                 self.valueBounds = self.parent.vals[1]
 193                 self.censorChar = self.parent.vals[2]
 194
 195         def checkValues(self):
 196                 maxPos = 0
 197                 num = 0
 198                 for i in self.parent.value:
 199                         maxPos += len(str(self.valueBounds[num][1]))
 200                         while (self.valueBounds[num][0] > self.parent.value[num]):
 201                                 self.parent.value[num] += 1
 202
 203                         while (self.valueBounds[num][1] < self.parent.value[num]):
 204                                 self.parent.value[num] -= 1
 205
 206 #                       if (self.valueBounds[num][0] <= i <= self.valueBounds[num][1]):
 207                                 #pass
 208                         #else:
 209                                 #self.parent.value[num] = self.valueBounds[num][0]
 210                         num += 1
 211
 212                 if self.markedPos >= maxPos:
 213                         self.markedPos = maxPos - 1
 214                 if self.markedPos < 0:
 215                         self.markedPos = 0
 216
 217         def cancel(self):
 218                 self.parent.reload()
 219
 220         def save(self):
 221                 self.parent.save()
 222
 223         def handleKey(self, key):
 224                 #this will no change anything on the value itself
 225                 #so we can handle it here in gui element
 226                 if key == config.key["prevElement"]:
 227                         self.markedPos -= 1
 228                 if key == config.key["nextElement"]:
 229                         self.markedPos += 1
 230
 231                 if key >= config.key["0"] and key <= config.key["9"]:
 232                         self.blockLen = []
 233                         for x in self.valueBounds:
 234                                 self.blockLen.append(len(str(x[1])))
 235
 236                         pos = 0
 237                         blocknumber = 0
 238                         self.blockLenTotal = [0,]
 239                         for x in self.blockLen:
 240                                 pos += self.blockLen[blocknumber]
 241                                 self.blockLenTotal.append(pos)
 242                                 if (pos - 1 >= self.markedPos):
 243                                         pass
 244                                 else:
 245                                         blocknumber += 1
 246
 247                         number = 9 - config.key["9"] + key
 248                         # length of numberblock
 249                         numberLen = len(str(self.valueBounds[blocknumber][1]))
 250                         # position in the block
 251                         posinblock = self.markedPos - self.blockLenTotal[blocknumber]
 252
 253                         oldvalue = self.parent.value[blocknumber]
 254                         olddec = oldvalue % 10 ** (numberLen - posinblock) - (oldvalue % 10 ** (numberLen - posinblock - 1))
 255                         newvalue = oldvalue - olddec + (10 ** (numberLen - posinblock - 1) * number)
 256
 257                         print "You actually pressed a number (" + str(number) + ") which will be added at block number " + str(blocknumber) + " on position " + str(posinblock)
 258                         print "Old value: " + str(oldvalue) + " olddec: " + str(olddec) + " newvalue: " + str(newvalue)
 259                         self.parent.value[blocknumber] = newvalue
 260                         self.markedPos += 1
 261
 262                 self.checkValues()
 263
 264                 print "markPos:",
 265                 print self.markedPos
 266
 267                 #FIXME: dont call when press left/right
 268                 self.parent.change()
 269
 270         def __call__(self, selected):                   #needed by configlist
 271                 value = ""
 272                 mPos = self.markedPos
 273                 print "Positon: " + str(mPos)
 274                 num = 0;
 275                 for i in self.parent.value:
 276                         if len(value):  #fixme no heading separator possible
 277                                 value += self.seperator
 278                                 if mPos >= len(value) - 1:
 279                                         mPos += 1
 280
 281                         #diff =         self.valueBounds - len(str(i))
 282                         #if diff > 0:
 283                                 ## if this helps?!
 284                                 #value += " " * diff
 285                         print (("%0" + str(len(str(self.valueBounds[num][1]))) + "d") % i)
 286                         if (self.censorChar == ""):
 287                                 value += ("%0" + str(len(str(self.valueBounds[num][1]))) + "d") % i
 288                         else:
 289                                 value += (self.censorChar * len(str(self.valueBounds[num][1])))
 290                         num += 1
 291                         # only mark cursor when we are selected
 292                         # (this code is heavily ink optimized!)
 293                 if (self.parent.enabled == True):
 294                         return ("mtext"[1-selected:], value, [mPos])
 295                 else:
 296                         return ("text", value)
 297
 298 class configText:
 299         # used as first parameter
 300         # is the text of a fixed size or is the user able to extend the length of the text
 301         extendableSize = 1
 302         fixedSize = 2
 303
 304         def __init__(self, parent):
 305                 self.parent = parent
 306                 self.markedPos = 0
 307                 self.mode = self.parent.vals[0]
 308                 self.textInput = NumericalTextInput(self.nextEntry)
 309
 310         def checkValues(self):
 311                 if (self.markedPos < 0):
 312                         self.markedPos = 0
 313                 if (self.markedPos >= len(self.parent.value)):
 314                         self.markedPos = len(self.parent.value) - 1
 315
 316         def cancel(self):
 317                 self.parent.reload()
 318
 319         def save(self):
 320                 self.parent.save()
 321
 322         def nextEntry(self):
 323                 print self.parent
 324                 self.parent.vals[1](self.parent.getConfigPath())
 325
 326         def handleKey(self, key):
 327                 #this will no change anything on the value itself
 328                 #so we can handle it here in gui element
 329                 if key == config.key["delete"]:
 330                         self.parent.value = self.parent.value[0:self.markedPos] + self.parent.value[self.markedPos + 1:]
 331                 if key == config.key["prevElement"]:
 332                         self.textInput.nextKey()
 333                         self.markedPos -= 1
 334
 335                 if key == config.key["nextElement"]:
 336                         self.textInput.nextKey()
 337                         self.markedPos += 1
 338                         if (self.mode == self.extendableSize):
 339                                 if (self.markedPos >= len(self.parent.value)):
 340                                         self.parent.value = self.parent.value.ljust(len(self.parent.value) + 1)
 341
 342
 343                 if key >= config.key["0"] and key <= config.key["9"]:
 344                         number = 9 - config.key["9"] + key
 345
 346                         self.parent.value = self.parent.value[0:self.markedPos] + str(self.textInput.getKey(number)) + self.parent.value[self.markedPos + 1:]
 347
 348                 self.checkValues()
 349
 350                 self.parent.change()
 351
 352         def __call__(self, selected):                   #needed by configlist
 353                 return ("mtext"[1-selected:], str(self.parent.value), [self.markedPos])
 354
 355 class configValue:
 356         def __init__(self, obj):
 357                 self.obj = obj
 358
 359         def __str__(self):
 360                 return self.obj
 361
 362 class Config:
 363         def __init__(self):
 364                 self.key = { "choseElement": 0,
 365                                          "prevElement": 1,
 366                                          "nextElement": 2,
 367                                          "delete": 3,
 368                                          "0": 10,
 369                                          "1": 11,
 370                                          "2": 12,
 371                                          "3": 13,
 372                                          "4": 14,
 373                                          "5": 15,
 374                                          "6": 16,
 375                                          "7": 17,
 376                                          "8": 18,
 377                                          "9": 19 }
 378
 379 config = Config();
 380
 381 configfile = configFile()
 382
 383 class ConfigSlider:
 384         def __init__(self, parent):
 385                 self.parent = parent
 386
 387         def cancel(self):
 388                 self.parent.reload()
 389
 390         def save(self):
 391                 self.parent.save()
 392
 393         def checkValues(self):
 394                 if self.parent.value < 0:
 395                         self.parent.value = 0
 396
 397                 if self.parent.value > 10:
 398                         self.parent.value = 10
 399
 400         def handleKey(self, key):
 401                 if key == config.key["prevElement"]:
 402                         self.parent.value = self.parent.value - 1
 403                 if key == config.key["nextElement"]:
 404                         self.parent.value = self.parent.value + 1
 405
 406                 self.checkValues()
 407                 self.parent.change()
 408
 409         def __call__(self, selected):                   #needed by configlist
 410                 self.checkValues()
 411                 return ("slider", self.parent.value * 10)
 412
 413 class ConfigSubsection:
 414         def __init__(self):
 415                 pass
 416
 417 class configElement:
 418
 419         def getIndexbyEntry(self, data):
 420                 cnt = 0;
 421                 tcnt = -1; #for defaultval
 422                 for x in self.vals:
 423                         if int(x[1]) == int(data):
 424                                         return cnt
 425                         if int(x[1]) == int(self.defaultValue):
 426                                         tcnt = cnt
 427                         cnt += 1
 428                 if tcnt != -1:
 429                         return tcnt
 430                 return 0        #prevent bigger then array
 431
 432         def datafromFile(self, control, data):
 433                 if control == ConfigSlider:
 434                         return int(data)
 435                 elif control == configSelection:
 436                         try:
 437                                 return int(data)
 438                         except:
 439                                 for x in data.split(":"):
 440                                         if x[0] == "*":
 441                                                 count = 0
 442                                                 for y in self.vals:
 443                                                         if y[0] == x[1:-1]:
 444                                                                 return count
 445                                                         count += 1
 446                                 return self.defaultValue
 447                 elif control == configDateTime:
 448                         return int(data)
 449                 elif control == configText:
 450                         return str(data)
 451                 elif control == configSequence:
 452                         list = [ ]
 453                         part = data.split(self.vals[0])
 454                         for x in part:
 455                                 list.append(int(x))
 456                         return list
 457                 elif control == configSatlist:
 458                         return self.getIndexbyEntry(data)
 459                 else:
 460                         return ""
 461
 462         def datatoFile(self, control, data):
 463                 if control == ConfigSlider:
 464                         return str(data)
 465                 elif control == configSelection:
 466                         if isinstance(self.vals[data], str):
 467                                 return str(data)
 468                         else:
 469                                 confList = []
 470                                 count = 0
 471                                 for x in self.vals:
 472                                         if count == data:
 473                                                 confList.append("*" + str(x[0] + "*"))
 474                                         else:
 475                                                 confList.append(x[0])
 476                                         count += 1
 477                                 return ":".join(confList)
 478                         return str(data)
 479                 elif control == configDateTime:
 480                         return str(data)
 481                 elif control == configText:
 482                         return str(data.strip())
 483
 484                 elif control == configSequence:
 485                         print data
 486                         try:
 487                                 value = ((len(data) * ("%d" + self.vals[0]))[0:-1]) % tuple(data)
 488                         except:
 489                                 value = str(data)
 490 #                       just in case you don't understand the above, here an equivalent:
 491 #                       value = ""
 492 #                       for i in data:
 493 #                               if value !="":
 494 #                                       value += self.vals[0]
 495 #                               value += str(i)
 496                         return value
 497                 elif control == configSatlist:
 498                         return str(self.vals[self.value][1]);
 499                 else:
 500                         return ""
 501
 502         def loadData(self):
 503                 #print "load:" + self.configPath
 504                 try:
 505                         value = self.datafromFile(self.controlType, configfile.getKey(self.configPath))
 506                 except:
 507                         value = ""
 508
 509                 if value == "":
 510                         #print "value not found - using default"
 511
 512                         if self.controlType == configSatlist:
 513                                 self.value = self.getIndexbyEntry(self.defaultValue)
 514                         else:
 515                                 self.value = self.defaultValue
 516
 517                         self.save()             #add missing value to dict
 518                 else:
 519                         #print "set val:" + str(value)
 520                         self.value = value
 521
 522                 #is this right? activate settings after load/cancel and use default
 523                 self.change()
 524
 525         def __init__(self, configPath, control, defaultValue, vals):
 526                 self.configPath = configPath
 527                 self.defaultValue = defaultValue
 528                 self.controlType = control
 529                 self.vals = vals
 530                 self.notifierList = [ ]
 531                 self.enabled = True
 532                 self.loadData()
 533
 534         def getConfigPath(self):
 535                 return self.configPath
 536
 537         def addNotifier(self, notifier):
 538                 self.notifierList.append(notifier);
 539                 notifier(self);
 540         def change(self):
 541                 for notifier in self.notifierList:
 542                         notifier(self)
 543         def reload(self):
 544                 self.loadData()
 545         def save(self):
 546                 configfile.setKey(self.configPath, self.datatoFile(self.controlType,self.value))
 547
 548 class configElement_nonSave(configElement):
 549         def __init__(self, configPath, control, defaultValue, vals):
 550                 configElement.__init__(self, configPath, control, defaultValue, vals)
 551
 552         def save(self):
 553                 pass
 554
 555 def getConfigListEntry(description, element):
 556         b = element
 557         item = b.controlType(b)
 558         return ((description, item))
 559
 560
 561 def configElementBoolean(name, default, texts=(_("Enable"), _("Disable"))):
 562         return configElement(name, configSelection, default, texts)
 563
 564 config.misc = ConfigSubsection()