X-Git-Url: http://gitweb.pimeys.fr/?p=today.git;a=blobdiff_plain;f=today.py;h=e430fd881a3030ab8768ced77b3ca24f4ae4366d;hp=c8a9e3194d6c1a9062a883e38c00cc62c6df4a2a;hb=b28fd9bf0a3d390494519bf1f825763906ff73a2;hpb=76969cd0e6992178db5f553250c6fa7d17a4c35f diff --git a/today.py b/today.py index c8a9e31..e430fd8 100755 --- a/today.py +++ b/today.py @@ -6,6 +6,7 @@ or that kind of stuff""" import time, datetime +import re import os import sys import subprocess @@ -16,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"" - self.blue = u"" - self.red = u"" - self.green = u"" - else: - self.endcolor=self.blue=self.red=self.green="" + self.endcolor = u"" + self.blue = u"" + self.red = u"" + self.green = u"" + 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 @@ -32,13 +34,17 @@ class Config(object): #: Fichier contenant les évènements à venir self.timers_file = "timers.txt" #: Fichier contenant les fêtes à souhaiter - self.saints_file = "saints.txt" + 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"" if "--color" in sys.argv: sys.argv.remove("--color") @@ -109,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""" @@ -142,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 @@ -157,6 +156,53 @@ def get_birthdays(*search): birthdays = add_title(u"Anniversaires", birthdays) return birthdays +def _parse_saint(s): + """Renvoie la liste des fêtes contenue dans un jour""" + l = s.split(",") + ll = [] + for st in l: + if st[0] == "&": + ll.append(["St ", st[1:]]) + elif st[0] == "!": + ll.append(["Ste ", st[1:]]) + return ll + +def _get_firstnames(): + """Récupère la liste des noms des gens connus""" + birthdays = parse_datefile(config.birthdays_file) + firstnames = [b[1].split()[0] for b in birthdays] + return firstnames + +def get_saints(): + """Renvoie la liste des fêtes à souhaiter aujourd'hui et demain""" + sourcesaints = json.load(open(config.saints_file)) + now = get_now() + saints = {} + saints["today"] = _parse_saint(sourcesaints[now.month - 1][now.day - 1]) + nbdays = len(sourcesaints[now.month - 1]) + if now.day == nbdays: # il faut regarder le mois suivant + if now.month == 12: # il faut regarder l'année suivante + saints["tomorrow"] = _parse_saint(sourcesaints[0][0]) + else: + saints["tomorrow"] = _parse_saint(sourcesaints[now.month][0]) + else: + saints["tomorrow"] = _parse_saint(sourcesaints[now.month - 1][now.day]) + firstnames = _get_firstnames() + towish = {"today" : [], "tomorrow" : []} + for day in ["today", "tomorrow"]: + ssaints = saints[day] + for (sexe, saint) in ssaints: + if any([firstname.lower() in saint.lower().split() for firstname in firstnames]): + towish[day].append(sexe + saint) + ttowish = [] + if towish["today"]: + ttowish.append(u"Aujourd'hui :\n" + "\n".join(towish["today"])) + if towish["tomorrow"]: + ttowish.append(u"Demain :\n" + "\n".join(towish["tomorrow"])) + saints = "\n".join(ttowish) + saints = add_title(u"Fêtes à souhaiter", saints) + return saints + def check_birthdays(): """Compte combien il y a d'anniversaires à afficher""" birthdays = get_birthdays() @@ -166,6 +212,11 @@ def check_birthdays(): n = 0 return n +def check_saints(): + """Compte combien il y a de fêtes à afficher""" + saints = get_saints() + return len(re.findall("\nSt", saints)) + def format_quotes(liste): """Formate les quotes de dicos à texte""" t = (u"\n" + u"_"*80 + u"\n").join([u"%(id)s (%(date)s)\n%(quote)s" % q for q in liste]) @@ -182,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) @@ -201,37 +252,65 @@ def update_xantah(newid): def update_visiteur(newid): update_last_seen({"visiteur" : int(newid)}) +def update_noob(newid): + update_last_seen({"noob" : int(newid)}) + +def update_warpzone(newid): + update_last_seen({"warpzone" : int(newid)}) + +def update_hugo(newid): + update_last_seen({"hugo" : int(newid)}) + +def update_norman(newid): + update_last_seen({"norman" : int(newid)}) + +def update_cyprien(newid): + update_last_seen({"cyprien" : int(newid)}) + +def update_grenier(newid): + update_last_seen({"grenier" : int(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", "birthdays" : u"Anniversaires à souhaiter", + "saints" : u"Fêtes à souhaiter", } 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() news["birthdays"] = check_birthdays() + news["saints"] = check_saints() checks = [] for (thing, comm) in THINGS.iteritems(): n = news[thing] - seen.get(thing, 0) if type(n) != int: print n elif n > 0: - checks.append("%s : %s" % (comm, n)) + checks.append("%s : %s (last : %s)" % (comm, n, news[thing])) checks = u"\n".join(checks) checks = add_title(u"Checks", checks) 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 @@ -262,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()): @@ -279,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) @@ -291,12 +371,12 @@ def initialize(): config.last_seen_file] contents = [u"# Anniversaires\n#jj/mm/aaaa Prénom Nom\n", u"# Évènements à venir\n#jj/mm/aaaa Évènement\n", - u"# Fêtes\n#jj/mm/aaaa Saint\n", + json.dumps([[""]*31]*12).decode(), u"{}"] 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")) @@ -307,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) @@ -316,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") @@ -324,6 +404,7 @@ def sync(): AUTOMATED_ACTIONS = { "timers" : get_timers, "birth" : get_birthdays, + "saints" : get_saints, "check" : check_all, } @@ -332,6 +413,12 @@ 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, "dtc" : get_dtc, "ping" : ping, @@ -341,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) @@ -350,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]() @@ -359,4 +453,4 @@ if __name__ == "__main__": args = sys.argv[2:] output = ACTIONS[commande](*args) if output: - print output + print output.encode("utf-8")