]> gitweb.pimeys.fr Git - today.git/blobdiff - today.py
La futurs annivs affichés le sont sur un espace temporel connexe
[today.git] / today.py
index 5b5162116339b71d5be6d2080f04e0717617559a..2fb940e2420faa754cb84446de1d8dfe9b14d944 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"""
@@ -87,7 +58,7 @@ def is_int(a):
     except ValueError:
         return False
 
-def update_last_seen(newdict):
+def update_last_seen(newdict, lasts_seen=None):
     """Met à jour un des derniers trucs vus/lus"""
     if not all([is_int(v) or (v == "last") for v in newdict.values()]):
         raise ValueError("Donner un entier ou last")
@@ -99,10 +70,12 @@ def update_last_seen(newdict):
             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()
+    if lasts_seen is None:
+        lasts_seen = get_last_seen()
     lasts_seen.update(newdict)
     with open(config.last_seen_file, "w") as f:
         f.write(json.dumps(lasts_seen))
+    return ", ".join(["last %s to %s" % (id, n) for (id, n) in newdict.items()])
 
 def parse_datefile(namefile):
     """Ouvre et parse un fichier avec des lignes de la forme
@@ -149,9 +122,7 @@ def get_birthdays(*search):
                     birthdays.append([config.red, u"%s a %s ans AUJOURD'HUI !" % (nom, age), 0, config.endcolor])
                 elif delta < datetime.timedelta(7):
                     birthdays.append([config.red, u"%s va avoir %s ans" % (nom, age), -delta.days, config.endcolor])
-                elif delta < datetime.timedelta(14):
-                    birthdays.append([u"", u"%s va avoir %s ans" % (nom, age), -delta.days, u""])
-                elif datetime.timedelta(30-4) < delta < datetime.timedelta(30+4):
+                elif delta < datetime.timedelta(30):
                     birthdays.append([u"", u"%s va avoir %s ans" % (nom, age), -delta.days, u""])
     else:
         # Recherche spécifique
@@ -262,56 +233,9 @@ def get_dtc(*args):
     textquotes = format_quotes(quotes)
     return textquotes
 
-def update_xkcd(newid):
-    update_last_seen({"xkcd" : newid})
-
-def update_xantah(newid):
-    update_last_seen({"xantah" : newid})
-
-def update_visiteur(newid):
-    update_last_seen({"visiteur" : newid})
-
-def update_noob(newid):
-    update_last_seen({"noob" : newid})
-
-def update_warpzone(newid):
-    update_last_seen({"warpzone" : newid})
-
-def update_hugo(newid):
-    update_last_seen({"hugo" : newid})
-
-def update_norman(newid):
-    update_last_seen({"norman" : newid})
-
-def update_cyprien(newid):
-    update_last_seen({"cyprien" : newid})
-
-def update_grenier(newid):
-    update_last_seen({"grenier" : newid})
-
-def update_jl8(newid):
-    update_last_seen({"dc" : newid})
-
-
-THINGS = {
-          "dtc" : u"Quotes DTC",
-          "xkcd" : u"Épisodes de XKCD",
-          "xantah" : u"Épisodes de La Légende de Xantah",
-          "visiteur" : u"Épisodes du Visiteur du Futur",
-          "noob" : u"Épisodes de NOOB",
-          "warpzone" : u"Épisodes de WARPZONE PROJECT",
-          "hugo" : u"Vidéos de Hugo Tout Seul",
-          "norman" : u"Vidéos de Norman",
-          "cyprien" : u"Vidéos de Cyprien",
-          "grenier" : u"Épisodes du joueur du grenier",
-          "dc" : u"Épisodes de la BD youngDC",
-
-          "birthdays" : u"Anniversaires à souhaiter",
-          "saints" : u"Fêtes à souhaiter",
-         }
 
 def get_whatsup():
-    """Récuère sur le serveur les derniers trucs."""
+    """Récupè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()
@@ -322,15 +246,14 @@ 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()
     checks = []
-    for (thing, comm) in THINGS.iteritems():
-        n = news[thing] - seen.get(thing, 0)
+    length = max([len(thing) for thing in news.keys()])
+    for (thing, lastid) in news.items():
+        n = lastid - seen.get(thing, 0)
         if type(n) != int:
             print n
         elif n > 0:
-            checks.append("%s : %s (last : %s)" % (comm, n, news[thing]))
+            checks.append("%s :%s%s (last : %s)" % (thing, " " *(length + 1 - len(thing)), n, news[thing]))
     checks = u"\n".join(checks)
     checks = add_title(u"Checks", checks)
     return checks
@@ -438,17 +361,6 @@ AUTOMATED_ACTIONS = {
 
 #: Les actions qu'on peut effectuer en rajoutant des paramètres
 OTHER_ACTIONS = {
-           "xkcd" : update_xkcd,
-           "xantah" : update_xantah,
-           "visiteur" : update_visiteur,
-           "noob" : update_noob,
-           "warpzone" : update_warpzone,
-           "hugo" : update_hugo,
-           "norman" : update_norman,
-           "cyprien" : update_cyprien,
-           "grenier" : update_grenier,
-           "dc" : update_jl8,
-           
            "dtc" : get_dtc,
            "ping" : ping,
            "show" : affiche,
@@ -474,12 +386,20 @@ if __name__ == "__main__":
     if "--no-remote" in sys.argv:
         config.do_remote = False
         sys.argv.remove("--no-remote")
+    output = None
     if len(sys.argv) == 1:
         # Juste un today
         output = ACTIONS[None]()
     else:
         commande = sys.argv[1]
         args = [s.decode("utf-8") for s in sys.argv[2:]]
-        output = ACTIONS[commande](*args)
+        if commande in ACTIONS:
+            output = ACTIONS[commande](*args)
+        else:
+            # on tente un update
+            lasts_seen  = get_last_seen()
+            if not commande in lasts_seen:
+                print "Attention : n'existait pas avant."
+            output = update_last_seen({commande : args[0]}, lasts_seen)
     if output:
         print output.encode("utf-8")