Revert "pimeys est temporairement down" : Il l'est plus \o/

[today.git] / today.py

diff --git a/today.py b/today.py
index 80d05fb3caef6e0c5e675307c023c65e1b9d37e0..e430fd881a3030ab8768ced77b3ca24f4ae4366d 100755 (executable)
--- a/today.py
+++ b/today.py
@@ -17,13 +17,14 @@ 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
         self.distant_server = "pimeys"
         #: path de today-server.py sur le serveur distant
@@ -40,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")
@@ -110,13 +115,6 @@ def get_timers():
     timers = add_title(u"Timers", timers)
     return timers
 
-def exists(l):
-    """Renvoie True si au moins un élément de l est vrai"""
-    for i in l:
-        if i:
-            return True
-    return False
-
 def get_birthdays(*search):
     """Obtenir la liste des anniversaires à venir,
        ou la liste des anniversaires correspondants à une recherche"""
@@ -143,7 +141,7 @@ def get_birthdays(*search):
         search = [i.lower() for i in search]
         tous = ("--all" in search)
         for date, nom in liste:
-            if tous or exists([term.lower() in nom.lower() for term in search]):
+            if tous or any([term.lower() in nom.lower() for term in search]):
                 thisyeardate = datetime.datetime(now.year, date.month, date.day)
                 delta = thisyeardate - now + datetime.timedelta(1)
                 age = now.year - date.year
@@ -194,7 +192,7 @@ def get_saints():
     for day in ["today", "tomorrow"]:
         ssaints = saints[day]
         for (sexe, saint) in ssaints:
-            if exists([firstname.lower() in saint.lower().split() for firstname in firstnames]):
+            if any([firstname.lower() in saint.lower().split() for firstname in firstnames]):
                 towish[day].append(sexe + saint)
     ttowish = []
     if towish["today"]:
@@ -235,7 +233,7 @@ def get_dtc(*args):
         cmd = "~/bin/dtc %s %s --json" % (args[0], args[1])
     else:
         return None
-    proc = subprocess.Popen(["ssh", config.distant_server, cmd], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    proc = subprocess.Popen(["ssh", "-4", config.distant_server, cmd], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     out, err = proc.communicate()
     out += err
     quotes = json.loads(out)
@@ -292,7 +290,7 @@ THINGS = {
 def check_all():
     """Vérifie si il y a des derniers trucs non lus/vus."""
     cmd = "%s whatsup" % (config.path_today_server,)
-    proc = subprocess.Popen(["ssh", config.distant_server, cmd], stdout=subprocess.PIPE)
+    proc = subprocess.Popen(["ssh", "-4", config.distant_server, cmd], stdout=subprocess.PIPE)
     out, err = proc.communicate()
     news = json.loads(out)
     seen = get_last_seen()
@@ -310,8 +308,9 @@ def check_all():
     return checks
 
 def get_everything():
-    """Récupère toutes les infos"""
-    work = [action() for action in AUTOMATED_ACTIONS.values()]
+    """Récupère toutes les infos."""
+    work = [action() for (keyword, action) in AUTOMATED_ACTIONS.iteritems()
+            if (config.do_remote or not keyword in REMOTE_ACTIONS)]
     chain = u"\n\n".join([result for result in work if result])
     return chain
 
@@ -342,7 +341,8 @@ def _update_lasttime(when):
 
 def ping():
     """Dit juste si il y a quelque chose à voir.
-       La première exécution de la journée peut être lente parce qu'elle va bosser avnt de répondre."""
+       La première exécution de la journée peut être lente parce qu'elle va bosser avant de répondre.
+       """
     now = get_now()
     lasttime = _get_lasttime()
     if (lasttime.date() < now.date()):
@@ -359,7 +359,7 @@ def ping():
             return u"You have something in %stoday%s" % (config.red, config.endcolor)
 
 def affiche():
-    """Action par défaut, affiche toutes les infos"""
+    """Action par défaut, affiche toutes les infos."""
     out = print_date()
     out += get_everything()
     _there_is_something_in_today(False)
@@ -376,7 +376,7 @@ def initialize():
     for ifile in range(len(files)):
         namefile = files[ifile]
         if os.path.isfile(namefile):
-            print "%s exists, skipping." % (namefile,)
+            print (u"%s exists, skipping." % (namefile,)).encode("utf-8")
         else:
             f = open(namefile, "w")
             f.write(contents[ifile].encode("utf-8"))
@@ -387,7 +387,7 @@ def sync():
        le maximum de chaque truc vu est gardé des deux côtés."""
     lasts = get_last_seen()
     cmd = "%s sync" % (config.path_today_server,)
-    proc = subprocess.Popen(["ssh", config.distant_server, cmd],
+    proc = subprocess.Popen(["ssh", "-4", config.distant_server, cmd],
                             stdin = subprocess.PIPE, stdout=subprocess.PIPE,
                             close_fds = True)
     lasts_raw = json.dumps(lasts)
@@ -396,7 +396,7 @@ def sync():
     out = proc.stdout.read()
     newdict = json.loads(out)
     update_last_seen(newdict)
-    print u"Nouvel état : %r" % newdict
+    print (u"Nouvel état : %r" % newdict).encode("utf-8")
     
     
 
@@ -428,6 +428,9 @@ OTHER_ACTIONS = {
            "init" : initialize,
            }
 
+#: Les actions qui nécessitent un accès au serveur distant
+REMOTE_ACTIONS = ["check", "sync"]
+
 #: Toutes les actions
 ACTIONS = dict(AUTOMATED_ACTIONS)
 ACTIONS.update(OTHER_ACTIONS)
@@ -437,7 +440,11 @@ 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
+        sys.argv.remove("--no-remote")
     if len(sys.argv) == 1:
         # Juste un today
         output = ACTIONS[None]()
@@ -446,4 +453,4 @@ if __name__ == "__main__":
         args = sys.argv[2:]
         output = ACTIONS[commande](*args)
     if output:
-        print output
+        print output.encode("utf-8")