#!/usr/bin/python2.6 # -*- coding: UTF-8 -*- """Creates an Atom-Feed for single or multiple winterrodeln sled reports. Format: http://www.winterrodeln.org/feed/berichte/alle http://www.winterrodeln.org/feed/berichte/bahn/kemater_alm http://www.winterrodeln.org/feed/berichte/bahnen/5+280+251 See: http://www.atompub.org/ http://effbot.org/zone/element.htm http://www.winterrodeln.org/trac/wiki/UrlSchema """ import sys import datetime from xml.etree.ElementTree import Element, SubElement, tostring from sqlalchemy.engine import create_engine import logging from pylons import request, response, session, config, tmpl_context as c, url from pylons.controllers.util import abort, redirect from wrfeed.lib.base import BaseController, render log = logging.getLogger(__name__) def create_feed(page_title=None, page_ids=None): """If a page_title is given, only the reports for the given sledrun are shown. If a list of page_ids is given, only the reports for the selected pages are shown. Otherwise, all reports are shown.""" engine = create_engine(config['sqlalchemy.url']) limit = int(config['feedentrylimit']) conn = engine.connect() select = "select wrreport.page_id, wrreport.page_title, wrreport.id, date_report, date_entry, `condition`, description, author_name, author_userid, author_username, position_longitude, position_latitude from wrreport left outer join wrsledruncache on wrreport.page_id=wrsledruncache.page_id " if not page_title is None: # page_title is given page_title = page_title.replace('_', ' ') sql = select + "where lcase(wrreport.page_title) = lcase(%s) and date_invalid > now() and delete_date is null order by id desc limit {0}".format(limit) result = conn.execute(sql, page_title) elif not page_ids is None: # a list of page_ids is given sql = [select + "where "] if len(page_ids) > 0: sql += '(' sql += " or ".join(['wrreport.page_id=%s' for page_id in page_ids]) sql += ') ' sql += 'and date_invalid > now() and delete_date is null order by id desc limit {0}'.format(limit) page_ids_str = [str(page_id) for page_id in page_ids] result = conn.execute("".join(sql), *page_ids_str) else: # user wants to have all reports sql = select + "where date_invalid > now() and delete_date is null order by id desc limit {0}".format(limit) result = conn.execute(sql) feed = Element("feed", xmlns="http://www.w3.org/2005/Atom", attrib={'xmlns:georss': 'http://www.georss.org/georss', 'xmlns:wr': 'http://www.winterrodeln.org/schema/wrreport'}) feed_title = SubElement(feed, "title") feed_title.text = "Winterrodeln Rodelbahnberichte" feed_id = SubElement(feed, "id") if not page_title is None: feed_id.text = url(qualified=True, controller='berichte', action='bahn', id=page_title) elif not page_ids is None: feed_id.text = url(qualified=True, controller='berichte', action='bahnen', id="+".join(page_ids_str)) else: feed_id.text = url(qualified=True, controller='berichte', action='alle') feed_updated = SubElement(feed, "updated") feed.append(Element("link", rel="self", href=feed_id.text)) last_updated = None for row in result: page_id, page_title, report_id, date_report, date_entry, condition, description, author_name, author_userid, author_username, lon, lat = row page_title_url = page_title.replace(u' ', u'_') entry = SubElement(feed, "entry") entry_title = SubElement(entry, "title") entry_title.text = page_title entry.append(Element("link", rel="alternate", href=u"http://www.winterrodeln.org/wiki/{0}".format(page_title_url), type="text/html", hreflang="de")) entry_id = SubElement(entry, "id") entry_id.text = u"http://www.winterrodeln.org/wiki/{0}#{1}".format(page_title_url, report_id) entry_updated = SubElement(entry, "updated") entry_updated.text = date_entry.isoformat() + "+01:00" if last_updated is None: last_updated = date_entry # entry_summary = SubElement(entry, "summary") # entry_summary.text = str(condition) entry_content = SubElement(entry, "content") entry_content.attrib["type"] = "xhtml" entry_content_div = SubElement(entry_content, "div") entry_content_div.attrib["xmlns"] = "http://www.w3.org/1999/xhtml" entry_content_ul = SubElement(entry_content_div, "ul") if not date_report is None: entry_content_date = SubElement(entry_content_ul, "li") entry_content_date.text = u"Bericht für " + date_report.isoformat() if not condition is None: entry_content_condition = SubElement(entry_content_ul, "li") entry_content_condition.text = u"Schneelage: " + {1: u'Sehr gut', 2: u'Gut', 3: u'Mittelmäßig', 4: u'Schlecht', 5: u'Geht nicht'}[condition] entry_content_description = SubElement(entry_content_ul, "li") entry_content_description.text = description entry_author = SubElement(entry, "author") entry_author_name = SubElement(entry_author, "name") if author_name is None or len(author_name.strip()) == 0: entry_author_name.text = "Anonymous user" else: entry_author_name.text = author_name if not lon is None and not lat is None: entry_geo = SubElement(entry, "georss:point") entry_geo.text = "{lat} {lon}".format(lat=lat, lon=lon) entry_wrreport = SubElement(entry, "wr:report") entry_wrreport.attrib['report_id'] = str(report_id) entry_wrreport.attrib['page_id'] = str(page_id) entry_wrreport.attrib['page_title'] = page_title entry_wrreport.attrib['date_report'] = str(date_report) entry_wrreport.attrib['date_entry'] = str(date_entry) entry_wrreport.attrib['condition'] = "0" if condition is None else str(condition) entry_wrreport.attrib['author_name'] = author_name entry_wrreport.attrib['author_username'] = "" if author_userid is None else author_username entry_wrreport.text = description if last_updated is None: last_updated = datetime.datetime.now() feed_updated.text = last_updated.isoformat() + "+01:00" feed_xml = '\n' + tostring(feed) conn.close() return feed_xml class BerichteController(BaseController): def alle(self): """Handles URLs like http://127.0.0.1:5000/berichte/alle http://www.winterrodeln.org/feed/berichte/alle """ response.content_type = 'application/atom+xml' return create_feed() def bahn(self, id): """Handles URLs like http://127.0.0.1:5000/berichte/bahn/kemater_alm http://www.winterrodeln.org/feed/berichte/bahn/kemater_alm """ response.content_type = 'application/atom+xml' return create_feed(page_title=id) def bahnen(self, id): """Handles URLs like http://127.0.0.1:5000/berichte/bahnen/5+280+251 http://www.winterrodeln.org/feed/berichte/bahnen/5+280+251 """ page_ids = id.split('+') try: page_ids = [int(page_id) for page_id in page_ids] except ValueError: abort(400) # bad request response.content_type = 'application/atom+xml' return create_feed(page_ids=page_ids)