X-Git-Url: http://gitweb.pimeys.fr/?a=blobdiff_plain;ds=inline;f=today_server.py;h=ef73747f94dfa130555931a51c0ddd45a240b364;hb=dabb5eeb6a9c41bd3c415f2a73278d2fcd58dd80;hp=209dd04b0e6bbc39f50c244ddf9feea62e1dcf91;hpb=e752abb6ffde566d0527ad46043fd6c9460d739a;p=today.git
diff --git a/today_server.py b/today_server.py
index 209dd04..ef73747 100755
--- a/today_server.py
+++ b/today_server.py
@@ -10,8 +10,8 @@ et répondre à un check.
"""
-import time, datetime
import re
+from lxml import etree
import os
import sys
import urllib
@@ -20,53 +20,67 @@ 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
+#: Récupération de toutes les nouveautés
+import gather
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"
+ 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_jl8():
+ rss = urllib.urlopen("http://limbero.org/jl8/rss/")
+ t = rss.read()
+ x = etree.fromstring(t)
+ links = x.xpath("//link")
+ maxnum = links[1].text.split("/")[-1]
+ maxnum = int(maxnum)
+ return maxnum
+
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,
+ "dc" : last_jl8,
}
+
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 +89,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)