X-Git-Url: http://gitweb.pimeys.fr/?p=today.git;a=blobdiff_plain;f=today_server.py;h=ec60fe4ee9da16af02ca44b5266cc654507257d0;hp=cadb02dc1b331b735e47d8eb1434c1f455f3d3dd;hb=HEAD;hpb=4fdc7d41d82ff89bcb92cf6d93e2111399149c1e diff --git a/today_server.py b/today_server.py index cadb02d..ec60fe4 100755 --- a/today_server.py +++ b/today_server.py @@ -10,89 +10,33 @@ et répondre à un check. """ -import time, datetime import re +from lxml import etree import os import sys import urllib import json -import traceback os.chdir('/home/vincent/scripts/today/') -sys.path.append("/home/vincent/scripts/dtc/") -import dtc -#: Fichier où sont stockés les derniers IDs des trucs -store_published_file = "lasts_published" +#: Config serveur +import serverconfig -#: Fichier où est stockée une copie de la liste des derniers IDs *lus* -store_seen_file = "lasts_sync" - -def get_now(): - """Obtenir la date actuelle sous le bon format""" - timestamp = time.time() - now = datetime.datetime(*time.localtime(timestamp)[:7]) - return now - -def last_dtc(): - """Vérifie si il y a des quotes DTC non lues""" - return dtc.last_inserted() - -def last_xkcd(): - p = urllib.urlopen("http://xkcd.com") - t = p.read() - current_id = int(re.findall("Permanent link to this comic: http://xkcd.com/(.*?)/", t)[0]) - return current_id - -def last_xantah(): - p = urllib.urlopen("http://www.adoprixtoxis.com/lite/download/xantah_downloads.php") - t = p.read() - ids = re.findall("""
Xantah (.*?)
""", t) - ids = [int(i) for i in ids] - return max(ids) - -def last_visiteur(): - p = urllib.urlopen("http://www.levisiteurdufutur.com/episodes.html") - t = p.read() - # On parse - soup = BeautifulSoup.BeautifulSoup(t) - # On récupère les différentes saisons - saisons = soup.findAll("div", attrs={"id" : "episodes_list"}) - nsaisons = len(saisons) - # La saison en cours est la première dans la liste - episodes = saisons[0].findAll("div", attrs={"class" : "thumbCaption"}) - nepisodes = len(episodes) - return nsaisons * 100 + nepisodes +#: Récupération de toutes les nouveautés +import gather def get_file(): - """Récupère la liste des derniers ids de chaque truc, stockée dans le fichiers.""" - f = open(store_published_file) + """Récupère la liste des derniers ids de chaque truc, stockée dans le fichier.""" + f = open(serverconfig.store_published_file) news = json.load(f) f.close() return news def update_file(news): """Met à jour la liste des derniers ids dans le fichier.""" - f = open(store_published_file, 'w') + f = open(serverconfig.store_published_file, 'w') json.dump(news, f) f.close() -FETCHS = { - "xkcd" : last_xkcd, - "dtc" : last_dtc, - "xantah" : last_xantah, - "visiteur" : last_visiteur, - } - -def fetch_all(): - """Va chercher sur les différents sites les nouveaux trucs.""" - news = {} - for (k, f) in FETCHS.iteritems(): - try: - news[k] = f() - except Exception as e: - print "Erreur à la récupération de %s :" % k - traceback.print_exc() - return news def sync(): """Reçoit une requête de synchronisation.""" @@ -100,23 +44,27 @@ def sync(): t = sys.stdin.read() on_client = json.loads(t) # On récupère où en est le serveur dans le fichier idoine - if os.path.isfile(store_seen_file): - on_server = json.load(open(store_seen_file)) + if os.path.isfile(serverconfig.store_seen_file): + on_server = json.load(open(serverconfig.store_seen_file)) else: on_server = {} # On garde le maximum for k in set(on_client.keys() + on_server.keys()): on_server[k] = max(on_client.get(k, 0), on_server.get(k, 0)) # On enregistre ce nouveau dico - json.dump(on_server, open(store_seen_file, "w")) + json.dump(on_server, open(serverconfig.store_seen_file, "w")) # On envoie au client ce nouveau dico print json.dumps(on_server) if __name__ == "__main__": + DEBUG = ("--debug" in sys.argv) or ("--verbose" in sys.argv) or serverconfig.DEBUG if sys.argv[1] == "check": - news = fetch_all() - olds = get_file() - olds.update(news) + news = gather.gather() + if "--init" in sys.argv: + olds = news + else: + olds = get_file() + olds.update(news) update_file(olds) elif sys.argv[1] == "whatsup": news = get_file()