Meilleure façon de gérer --no-color (en plus j'avais oublié green)

diff --git a/today.py b/today.py
index 9f9759419e9dcaffce0f73640336155b8a750387..e430fd881a3030ab8768ced77b3ca24f4ae4366d 100755 (executable)
--- a/today.py
+++ b/today.py
@@ -17,13 +17,12 @@ os.chdir('/home/vincent/scripts/today/')
 class Config(object):
     """Configuration (pas de couleurs si on n'output pas dans un terminal"""
     def __init__(self, color=True):
-        if color:
-            self.endcolor = u"\e[0m"
-            self.blue = u"\e[1;36m"
-            self.red = u"\e[1;31m"
-            self.green = u"\e[1;32m"
-        else:
-            self.endcolor = self.blue = self.red = self.green = ""
+        self.endcolor = u"\e[0m"
+        self.blue = u"\e[1;36m"
+        self.red = u"\e[1;31m"
+        self.green = u"\e[1;32m"
+        if not color:
+            self.nocolor()
         #: Faut-il contacter le serveur distant
         self.do_remote = True
         #: Serveur distant où aller récupérer les checks
@@ -42,6 +41,10 @@ class Config(object):
         self.lasttime_file = ".lasttime"
         #: Fichier contenant un booléen mémorisant si il y a quelquechose dans le today du jour et qu'il n'a pas encore été regardé
         self.something_file = ".something"
+    
+    def nocolor(self):
+        """Passe en mode sans couleur."""
+        self.endcolor = self.blue = self.red = self.green = u""
 
 if "--color" in sys.argv:
     sys.argv.remove("--color")
@@ -437,7 +440,7 @@ ACTIONS[None] = affiche # action par défaut
 if __name__ == "__main__":
     import sys
     if "--no-color" in sys.argv:
-        config.endcolor, config.red, config.blue = u"", u"", u""
+        config.nocolor()
         sys.argv.remove("--no-color")
     if "--no-remote" in sys.argv:
         config.do_remote = False