X-Git-Url: http://gitweb.pimeys.fr/?a=blobdiff_plain;f=today.py;h=5b5162116339b71d5be6d2080f04e0717617559a;hb=b786fa6bc6d8b8d92d13a680da857a62d8fccef2;hp=80d05fb3caef6e0c5e675307c023c65e1b9d37e0;hpb=7ad2e9146c3b7276edf5d36359bb1015a6cb156e;p=today.git diff --git a/today.py b/today.py index 80d05fb..5b51621 100755 --- 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"" - 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 @@ -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") @@ -74,13 +79,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 + 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 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 @@ -110,13 +132,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 +158,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 @@ -161,6 +176,8 @@ def get_birthdays(*search): def _parse_saint(s): """Renvoie la liste des fêtes contenue dans un jour""" l = s.split(",") + if l == [""]: + l = [] ll = [] for st in l: if st[0] == "&": @@ -194,7 +211,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 +252,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) @@ -246,31 +263,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" : newid}) THINGS = { @@ -284,17 +304,23 @@ THINGS = { "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 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", 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) + 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() @@ -310,8 +336,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 +369,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 +387,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 +404,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 +415,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 +424,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") @@ -419,6 +447,7 @@ OTHER_ACTIONS = { "norman" : update_norman, "cyprien" : update_cyprien, "grenier" : update_grenier, + "dc" : update_jl8, "dtc" : get_dtc, "ping" : ping, @@ -428,6 +457,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,13 +469,17 @@ 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]() else: commande = sys.argv[1] - args = sys.argv[2:] + args = [s.decode("utf-8") for s in sys.argv[2:]] output = ACTIONS[commande](*args) if output: - print output + print output.encode("utf-8")