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
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")
"""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
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"""
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
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] == "&":
now = get_now()
saints = {}
saints["today"] = _parse_saint(sourcesaints[now.month - 1][now.day - 1])
- saints["tomorrow"] = _parse_saint(sourcesaints[now.month - 1][now.day])
+ 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 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"]:
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)
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 = {
"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()
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
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()):
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)
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"))
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)
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")
"norman" : update_norman,
"cyprien" : update_cyprien,
"grenier" : update_grenier,
+ "dc" : update_jl8,
"dtc" : get_dtc,
"ping" : ping,
"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)
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")