Changeset 132:92f78dc5b965 for src/mlx/config.py

Timestamp:

04/30/12 10:55:37 (12 years ago)

Author:

István Váradi <ivaradi@…>

Branch:

default

hg-Phase:

(<MercurialRepository 1 'hg:/home/ivaradi/mlx/hg' '/'>, 'public')

Message:

Added support for enabling/disabling message types

File:

• src/mlx/config.py (modified) (8 diffs)

src/mlx/config.py

@@ -3 +3 @@
 
 #-------------------------------------------------------------------------------
+
+import const
 
 import os
@@ -24 +26 @@
     """Our configuration."""
     DEFAULT_UPDATE_URL = "http://mlx.varadiistvan.hu/update"
+
+    _messageTypesSection = "messageTypes"
     
     def __init__(self):
@@ -32 +36 @@
         self._rememberPassword = False
 
+        self._language = ""
+        self._flareTimeFromFS = False
+        
         self._autoUpdate = True        
         self._updateURL = Config.DEFAULT_UPDATE_URL
-        
-        self._language = ""
-        self._flareTimeFromFS = False
-
+
+        self._messageTypeLevels = {}
+        
         self._modified = False
 
@@ -102 +108 @@
             self._modified = True
 
+    def getMessageTypeLevel(self, messageType):
+        """Get the level for the given message type."""
+        return self._messageTypeLevels[messageType] \
+               if messageType in self._messageTypeLevels \
+               else const.MESSAGELEVEL_NONE
+
+    def setMessageTypeLevel(self, messageType, level):
+        """Set the level of the given message type."""
+        if messageType not in self._messageTypeLevels or \
+           self._messageTypeLevels[messageType]!=level:
+            self._messageTypeLevels[messageType] = level
+            self._modified = True
+
     @property
     def autoUpdate(self):
@@ -136 +155 @@
                                                   "rememberPassword", False)
 
-        self._autoUpdate = self._getBoolean(config, "update", "auto", True)
-        self._updateURL = self._get(config, "update", "url",
-                                    Config.DEFAULT_UPDATE_URL)
-
         self._language = self._get(config, "general", "language", "")
         self._flareTimeFromFS = self._getBoolean(config, "general",
                                                  "flareTimeFromFS",
                                                  False)
-        
+
+        self._messageTypeLevels = {}
+        for messageType in const.messageTypes:
+            self._messageTypeLevels[messageType] = \
+                self._getMessageTypeLevel(config, messageType)
+            
+        self._autoUpdate = self._getBoolean(config, "update", "auto", True)
+        self._updateURL = self._get(config, "update", "url",
+                                    Config.DEFAULT_UPDATE_URL)
+
         self._modified = False
 
@@ -160 +184 @@
                    "yes" if self._rememberPassword else "no")
 
-        config.add_section("update")
-        config.set("update", "auto",
-                   "yes" if self._autoUpdate else "no")
-        config.set("update", "url", self._updateURL)
-
         config.add_section("general")
         if self._language:
@@ -170 +189 @@
         config.set("general", "flareTimeFromFS",
                    "yes" if self._flareTimeFromFS else "no")
-        
+
+        config.add_section(Config._messageTypesSection)
+        for messageType in const.messageTypes:
+            if messageType in self._messageTypeLevels:
+                option = self._getMessageTypeLevelOptionName(messageType)
+                level = self._messageTypeLevels[messageType]                
+                config.set(Config._messageTypesSection, option,
+                           const.messageLevel2string(level))
+        
+        config.add_section("update")
+        config.set("update", "auto",
+                   "yes" if self._autoUpdate else "no")
+        config.set("update", "url", self._updateURL)
+
         try:
             fd = os.open(configPath, os.O_CREAT|os.O_TRUNC|os.O_WRONLY,
@@ -194 +226 @@
                else default
 
+    def _getMessageTypeLevel(self, config, messageType):
+        """Get the message type level for the given message type."""
+        option = self._getMessageTypeLevelOptionName(messageType)
+        if config.has_option(Config._messageTypesSection, option):
+            value = config.get(Config._messageTypesSection, option)
+            return const.string2messageLevel(value)
+        else:
+            return const.MESSAGELEVEL_NONE
+
+    def _getMessageTypeLevelOptionName(self, messageType):
+        """Get the option name for the given message type level."""
+        return const.messageType2string(messageType)
+        
     def getLanguage(self):
         """Get the language to be used."""