today_server.py

   1 #!/usr/bin/python
   2 # -*- encoding: utf-8 -*-
   3
   4 """ Codé par 20-100
   5 script qui affiche des trucs à penser, des J-n des conneries
   6 or that kind of stuff.
   7
   8 Partie serveur, prévue pour chercher périodiquement les trucs non lus
   9 et répondre à un check.
  10
  11 """
  12
  13 import time, datetime
  14 import re
  15 import os
  16 import sys
  17 import urllib
  18 import json
  19 import traceback
  20 os.chdir('/home/vincent/scripts/today/')
  21 sys.path.append("/home/vincent/scripts/dtc/")
  22 import dtc
  23
  24 #: Fichier où sont stockés les derniers IDs des trucs
  25 store_published_file = "lasts_published"
  26
  27 #: Fichier où est stockée une copie de la liste des derniers IDs *lus*
  28 store_seen_file = "lasts_sync"
  29
  30 def get_now():
  31     """Obtenir la date actuelle sous le bon format"""
  32     timestamp = time.time()
  33     now = datetime.datetime(*time.localtime(timestamp)[:7])
  34     return now
  35
  36 def last_dtc():
  37     """Vérifie si il y a des quotes DTC non lues"""
  38     return dtc.last_inserted()
  39
  40 def last_xkcd():
  41     p = urllib.urlopen("http://xkcd.com")
  42     t = p.read()
  43     current_id = int(re.findall("Permanent link to this comic: http://xkcd.com/(.*?)/", t)[0])
  44     return current_id
  45
  46 def last_xantah():
  47     p = urllib.urlopen("http://www.adoprixtoxis.com/lite/download/xantah_downloads.php")
  48     t = p.read()
  49     ids = re.findall("""<div class="top">Xantah (.*?)</div>""", t)
  50     ids = [int(i) for i in ids]
  51     return max(ids)
  52
  53 def last_visiteur():
  54     p = urllib.urlopen("http://www.levisiteurdufutur.com/episodes.html")
  55     t = p.read()
  56     # On parse
  57     soup = BeautifulSoup.BeautifulSoup(t)
  58     # On récupère les différentes saisons
  59     saisons = soup.findAll("div", attrs={"id" : "episodes_list"})
  60     nsaisons = len(saisons)
  61     # La saison en cours est la première dans la liste
  62     episodes = saisons[0].findAll("div", attrs={"class" : "thumbCaption"})
  63     nepisodes = len(episodes)
  64     return nsaisons * 100 + nepisodes
  65
  66 def get_file():
  67     """Récupère la liste des derniers ids de chaque truc, stockée dans le fichiers."""
  68     f = open(store_published_file)
  69     news = json.load(f)
  70     f.close()
  71     return news
  72
  73 def update_file(news):
  74     """Met à jour la liste des derniers ids dans le fichier."""
  75     f = open(store_published_file, 'w')
  76     json.dump(news, f)
  77     f.close()
  78
  79 FETCHS = {
  80           "xkcd" : last_xkcd,
  81           "dtc" : last_dtc,
  82           "xantah" : last_xantah,
  83           "visiteur" : last_visiteur,
  84          }
  85
  86 def fetch_all():
  87     """Va chercher sur les différents sites les nouveaux trucs."""
  88     news = {}
  89     for (k, f) in FETCHS.iteritems():
  90         try:
  91             news[k] = f()
  92         except Exception as e:
  93             print "Erreur à la récupération de %s :" % k
  94             traceback.print_exc()
  95     return news
  96
  97 def sync():
  98     """Reçoit une requête de synchronisation."""
  99     # On récupère où en est le client sur stdin
 100     t = sys.stdin.read()
 101     on_client = json.loads(t)
 102     # On récupère où en est le serveur dans le fichier idoine
 103     if os.path.isfile(store_seen_file):
 104         on_server = json.load(open(store_seen_file))
 105     else:
 106         on_server = {}
 107     # On garde le maximum
 108     for k in set(on_client.keys() + on_server.keys()):
 109         on_server[k] = max(on_client.get(k, 0), on_server.get(k, 0))
 110     # On enregistre ce nouveau dico
 111     json.dump(on_server, open(store_seen_file, "w"))
 112     # On envoie au client ce nouveau dico
 113     print json.dumps(on_server)
 114
 115 if __name__ == "__main__":
 116     if sys.argv[1] == "check":
 117         news = fetch_all()
 118         olds = get_file()
 119         olds.update(news)
 120         update_file(olds)
 121     elif sys.argv[1] == "whatsup":
 122         news = get_file()
 123         print json.dumps(news)
 124     elif sys.argv[1] == "sync":
 125         sync()