Modularisation++ : module comics pour xkcd et jl8 (young DC)

[today.git] / today_server.py

diff --git a/today_server.py b/today_server.py
index 209dd04b0e6bbc39f50c244ddf9feea62e1dcf91..0c5016048f04f009ed06e71982598bbd0a1dd543 100755 (executable)
--- a/today_server.py
+++ b/today_server.py
@@ -10,63 +10,53 @@ et répondre à un check.
 
 """
 
-import time, datetime
 import re
+from lxml import etree
 import os
 import sys
 import urllib
 import json
 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"
+#: Récupération de toutes les nouveautés
+import gather
 
-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():
-    try:
-        p = urllib.urlopen("http://xkcd.com")
-    except IOError:
-        return "Impossible de se connecter à xkcd"
+def last_xantah():
+    p = urllib.urlopen("http://www.adoprixtoxis.com/lite/download/xantah_downloads.php")
     t = p.read()
-    current_id = int(re.findall("Permanent link to this comic: http://xkcd.com/(.*?)/", t)[0])
-    return current_id
+    ids = re.findall("""<div class="top">Xantah (.*?)</div>""", t)
+    ids = [int(i) for i in ids]
+    return max(ids)
 
 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,
          }
+
 def fetch_all():
     """Va chercher sur les différents sites les nouveaux trucs."""
     news = {}
     for (k, f) in FETCHS.iteritems():
-        news[k] = f()
+        try:
+            news[k] = f()
+        except Exception as e:
+            raise
+    news.update(gather.gather())
     return news
 
 def sync():
@@ -75,22 +65,28 @@ 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()
-        update_file(news)
+        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()
         print json.dumps(news)