annotate myrss/myrss_app.py @ 71:6318de36e334

pics2: limit access to index
author paulo
date Wed, 28 Oct 2015 00:57:23 -0700
parents ae0f2f438a95
children c7bbd3805509
rev   line source
paulo@39 1 import os
paulo@40 2 import sys
paulo@39 3 import re
paulo@40 4 import urllib2
paulo@40 5 import threading
paulo@40 6 import Queue
paulo@41 7 import datetime
paulo@41 8 import time
paulo@70 9 import traceback
paulo@47 10
paulo@42 11 import logging
paulo@68 12 logging.basicConfig(
paulo@70 13 #level=logging.DEBUG,
paulo@68 14 #filename="_LOG",
paulo@68 15 #format="%(asctime)s %(levelname)-8s %(message)s",
paulo@68 16 )
paulo@39 17
paulo@47 18 import xml.etree.ElementTree
paulo@47 19 import HTMLParser
paulo@47 20
paulo@39 21 import html
paulo@39 22
paulo@39 23
paulo@41 24 FEEDS_FILE = "FEEDS"
paulo@41 25 CACHE_HTML_FILE = "__cache__.html"
paulo@41 26
paulo@44 27 CACHE_LIFE = 1200 # [seconds]
paulo@47 28 MAX_ITEMS = 50
paulo@39 29 MAX_LINK_Z = 4
paulo@40 30 MAX_THREADS = 20
paulo@46 31 URLOPEN_TIMEOUT = 60 # [seconds]
paulo@39 32
paulo@39 33
paulo@39 34 _PARSE_ROOT_TAG_RE = re.compile(r"(\{(.+)\})?(.+)")
paulo@39 35
paulo@39 36 def _parse_root_tag(root_tag):
paulo@39 37 re_match = _PARSE_ROOT_TAG_RE.match(root_tag)
paulo@39 38
paulo@39 39 if re_match is None:
paulo@39 40 return (None, None)
paulo@39 41 else:
paulo@39 42 return re_match.group(2, 3)
paulo@39 43
paulo@39 44
paulo@47 45 def _strip_if_not_none(txt):
paulo@47 46 return txt.strip() if txt is not None else ''
paulo@47 47
paulo@47 48
paulo@39 49 def _go_rss(elementTree):
paulo@47 50 title = _strip_if_not_none(elementTree.find("channel/title").text)
paulo@39 51 link = elementTree.find("channel/link").text
paulo@39 52
paulo@39 53 items = []
paulo@39 54
paulo@39 55 for i in elementTree.findall("channel/item")[:MAX_ITEMS]:
paulo@47 56 it_title = _strip_if_not_none(i.find("title").text)
paulo@39 57 it_link = i.find("link").text
paulo@39 58
paulo@39 59 items.append((it_title, it_link))
paulo@39 60
paulo@39 61 return (title, link, items)
paulo@39 62
paulo@39 63
paulo@39 64 def _go_atom(elementTree):
paulo@39 65 ns = "http://www.w3.org/2005/Atom"
paulo@39 66
paulo@47 67 title = _strip_if_not_none(elementTree.find("{%s}title" % ns).text)
paulo@39 68 link = ''
paulo@39 69
paulo@39 70 for i in elementTree.findall("{%s}link" % ns):
paulo@39 71 if i.get("type") == "text/html" and i.get("rel") == "alternate":
paulo@39 72 link = i.get("href")
paulo@39 73 break
paulo@39 74
paulo@39 75 items = []
paulo@39 76
paulo@39 77 for i in elementTree.findall("{%s}entry" % ns)[:MAX_ITEMS]:
paulo@47 78 it_title = _strip_if_not_none(i.find("{%s}title" % ns).text)
paulo@39 79 it_link = ''
paulo@39 80
paulo@39 81 for j in i.findall("{%s}link" % ns):
paulo@39 82 if j.get("type") == "text/html" and j.get("rel") == "alternate":
paulo@39 83 it_link = j.get("href")
paulo@39 84 break
paulo@39 85
paulo@39 86 items.append((it_title, it_link))
paulo@39 87
paulo@39 88 return (title, link, items)
paulo@39 89
paulo@39 90
paulo@69 91 def _go_purl_rss(elementTree):
paulo@69 92 ns = "http://purl.org/rss/1.0/"
paulo@69 93
paulo@69 94 title = _strip_if_not_none(elementTree.find("{%s}channel/{%s}title" % (ns, ns)).text)
paulo@69 95 link = elementTree.find("{%s}channel/{%s}link" % (ns, ns)).text
paulo@69 96
paulo@69 97 items = []
paulo@69 98
paulo@69 99 for i in elementTree.findall("{%s}item" % ns)[:MAX_ITEMS]:
paulo@69 100 it_title = _strip_if_not_none(i.find("{%s}title" % ns).text)
paulo@69 101 it_link = i.find("{%s}link" % ns).text
paulo@69 102
paulo@69 103 items.append((it_title, it_link))
paulo@69 104
paulo@69 105 return (title, link, items)
paulo@69 106
paulo@69 107
paulo@47 108 _STRIP_HTML_RE = re.compile(r"<.*?>")
paulo@47 109 _htmlParser = HTMLParser.HTMLParser()
paulo@47 110
paulo@47 111 def _strip_html(txt):
paulo@47 112 return _htmlParser.unescape(_STRIP_HTML_RE.sub('', txt))
paulo@47 113
paulo@47 114
paulo@41 115 def _to_html(dtnow, docstruct):
paulo@41 116 datetime_str = dtnow.strftime("%Y-%m-%d %H:%M %Z")
paulo@41 117 page_title = "myrss -- %s" % datetime_str
paulo@41 118
paulo@42 119 root = html.HTML("html")
paulo@39 120
paulo@39 121 header = root.header
paulo@41 122 header.title(page_title)
paulo@39 123 header.link(rel="stylesheet", type="text/css", href="index.css")
paulo@39 124
paulo@41 125 body = root.body
paulo@41 126 body.h1(page_title)
paulo@41 127
paulo@39 128 link_z = 0
paulo@39 129
paulo@39 130 for feed in docstruct:
paulo@40 131 if feed is None:
paulo@40 132 continue
paulo@40 133
paulo@39 134 (title, link, items) = feed
paulo@39 135
paulo@47 136 body.h2.a(_strip_html(title), href=link, klass="z%d" % (link_z % MAX_LINK_Z))
paulo@39 137 link_z += 1
paulo@41 138 p = body.p
paulo@39 139
paulo@39 140 for (i, (it_title, it_link)) in enumerate(items):
paulo@39 141 if i > 0:
paulo@39 142 p += " - "
paulo@39 143
paulo@47 144 p.a(_strip_html(it_title), href=it_link, klass="z%d" % (link_z % MAX_LINK_Z))
paulo@39 145 link_z += 1
paulo@39 146
paulo@46 147 dtdelta = datetime.datetime.now() - dtnow
paulo@46 148 root.div("%.3f" % (dtdelta.days*86400 + dtdelta.seconds + dtdelta.microseconds/1e6), klass="debug")
paulo@46 149
paulo@39 150 return unicode(root).encode("utf-8")
paulo@39 151
paulo@39 152
paulo@47 153 def _fetch_url(url):
paulo@40 154 try:
paulo@42 155 logging.info("processing %s" % url)
paulo@46 156 feed = urllib2.urlopen(urllib2.Request(url, headers={"User-Agent": ''}), timeout=URLOPEN_TIMEOUT)
paulo@40 157 except urllib2.HTTPError as e:
paulo@42 158 logging.info("(%s) %s" % (url, e))
paulo@47 159 return None
paulo@47 160
paulo@47 161 return feed
paulo@47 162
paulo@47 163
paulo@47 164 def _process_feed(feed):
paulo@47 165 ret = None
paulo@40 166
paulo@40 167 elementTree = xml.etree.ElementTree.parse(feed)
paulo@40 168 root = elementTree.getroot()
paulo@40 169
paulo@40 170 parsed_root_tag = _parse_root_tag(root.tag)
paulo@40 171
paulo@40 172 if parsed_root_tag == (None, "rss"):
paulo@40 173 version = float(root.get("version", 0.0))
paulo@40 174 if version >= 2.0:
paulo@40 175 ret = _go_rss(elementTree)
paulo@40 176 else:
paulo@40 177 raise NotImplementedError("Unsupported rss version")
paulo@40 178 elif parsed_root_tag == ("http://www.w3.org/2005/Atom", "feed"):
paulo@40 179 ret = _go_atom(elementTree)
paulo@69 180 elif parsed_root_tag == ("http://www.w3.org/1999/02/22-rdf-syntax-ns#", "RDF"):
paulo@69 181 ret = _go_purl_rss(elementTree)
paulo@40 182 else:
paulo@40 183 raise NotImplementedError("Unknown root tag")
paulo@40 184
paulo@40 185 return ret
paulo@40 186
paulo@40 187
paulo@40 188 class WorkerThread(threading.Thread):
paulo@40 189 def __init__(self, *args, **kwargs):
paulo@40 190 self._input_queue = kwargs.pop("input_queue")
paulo@40 191 self._output_queue = kwargs.pop("output_queue")
paulo@40 192 threading.Thread.__init__(self, *args, **kwargs)
paulo@40 193 self.daemon = True
paulo@40 194
paulo@40 195 def run(self):
paulo@40 196 while True:
paulo@40 197 (idx, url) = self._input_queue.get()
paulo@40 198 docfeed = None
paulo@40 199 try:
paulo@47 200 feed = _fetch_url(url)
paulo@47 201 if feed is not None:
paulo@47 202 docfeed = _process_feed(feed)
paulo@40 203 except Exception as e:
paulo@42 204 logging.info("(%s) exception: %s" % (url, e))
paulo@40 205 self._output_queue.put((idx, docfeed))
paulo@40 206
paulo@40 207
paulo@44 208 def main(input_queue, output_queue, lock):
paulo@41 209 ret = ''
paulo@41 210
paulo@44 211 with lock:
paulo@44 212 epoch_now = time.time()
paulo@44 213 dtnow = datetime.datetime.fromtimestamp(epoch_now)
paulo@41 214
paulo@44 215 if os.path.exists(CACHE_HTML_FILE) and (epoch_now - os.stat(CACHE_HTML_FILE).st_mtime) < float(CACHE_LIFE):
paulo@44 216 with open(CACHE_HTML_FILE) as cache_html_file:
paulo@44 217 ret = cache_html_file.read()
paulo@41 218
paulo@44 219 else:
paulo@44 220 with open(FEEDS_FILE) as feeds_file:
paulo@44 221 feedlines = feeds_file.readlines()
paulo@41 222
paulo@44 223 docstruct = [None]*len(feedlines)
paulo@44 224 num_input = 0
paulo@44 225 for (i, l) in enumerate(feedlines):
paulo@44 226 if l[0] != '#':
paulo@44 227 l = l.strip()
paulo@44 228 input_queue.put((i, l))
paulo@44 229 num_input += 1
paulo@41 230
paulo@44 231 for _ in range(num_input):
paulo@44 232 (idx, docfeed) = output_queue.get()
paulo@44 233 docstruct[idx] = docfeed
paulo@41 234
paulo@44 235 ret = _to_html(dtnow, docstruct)
paulo@41 236
paulo@44 237 with open(CACHE_HTML_FILE, 'w') as cache_html_file:
paulo@44 238 cache_html_file.write(ret)
paulo@41 239
paulo@41 240 return ret
paulo@41 241
paulo@41 242
paulo@42 243 class MyRssApp:
paulo@42 244 def __init__(self):
paulo@42 245 self._iq = Queue.Queue(MAX_THREADS)
paulo@42 246 self._oq = Queue.Queue(MAX_THREADS)
paulo@44 247 self._main_lock = threading.Lock()
paulo@39 248
paulo@42 249 for _ in range(MAX_THREADS):
paulo@42 250 WorkerThread(input_queue=self._iq, output_queue=self._oq).start()
paulo@42 251
paulo@42 252 def __call__(self, environ, start_response):
paulo@70 253 response_code = "500 Internal Server Error"
paulo@70 254 response_type = "text/plain; charset=UTF-8"
paulo@70 255
paulo@70 256 try:
paulo@70 257 response_body = main(self._iq, self._oq, self._main_lock)
paulo@70 258 response_code = "200 OK"
paulo@70 259 response_type = "text/html; charset=UTF-8"
paulo@70 260 except:
paulo@70 261 response_body = traceback.format_exc()
paulo@70 262
paulo@42 263 response_headers = [
paulo@70 264 ("Content-Type", response_type),
paulo@42 265 ("Content-Length", str(len(response_body))),
paulo@42 266 ]
paulo@70 267 start_response(response_code, response_headers)
paulo@42 268
paulo@42 269 return [response_body]