aachen.py

from urllib.request import urlopen
from bs4 import BeautifulSoup as parse
from bs4.element import NavigableString, Tag

from utils import Parser

from pyopenmensa.feed import OpenMensaCanteen, buildLegend

legend = None


def add_meals_from_table(canteen, table, day):
    for item in table.find_all('tr'):
        # category
        category = item.find('span', attrs={'class': 'menue-category'}).text.strip()
        # split names and notes
        name = ''
        notes = set()
        descs = item.find('span', attrs={'class': 'expand-nutr'})
        if not descs:
            return
        for namePart in descs.children:
            if type(namePart) is NavigableString:
                name += namePart.string
            elif type(namePart) is Tag:
                if namePart.name == 'sup':
                    notes.update(namePart.text.strip().split(','))
                else:
                    name += namePart.text
        name = name.strip()
        # Dieses Plus Zeichen entfernen
        name = name[1:] 
        notes = [legend.get(n, n) for n in notes if n]

        nutrition = item.find('div', attrs={'class': 'nutr-info'})

        first = True
        for br in nutrition.findAll('br'):
            prev_s = br.prevSibling
            next_s = br.nextSibling        
            if first:
                first = False
                if prev_s:
                    notes.append(prev_s.strip())
                if next_s:
                    notes.append(next_s.strip())
            else:
                if next_s:
                    notes.append(next_s.strip())

        price_tag = item.find('span', attrs={'class': 'menue-price'})
        if not price_tag:
            canteen.addMeal(day, category, name, notes)
        else:
            canteen.addMeal(day, category, name, notes, price_tag.text.strip())


def parse_day(canteen, day, data):
    # 1. menues
    note = data.find(id='note')
    if note:
        canteen.setDayClosed(day)
        return
    add_meals_from_table(canteen, data.find(attrs={'class': 'menues'}), day)

    # 2. extras:
    extras_table = data.find(attrs={'class': 'extras'})

    if not extras_table:
        return
    add_meals_from_table(canteen, extras_table, day)


def parse_url(url, today=False):
    canteen = OpenMensaCanteen()
    # todo only for: Tellergericht, vegetarisch, Klassiker, Empfehlung des Tages:
    canteen.setAdditionalCharges('student', {'other': 1.5})

    document = parse(urlopen(url).read(), 'lxml')

    global legend
    regex = '\((?P<name>[\dA-Z]+)\)\s*(?P<value>[\w\s]+)'
    legend = buildLegend(legend, document.find(id='additives').text, regex=regex)

    days = ('Montag', 'Dienstag', 'Mittwoch', 'Donnerstag', 'Freitag',
            'Montag', 'Dienstag', 'Mittwoch', 'Donnerstag', 'Freitag')
    for day in days:
        data = document.find('div', id=day)
        headline = document.find('a', attrs={'data-anchor':day.lower()})
        parse_day(canteen, headline.text, data)
    return canteen.toXMLFeed()


parser = Parser('aachen', handler=parse_url,
                shared_prefix='http://www.studentenwerk-aachen.de/speiseplaene/')
parser.define('academica', suffix='academica-w.html')
parser.define('ahorn', suffix='ahornstrasse-w.html')
parser.define('templergraben', suffix='templergraben-w.html')
parser.define('bayernallee', suffix='bayernallee-w.html')
parser.define('eups', suffix='eupenerstrasse-w.html')
parser.define('goethe', suffix='goethestrasse-w.html')
parser.define('vita', suffix='vita-w.html')
parser.define('zeltmensa', suffix='forum-w.html')
parser.define('juelich', suffix='juelich-w.html')