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")
else:
color = False
-config = Config(color=color)
+config = config.Config(color=color)
def print_date(timestamp=None,color=True):
"""Afficher la date"""
"""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 <last_id>
+ 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 <last_id> 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
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" : 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",
- }
+ update_last_seen({"dc" : newid})
-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