[client] Séparation de la config

[today.git] / today.py

diff --git a/today.py b/today.py
index 6d28b1fe591d7d4486be9640697ac08ef2e21457..41479d8c9f1efceba0d1b82cfa56553d7a1185bf 100755 (executable)
--- a/today.py
+++ b/today.py
@@ -14,37 +14,8 @@ import json
 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):
-        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
-        self.distant_server = "pimeys"
-        #: path de today-server.py sur le serveur distant
-        self.path_today_server = "/home/vincent/scripts/today/today_server.py"
-        #: Fichier contenant les anniversaires
-        self.birthdays_file = "birthdays.txt"
-        #: Fichier contenant les évènements à venir
-        self.timers_file = "timers.txt"
-        #: Fichier contenant les fêtes à souhaiter
-        self.saints_file = "saints.json"
-        #: Fichier contenant les ids des derniers trucs vus/lus
-        self.last_seen_file = "lasts"
-        #: Fichier contenant le timestamp de dernière exécution
-        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""
+# Import de la configuration
+import config
 
 if "--color" in sys.argv:
     sys.argv.remove("--color")
@@ -54,7 +25,7 @@ elif sys.stdout.isatty():
 else:
     color = False
 
-config = Config(color=color)
+config = config.Config(color=color)
 
 def print_date(timestamp=None,color=True):
     """Afficher la date"""
@@ -79,13 +50,30 @@ def get_last_seen():
     """Récupère la liste des derniers trucs vus/lus"""
     with open(config.last_seen_file) as f:
         return json.loads(f.read())
-    
+
+def is_int(a):
+    try:
+        int(a)
+        return True
+    except ValueError:
+        return False
+
 def update_last_seen(newdict):
     """Met à jour un des derniers trucs vus/lus"""
-    lasts = get_last_seen()
-    lasts.update(newdict)
+    if not all([is_int(v) or (v == "last") for v in newdict.values()]):
+        raise ValueError("Donner un entier ou last")
+    # Il est possible qu'on ait mis "last" quelque part
+    if "last" in newdict.values():
+        # il faut alors récupérer les derniers pour convertir "last" en <last_id>
+        lasts_published = get_whatsup()
+        for key in [k for (k, v) in newdict.items() if v == "last"]:
+            newdict[key] = lasts_published[key]
+    # On convertit les chaînes en entier (pour les <last_id> il leur arrive rien)
+    newdict = {k: int(v) for (k,v) in newdict.items()}
+    lasts_seen = get_last_seen()
+    lasts_seen.update(newdict)
     with open(config.last_seen_file, "w") as f:
-        f.write(json.dumps(lasts))
+        f.write(json.dumps(lasts_seen))
 
 def parse_datefile(namefile):
     """Ouvre et parse un fichier avec des lignes de la forme
@@ -246,34 +234,34 @@ def get_dtc(*args):
     return textquotes
 
 def update_xkcd(newid):
-    update_last_seen({"xkcd" : int(newid)})
+    update_last_seen({"xkcd" : newid})
 
 def update_xantah(newid):
-    update_last_seen({"xantah" : int(newid)})
+    update_last_seen({"xantah" : newid})
 
 def update_visiteur(newid):
-    update_last_seen({"visiteur" : int(newid)})
+    update_last_seen({"visiteur" : newid})
 
 def update_noob(newid):
-    update_last_seen({"noob" : int(newid)})
+    update_last_seen({"noob" : newid})
 
 def update_warpzone(newid):
-    update_last_seen({"warpzone" : int(newid)})
+    update_last_seen({"warpzone" : newid})
 
 def update_hugo(newid):
-    update_last_seen({"hugo" : int(newid)})
+    update_last_seen({"hugo" : newid})
 
 def update_norman(newid):
-    update_last_seen({"norman" : int(newid)})
+    update_last_seen({"norman" : newid})
 
 def update_cyprien(newid):
-    update_last_seen({"cyprien" : int(newid)})
+    update_last_seen({"cyprien" : newid})
 
 def update_grenier(newid):
-    update_last_seen({"grenier" : int(newid)})
+    update_last_seen({"grenier" : newid})
 
 def update_jl8(newid):
-    update_last_seen({"dc" : int(newid)})
+    update_last_seen({"dc" : newid})
 
 
 THINGS = {
@@ -293,12 +281,17 @@ THINGS = {
           "saints" : u"Fêtes à souhaiter",
          }
 
-def check_all():
-    """Vérifie si il y a des derniers trucs non lus/vus."""
+def get_whatsup():
+    """Récuère sur le serveur les derniers trucs."""
     cmd = "%s whatsup" % (config.path_today_server,)
     proc = subprocess.Popen(["ssh", "-4", config.distant_server, cmd], stdout=subprocess.PIPE)
     out, err = proc.communicate()
     news = json.loads(out)
+    return news
+
+def check_all():
+    """Vérifie si il y a des derniers trucs non lus/vus."""
+    news = get_whatsup()
     seen = get_last_seen()
     news["birthdays"] = check_birthdays()
     news["saints"] = check_saints()