X-Git-Url: http://gitweb.pimeys.fr/?p=today.git;a=blobdiff_plain;f=today.py;h=2fb940e2420faa754cb84446de1d8dfe9b14d944;hp=c3f36295ec53afe20d457a5a9987e2e54f96ccc7;hb=HEAD;hpb=351bfd36ac35e89494d696987486e093425371f0 diff --git a/today.py b/today.py index c3f3629..2fb940e 100755 --- 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"" - 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 - 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,32 @@ 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 update_last_seen(newdict): + +def is_int(a): + try: + int(a) + return True + except ValueError: + return False + +def update_last_seen(newdict, lasts_seen=None): """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()} + 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)) + 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 @@ -132,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 @@ -159,6 +147,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] == "&": @@ -243,70 +233,27 @@ def get_dtc(*args): textquotes = format_quotes(quotes) return textquotes -def update_xkcd(newid): - update_last_seen({"xkcd" : int(newid)}) - -def update_xantah(newid): - update_last_seen({"xantah" : int(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)}) - -def update_jl8(newid): - update_last_seen({"dc" : 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", - "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é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() 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() 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 @@ -414,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, @@ -450,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")