lamperi
/
pna


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428
							#!/usr/bin/env python3

# Copyright 2018 Toni Fadjukoff. All Rights Reserved.

# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# The project is based on original rna.fi
# project by Timo Siiranen before 2007-2010

import sys

day_names = [ "Maanantai", "Tiistai", "Keskiviikko", "Torstai", 
	       "Perjantai", "Lauantai", "Sunnuntai" ]

import amica
import sodexo
import juvenes
import campusravita

allergies = [ "M", "L", "VL", "G", "K", "Ve" ]
allergy_descriptions = {
        "M": "Maidoton",
        "L": "Laktoositon","VL": "Vähälaktoosinen",
        "G": "Gluteiiniton",
        "K": "Kasvis",
        "Ve": "Vegaani"
        }

import os
import time
import datetime
import re

global_prefix = "";
use_old = False; # 1 is good for testing, 0 for production system!
unordered = []

l = time.localtime()
this_week = datetime.datetime.now().isocalendar()[1]

updateException = None
for restaurant_module in [sodexo, amica, juvenes, campusravita]:
    try:
        unordered += restaurant_module.get_restaurants(use_old, this_week)
    except Exception as e:
        updateException = e

max_week = 0;
for r in unordered:
  week = r[2]
  max_week = week if week > max_week or week == 1 else max_week

if l[6] != 0 and this_week != max_week:
  # it's not sunday, don't force next week's menu yet
  max_week = this_week

stamp = time.time() - 3600*24*7
max_week_daterange = ""
if max_week >= 1 and max_week <= 52:
  # figure out the date range
  while True:
    stamp_week = int(time.strftime("%W", time.localtime(stamp)))
    if stamp_week == max_week:
            break
    stamp += 3600*24
  l1 = time.localtime(stamp)
  l2 = time.localtime(stamp + 3600*24*6)
  if l1[1] == l2[1]:
    # same month
    max_week_daterange = "{monday}-{sunday}.{month}.".format(monday=l1[2], sunday=l2[2], month=l1[1])
  else:
    # different months
    max_week_daterange = "{monday}.{month}.-{sunday}.{next_month}.".format(monday=l1[2], month=l1[1],
            sunday=l2[2], next_month=l2[1])
  max_week_daterange = " (" + str(max_week_daterange) + ")"

file_header = '''<?xml version="1.0" encoding="utf-8"?>
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="fi" lang="fi">
<head>
  <title>Ruokalistat</title>
  <meta charset="UTF-8"/>
  <link rel="stylesheet" type="text/css" href="{resources_prefix}ruoka.css" />
</head>
<body>

<div id="notice" style="border: 1px solid black; border-radius: 5px; padding: 5px;">
PNA.fi on kolmannen osapuolen tarjoama palvelu. En voi taata ruokalistojen oikeellisuutta.  Virallisen ruokalistan saat näkyviin siirtymällä ravintolan omille sivuille painamalla sen nimestä. Jos huomaat ruokalistassa virheen, nopeiten virhe saadaan pois näkyvistä kun lähetät minulle siitä sähköpostia: <a href="mailto:lamperi+pna@gmail.com">lamperi+pna@gmail.com</a>
</div>

<form method="get" action="/cgi-bin/food.cgi">
'''

file_footer = """
      <div class="footer">Päivitetty {stamp}
        <input type="submit\" value=\"Päivitä nyt\" />
          / Palaute <a href=\"mailto:lamperi+pna@gmail.com\">lamperi+pna@gmail.com</a>
          / <a href="{{resources_prefix}}code.html\">Koodit täältä</a>
         / <a href="{{resources_prefix}}pna.html\">Mikä on PNA?</a>
      </div>
    </form>
  </body>
</html>
""".format(stamp=time.strftime("%d.%m.%Y %H:%M:%S"))

def find_last_day_with_foods(restaurants):
    last_day = 0
    for r in restaurants:
        for day in range(7):
            if day in r[3]:
                last_day = day if day > last_day else last_day
    return last_day

def write_days_header(fout, day, last_day):
    fout.write( "  <span class=\"days\">")
    for i in range(last_day):
        if i == day:
            fout.write("{0} ".format(day_names[i]))
        else:
            fout.write("<a href=\"{0}.html\">{1}</a> ".format(i+1, day_names[i]))
  
    if day < 0:
        fout.write("Taulukko")
    else:
        fout.write("<a href=\"table.html\">Taulukko</a>")
    fout.write("</span>\n")

def write_prefix_header(fout, prefix, day, resources_prefix):
    day = "table" if day == 0 else day
    fout.write("<span class=\"location\">")
    if prefix == "":
        fout.write("Kaikki ")
    else:
        fout.write("<a href=\"{resources_prefix}{day}.html\">Kaikki</a> ".format(resources_prefix=resources_prefix, day=day))
    if prefix == "tay/":
        fout.write("TaY ")
    else:
        fout.write("<a href=\"{resources_prefix}tay/{day}.html\">TaY</a> ".format(resources_prefix=resources_prefix, day=day))
    if prefix == "tays/":
        fout.write("TAYS ")
    else:
        fout.write("<a href=\"{resources_prefix}tays/{day}.html\">TAYS</a> ".format(resources_prefix=resources_prefix, day=day))
    if prefix == "tty/":
        fout.write("TTY ")
    else:
        fout.write("<a href=\"{resources_prefix}tty/{day}.html\">TTY</a> ".format(resources_prefix=resources_prefix, day=day))
    fout.write("</span>\n")

def write_day(day, header, outfname, last_day, restaurants, prefix, resources_prefix):
    with open(outfname, "w", encoding="utf-8") as fout:

        fout.write(file_header.format(resources_prefix=resources_prefix))
        fout.write("<h1>{header}</h1>\n".format(header=header))
        # print weekday links
        fout.write("<div class=\"title\">\n")
        write_days_header(fout, day, last_day)
        fout.write(" <span class=\"allergy\">Näytä: ")
        for a in allergies:
            fout.write("<input type=\"checkbox\" name=\"allergy_{a}\" id=\"allergy_{a}\" onclick=\"highlight()\" />".format(a=a))
            fout.write("<span title=\"{allergy_description}\">{a}</span>".format(allergy_description=allergy_descriptions[a], a=a))
        fout.write("</span>\n")
        write_prefix_header(fout, prefix, day+1, resources_prefix);
        fout.write("</div>\n")

        # print foods
        foodnum = 0
        eatable_food_numbers = {}
        maybe_eatable_food_numbers = {}
        for a in allergies:
            eatable_food_numbers[a] = []
            maybe_eatable_food_numbers[a] = []
        css_class = "left"
        fout.write("<div class=\"foods\"><div class=\"{css_class}\">\n".format(css_class=css_class))
        for r in restaurants:
            title, open_hours, week, week_foods, info = r[:5]
            exception = r[5].replace("\n", "<br>") if len(r) > 5 and r[5] else "Ruokalistaa ei saatavilla."
            title2, url, lazy_allergies, info_class = info[0:4]
            if day in week_foods or day < 5:
                if info_class != css_class:
                    css_class = info_class
                    fout.write("</div><div class=\"{css_class}\">\n".format(css_class=css_class))
                url = url.replace("&", "&amp;")
                fout.write("<h2><a href=\"{url}\">{title}</a></h2>\n".format(url=url, title=title))
                if not day in week_foods:
                    fout.write("<p class=\"missing\">{exception}</p>".format(exception=exception))
                    continue
                if week != "" and week != max_week:
                    if week > max_week or (week == 1 and max_week == 52):
                        # early..
                        fout.write("<p class=\"nextweek\">Viikon {week} ruokalista:</p>".format(week=week))
                    else:
                        fout.write("<p class=\"missing\">Saatavilla vain viikon {week} ruokalista.</p>".format(week=week))
                        continue
                if len(week_foods[day]) == 0:
                    fout.write("<p class=\"missing\">Ei ruokatietoja päivälle.</p>")
                    continue
                fout.write("<ul class=\"food\">\n")
                for food in week_foods[day]:
                    output = []
                    total_allergies = {}
                    maybe_allergies = {}
                    for a in allergies:
                        total_allergies[a] = 0
                        maybe_allergies[a] = 0
                    part_count = 0
                    for part in food.split("\n"):
                        # who cares?
                        if re.match("Peruna|Riisi", part):
                            continue
	                # fries: well, maybe we do care, but we don't care about allergy stuff
                        # and keep it in the same line as the previous food so as not to
	                # waste visible space
                        (part, fries) = re.subn("Tikkuperunat|Ranskalaiset perunat", "", part)
                        fries = fries > 0
                        part_count += 1

                        # add missing () around allergies
                        part = re.sub(" (([MLGKA]|VL|Ve|VE|Veg|Hot)(, *([MLGKA]|VL|Ve|VE|Veg|Hot|))+)$", " (\\1)", part)
                        match = re.match("^(.*) \\(([^\\)]+)\\)$", part)
                        if match:
                            # fix allergy issues
                            food = match.group(1)
                            allergy = match.group(2)
                            # standardization
                            allergy = re.sub("Kasvis", "K", allergy)
                            allergy = re.sub("([MLGK]|VL)([MLGK]|[VL])", "\\1,\\2", allergy)
                            # spaces to commas
                            allergy = re.sub("saatavana[: ]+(.*)$", "eriks: \\1", allergy)
                            allergy = re.sub(" +", ",", allergy)
                            # remove double commas
                            allergy = re.sub(",+", ",", allergy)
                            # eriks: standardization
                            allergy = re.sub(",?eriks:,?", ", eriks: ", allergy)
                            # remove extra commas/spaces from beginning/end
                            allergy = re.sub("^[, ]+", "", allergy)
                            allergy = re.sub("[, ]+$", "", allergy)
                            part = "{food} ({allergy})".format(food=food, allergy=allergy)

                        if output and not fries:
                            output.append("<br />\n")

                        match = re.search("Saatavana myös: (.*)", part)
                        if match:
                            alt = match.group(1)
                            alt = re.sub(r"^\((.*)\)$", r"\1", alt)
                            alt = re.sub("[, ]+", r",", alt)
                            alt = re.sub("^,+", "", alt)
                            alt = re.sub(",+$", "", alt)
                            part = re.sub(r"\)[- ]*Saatavana myös:.*", "eriks: {alt})".format(alt=alt), part)
                            part = re.sub(r"[- ]*Saatavana myös:.*", "(eriks: {alt})".format(alt=alt), part)

                        match = re.match(r"^(.*)(\([^\)]+\))$", part)
                        if match:
                            text = match.group(1)
                            allergy = match.group(2)
                            if fries:
                                output.append(", {text}".format(text=text))
                            else:
                                output.append("{text} <span class=\"allergy\">{allergy}</span>".format(text=text, allergy=allergy))
                            allergy = re.sub(r"^\((.*)\)$", r"\1", allergy)
                            allergy = re.sub(" *eriks: ", "", allergy)
                            this_allergies = set()
                            for a in re.split("[, ]", allergy):
                                for al in allergies:
                                    if a == al:
                                        this_allergies.add(a)
                                        break
                            if "L" in this_allergies:
                                this_allergies.add("VL")
                            for a in this_allergies:
                                if a in total_allergies:
                                    total_allergies[a] += 1
                                if a in maybe_allergies:
                                    maybe_allergies[a] += 1
                            match = re.search("M", lazy_allergies)
                            if match:
                                if "L" in this_allergies and not "M" in this_allergies:
                                    maybe_allergies["M"] += 1
                        else:
                            if lazy_allergies == "all":
                                for a in allergies:
                                    maybe_allergies[a] += 1

                            output.append(part)
                    allergy_output = ""
                    for a in allergies:
                        if total_allergies[a] == part_count:
                            eatable_food_numbers[a].append(foodnum)
                        elif maybe_allergies[a]== part_count:
                            maybe_eatable_food_numbers[a].append(foodnum)
                    fout.write("  <li id=\"f{foodnum}\">{output}</li>\n".format(foodnum=foodnum, output="".join(output)))
                    foodnum += 1
                fout.write("</ul>\n")
        # write allergy scripts
        fout.write('<script type="text/javascript" src="{resources_prefix}ruoka.js"></script>'.format(resources_prefix=resources_prefix))
        fout.write('<script type="text/javascript">\n')
        fout.write("var eatable_foods = [];\n")
        fout.write("var maybe_eatable_foods = [];\n")
        for a in allergies:
            fout.write("eatable_foods[\"{a}\"] = [{eatable_food_number}];\n".format(a=a, eatable_food_number=",".join(str(e) for e in eatable_food_numbers[a])))
            fout.write("maybe_eatable_foods[\"{a}\"] = [{maybe_eatable_food_number}];\n".format(a=a, maybe_eatable_food_number=",".join(str(e) for e in maybe_eatable_food_numbers[a])))
        allergy_string = ",".join('"{a}"'.format(a=a) for a in allergies)
        fout.write("var allergies = [{allergies}];\n".format(allergies = allergy_string))
        fout.write("var food_count = {foodnum};\n".format(foodnum = foodnum))
        fout.write("window.onload = function() { set_allergies(); show_warning(); };\n")
        fout.write("</script>\n")
        fout.write("</div></div>{file_footer}".format(file_footer=file_footer.format(resources_prefix=resources_prefix)))

def write_all_days(restaurants, prefix, title, resources_prefix):
    try:
        os.mkdir(prefix)
    except OSError as err:
        pass # hope it already exists
    last_day = find_last_day_with_foods(restaurants);
    for day in range(7):
        outfname = "{prefix}{day}.html".format(prefix=prefix, day=day+1)
        if day > last_day:
            try:
                os.unlink(outfname)
            except OSError as err:
                pass # probably did not exist
            continue
        header = "{day_name} - {title} vko {max_week}{max_week_daterange}".format(day_name=day_names[day], title=title,
                max_week=max_week, max_week_daterange=max_week_daterange)
        write_day(day, header, outfname, last_day, restaurants, prefix, resources_prefix)
 
def write_table(restaurants, prefix, title, resources_prefix):
    last_day = find_last_day_with_foods(restaurants);
    outfname = "{prefix}table.html".format(prefix=prefix)
    with open(outfname, "w", encoding="utf-8") as fout:
        header = "{title} vko {max_week}{max_week_daterange}".format(title=title, max_week=max_week, max_week_daterange=max_week_daterange)
        fout.write(file_header.format(resources_prefix=resources_prefix))
        fout.write("<h1>{header}</h1>\n".format(header=header))
        fout.write("<div class=\"title\">\n")

        write_days_header(fout, -1, last_day)
        write_prefix_header(fout, prefix, 0, resources_prefix)
        fout.write("</div><table border=\"1\"><tr><th>Päivä</th>")
        for r in restaurants:
            (title, open_hours, week, week_foods, info) = r[:5]
            (title2, url) = info[0:2]
            url = re.sub("&", "&nbsp;", url)
            fout.write("<th><a href=\"{url}\">{title}</a></th>".format(url=url, title=title))
        fout.write("</tr>\n")
        for day in range(last_day):
            fout.write("<tr><td>{day_name}</td>\n".format(day_name=day_names[day]))
            for r in restaurants:
                (title, open_hours, week, week_foods, info) = r[:5]
                if day in week_foods and (week == "" or week == max_week):
                    fout.write("<td><ul>\n")
                    for food in week_foods[day]:
                        fout.write("<li>{food}</li>".format(food=food))
                    fout.write("</ul></td>\n")
                else:
                    fout.write("<td></td>\n")
            fout.write("</tr\n")
        fout.write("</table>{file_footer}".format(file_footer=file_footer.format(resources_prefix=resources_prefix)))

def get_restaurants_sorted(restaurants):
    # consider writing comparator
    out = []
    for r in restaurants:
        if r[4][3] == "left":
            out.append(r)
    for r in restaurants:
        if r[4][3] == "right":
            out.append(r)
    for r in restaurants:
        if r[4][3] == "middle" and not re.search("TAMK", r[4][0]):
            out.append(r)
    for r in restaurants:
        if r[4][3] == "middle" and re.search("TAMK", r[4][0]):
            out.append(r)
    return out

def get_restaurants_with_prefix(prefix, restaurants):
    out = []
    for r in restaurants:
        if re.search(prefix, r[4][0]):
            out.append(r)
    return get_restaurants_sorted(out)

tty_title = "TTY:n ruokalistat"
tty = get_restaurants_with_prefix("TTY", unordered)
write_all_days(tty, "tty/", tty_title, "../")
write_table(tty, "tty/", tty_title, "../")

tay_title = "Tampereen yliopiston ruokalistat"
tay = get_restaurants_with_prefix("TaY", unordered)
write_all_days(tay, "tay/", tay_title, "../")
write_table(tay, "tay/", tay_title, "../")

tays_title = "TAYS:n ruokalistat"
tays = get_restaurants_with_prefix("TAYS", unordered)
write_all_days(tays, "tays/", tays_title, "../")
write_table(tays, "tays/", tays_title, "../")

for r in unordered:
    if re.search(r"^\(TaY\)", r[0]):
        r[4][3] = "left"
    if re.search(r"^\(TTY\)", r[0]):
        r[4][3] = "right"
    if re.search(r"^\(TAYS\)", r[0]):
        r[4][3] = "middle"

all_title = "Tampereen yliopistojen ruokalistat";
all_restaurants = get_restaurants_sorted(unordered);
# move fusion kitchen last
#fusion = splice(@all_restaurants, 1, 1);
#splice(@all_restaurants, 4, 0, @fusion);

write_all_days(all_restaurants, "", all_title, "");
write_table(all_restaurants, "", all_title, "");

if updateException is not None:
    print("Exception happened while fetching menus")
    raise updateException