Home Explore Help
Sign In
lamperi
/
pna
Watch 1
Star 0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
PNA.fi koodi
62 Commits
1 Branches
Tree: 751528f891
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '751528f891'
${ noResults }
pna/food.py

food.py 18KB
History Raw

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413 
  1. #!/usr/bin/env python3

  2. 

  3. # Ruokalistaparseri

  4. # Copyright (c) 2016 Toni Fadjukoff

  5. 

  6. # Based on food.pl by

  7. # Copyright (c) 2007-2010 Timo Sirainen

  8. #               2011-2016 Toni Fadjukoff

  9. # This is Public Domain

  10. 

  11. import sys

  12. 

  13. day_names = [ "Maanantai", "Tiistai", "Keskiviikko", "Torstai", 

  14. 	       "Perjantai", "Lauantai", "Sunnuntai" ]

  15. 

  16. import amica

  17. import sodexo

  18. import juvenes

  19. import campusravita

  20. 

  21. allergies = [ "M", "L", "VL", "G", "K", "Ve" ]

  22. allergy_descriptions = {

  23.         "M": "Maidoton",

  24.         "L": "Laktoositon","VL": "Vähälaktoosinen",

  25.         "G": "Gluteiiniton",

  26.         "K": "Kasvis",

  27.         "Ve": "Vegaani"

  28.         }

  29. 

  30. import os

  31. import time

  32. import datetime

  33. import re

  34. 

  35. global_prefix = "";

  36. use_old = False; # 1 is good for testing, 0 for production system!

  37. unordered = []

  38. 

  39. l = time.localtime()

  40. this_week = datetime.datetime.now().isocalendar()[1]

  41. 

  42. unordered += amica.get_restaurants(use_old, this_week)

  43. unordered += juvenes.get_restaurants(use_old, this_week)

  44. unordered += sodexo.get_restaurants(use_old, this_week)

  45. unordered += campusravita.get_restaurants(use_old, this_week)

  46. 

  47. 

  48. max_week = 0;

  49. for r in unordered:

  50.   week = r[2]

  51.   max_week = week if week > max_week or week == 1 else max_week

  52. 

  53. if l[6] != 0 and this_week != max_week:

  54.   # it's not sunday, don't force next week's menu yet

  55.   max_week = this_week

  56. 

  57. stamp = time.time() - 3600*24*7

  58. max_week_daterange = ""

  59. if max_week >= 1 and max_week <= 52:

  60.   # figure out the date range

  61.   while True:

  62.     stamp_week = int(time.strftime("%W", time.localtime(stamp)))

  63.     if stamp_week == max_week:

  64.             break

  65.     stamp += 3600*24

  66.   l1 = time.localtime(stamp)

  67.   l2 = time.localtime(stamp + 3600*24*6)

  68.   if l1[1] == l2[1]:

  69.     # same month

  70.     max_week_daterange = "{monday}-{sunday}.{month}.".format(monday=l1[2], sunday=l2[2], month=l1[1])

  71.   else:

  72.     # different months

  73.     max_week_daterange = "{monday}.{month}.-{sunday}.{next_month}.".format(monday=l1[2], month=l1[1],

  74.             sunday=l2[2], next_month=l2[1])

  75.   max_week_daterange = " (" + str(max_week_daterange) + ")"

  76. 

  77. file_header = '''<?xml version="1.0" encoding="utf-8"?>

  78. <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">

  79. <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="fi" lang="fi">

  80. <head>

  81.   <title>Ruokalistat</title>

  82.   <meta charset="UTF-8"/>

  83.   <link rel="stylesheet" type="text/css" href="{resources_prefix}ruoka.css" />

  84. </head>

  85. <body>

  86. 

  87. <div id="notice" style="border: 1px solid black; border-radius: 5px; padding: 5px;">

  88. 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>

  89. </div>

  90. 

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

  92. '''

  93. 

  94. file_footer = """

  95.       <div class="footer">Päivitetty {stamp}

  96.         <input type="submit\" value=\"Päivitä nyt\" />

  97.           / Palaute <a href=\"mailto:lamperi+pna@gmail.com\">lamperi+pna@gmail.com</a>

  98.           / <a href="{{resources_prefix}}code.html\">Koodit täältä</a>

  99.          / <a href="{{resources_prefix}}pna.html\">Mikä on PNA?</a>

  100.       </div>

  101.     </form>

  102.   </body>

  103. </html>

  104. """.format(stamp=time.strftime("%d.%m.%Y %H:%M:%S"))

  105. 

  106. def find_last_day_with_foods(restaurants):

  107.     last_day = 0

  108.     for r in restaurants:

  109.         for day in range(7):

  110.             if day in r[3]:

  111.                 last_day = day if day > last_day else last_day

  112.     return last_day

  113. 

  114. def write_days_header(fout, day, last_day):

  115.     fout.write( "  <span class=\"days\">")

  116.     for i in range(last_day):

  117.         if i == day:

  118.             fout.write("{0} ".format(day_names[i]))

  119.         else:

  120.             fout.write("<a href=\"{0}.html\">{1}</a> ".format(i+1, day_names[i]))

  121.   

  122.     if day < 0:

  123.         fout.write("Taulukko")

  124.     else:

  125.         fout.write("<a href=\"table.html\">Taulukko</a>")

  126.     fout.write("</span>\n")

  127. 

  128. def write_prefix_header(fout, prefix, day, resources_prefix):

  129.     day = "table" if day == 0 else day

  130.     fout.write("<span class=\"location\">")

  131.     if prefix == "":

  132.         fout.write("Kaikki ")

  133.     else:

  134.         fout.write("<a href=\"{resources_prefix}{day}.html\">Kaikki</a> ".format(resources_prefix=resources_prefix, day=day))

  135.     if prefix == "tay/":

  136.         fout.write("TaY ")

  137.     else:

  138.         fout.write("<a href=\"{resources_prefix}tay/{day}.html\">TaY</a> ".format(resources_prefix=resources_prefix, day=day))

  139.     if prefix == "tays/":

  140.         fout.write("TAYS ")

  141.     else:

  142.         fout.write("<a href=\"{resources_prefix}tays/{day}.html\">TAYS</a> ".format(resources_prefix=resources_prefix, day=day))

  143.     if prefix == "tty/":

  144.         fout.write("TTY ")

  145.     else:

  146.         fout.write("<a href=\"{resources_prefix}tty/{day}.html\">TTY</a> ".format(resources_prefix=resources_prefix, day=day))

  147.     fout.write("</span>\n")

  148. 

  149. def write_day(day, header, outfname, last_day, restaurants, prefix, resources_prefix):

  150.     with open(outfname, "w", encoding="utf-8") as fout:

  151. 

  152.         fout.write(file_header.format(resources_prefix=resources_prefix))

  153.         fout.write("<h1>{header}</h1>\n".format(header=header))

  154.         # print weekday links

  155.         fout.write("<div class=\"title\">\n")

  156.         write_days_header(fout, day, last_day)

  157.         fout.write(" <span class=\"allergy\">Näytä: ")

  158.         for a in allergies:

  159.             fout.write("<input type=\"checkbox\" name=\"allergy_{a}\" id=\"allergy_{a}\" onclick=\"highlight()\" />".format(a=a))

  160.             fout.write("<span title=\"{allergy_description}\">{a}</span>".format(allergy_description=allergy_descriptions[a], a=a))

  161.         fout.write("</span>\n")

  162.         write_prefix_header(fout, prefix, day+1, resources_prefix);

  163.         fout.write("</div>\n")

  164. 

  165.         # print foods

  166.         foodnum = 0

  167.         eatable_food_numbers = {}

  168.         maybe_eatable_food_numbers = {}

  169.         for a in allergies:

  170.             eatable_food_numbers[a] = []

  171.             maybe_eatable_food_numbers[a] = []

  172.         css_class = "left"

  173.         fout.write("<div class=\"foods\"><div class=\"{css_class}\">\n".format(css_class=css_class))

  174.         for r in restaurants:

  175.             title, open_hours, week, week_foods, info = r

  176.             title2, url, lazy_allergies, info_class = info[0:4]

  177.             if day in week_foods or day < 5:

  178.                 if info_class != css_class:

  179.                     css_class = info_class

  180.                     fout.write("</div><div class=\"{css_class}\">\n".format(css_class=css_class))

  181.                 url = url.replace("&", "&amp;")

  182.                 fout.write("<h2><a href=\"{url}\">{title}</a></h2>\n".format(url=url, title=title))

  183.                 if not day in week_foods:

  184.                     fout.write("<p class=\"missing\">Ruokalistaa ei saatavilla.</p>")

  185.                     continue

  186.                 if week != "" and week != max_week:

  187.                     if week > max_week or (week == 1 and max_week == 52):

  188.                         # early..

  189.                         fout.write("<p class=\"nextweek\">Viikon {week} ruokalista:</p>".format(week=week))

  190.                     else:

  191.                         fout.write("<p class=\"missing\">Saatavilla vain viikon {week} ruokalista.</p>".format(week=week))

  192.                         continue

  193.                 if len(week_foods[day]) == 0:

  194.                     fout.write("<p class=\"missing\">Ei ruokatietoja päivälle.</p>")

  195.                     continue

  196.                 fout.write("<ul class=\"food\">\n")

  197.                 for food in week_foods[day]:

  198.                     output = []

  199.                     total_allergies = {}

  200.                     maybe_allergies = {}

  201.                     for a in allergies:

  202.                         total_allergies[a] = 0

  203.                         maybe_allergies[a] = 0

  204.                     part_count = 0

  205.                     for part in food.split("\n"):

  206.                         # who cares?

  207.                         if re.match("Peruna|Riisi", part):

  208.                             continue

  209. 	                # fries: well, maybe we do care, but we don't care about allergy stuff

  210.                         # and keep it in the same line as the previous food so as not to

  211. 	                # waste visible space

  212.                         (part, fries) = re.subn("Tikkuperunat|Ranskalaiset perunat", "", part)

  213.                         fries = fries > 0

  214.                         part_count += 1

  215. 

  216.                         # add missing () around allergies

  217.                         part = re.sub(" (([MLGKA]|VL|Ve|VE|Veg|Hot)(, *([MLGKA]|VL|Ve|VE|Veg|Hot|))+)$", " (\\1)", part)

  218.                         match = re.match("^(.*) \\(([^\\)]+)\\)$", part)

  219.                         if match:

  220.                             # fix allergy issues

  221.                             food = match.group(1)

  222.                             allergy = match.group(2)

  223.                             # standardization

  224.                             allergy = re.sub("Kasvis", "K", allergy)

  225.                             allergy = re.sub("([MLGK]|VL)([MLGK]|[VL])", "\\1,\\2", allergy)

  226.                             # spaces to commas

  227.                             allergy = re.sub("saatavana[: ]+(.*)$", "eriks: \\1", allergy)

  228.                             allergy = re.sub(" +", ",", allergy)

  229.                             # remove double commas

  230.                             allergy = re.sub(",+", ",", allergy)

  231.                             # eriks: standardization

  232.                             allergy = re.sub(",?eriks:,?", ", eriks: ", allergy)

  233.                             # remove extra commas/spaces from beginning/end

  234.                             allergy = re.sub("^[, ]+", "", allergy)

  235.                             allergy = re.sub("[, ]+$", "", allergy)

  236.                             part = "{food} ({allergy})".format(food=food, allergy=allergy)

  237. 

  238.                         if output and not fries:

  239.                             output.append("<br />\n")

  240. 

  241.                         match = re.search("Saatavana myös: (.*)", part)

  242.                         if match:

  243.                             alt = match.group(1)

  244.                             alt = re.sub(r"^\((.*)\)$", r"\1", alt)

  245.                             alt = re.sub("[, ]+", r",", alt)

  246.                             alt = re.sub("^,+", "", alt)

  247.                             alt = re.sub(",+$", "", alt)

  248.                             part = re.sub(r"\)[- ]*Saatavana myös:.*", "eriks: {alt})".format(alt=alt), part)

  249.                             part = re.sub(r"[- ]*Saatavana myös:.*", "(eriks: {alt})".format(alt=alt), part)

  250. 

  251.                         match = re.match(r"^(.*)(\([^\)]+\))$", part)

  252.                         if match:

  253.                             text = match.group(1)

  254.                             allergy = match.group(2)

  255.                             if fries:

  256.                                 output.append(", {text}".format(text=text))

  257.                             else:

  258.                                 output.append("{text} <span class=\"allergy\">{allergy}</span>".format(text=text, allergy=allergy))

  259.                             allergy = re.sub(r"^\((.*)\)$", r"\1", allergy)

  260.                             allergy = re.sub(" *eriks: ", "", allergy)

  261.                             this_allergies = set()

  262.                             for a in re.split("[, ]", allergy):

  263.                                 for al in allergies:

  264.                                     if a == al:

  265.                                         this_allergies.add(a)

  266.                                         break

  267.                             if "L" in this_allergies:

  268.                                 this_allergies.add("VL")

  269.                             for a in this_allergies:

  270.                                 if a in total_allergies:

  271.                                     total_allergies[a] += 1

  272.                                 if a in maybe_allergies:

  273.                                     maybe_allergies[a] += 1

  274.                             match = re.search("M", lazy_allergies)

  275.                             if match:

  276.                                 if "L" in this_allergies and not "M" in this_allergies:

  277.                                     maybe_allergies["M"] += 1

  278.                         else:

  279.                             if lazy_allergies == "all":

  280.                                 for a in allergies:

  281.                                     maybe_allergies[a] += 1

  282. 

  283.                             output.append(part)

  284.                     allergy_output = ""

  285.                     for a in allergies:

  286.                         if total_allergies[a] == part_count:

  287.                             eatable_food_numbers[a].append(foodnum)

  288.                         elif maybe_allergies[a]== part_count:

  289.                             maybe_eatable_food_numbers[a].append(foodnum)

  290.                     fout.write("  <li id=\"f{foodnum}\">{output}</li>\n".format(foodnum=foodnum, output="".join(output)))

  291.                     foodnum += 1

  292.                 fout.write("</ul>\n")

  293.         # write allergy scripts

  294.         fout.write('<script type="text/javascript" src="{resources_prefix}ruoka.js"></script>'.format(resources_prefix=resources_prefix))

  295.         fout.write('<script type="text/javascript">\n')

  296.         fout.write("var eatable_foods = [];\n")

  297.         fout.write("var maybe_eatable_foods = [];\n")

  298.         for a in allergies:

  299.             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])))

  300.             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])))

  301.         allergy_string = ",".join('"{a}"'.format(a=a) for a in allergies)

  302.         fout.write("var allergies = [{allergies}];\n".format(allergies = allergy_string))

  303.         fout.write("var food_count = {foodnum};\n".format(foodnum = foodnum))

  304.         fout.write("window.onload = function() { set_allergies(); show_warning(); };\n")

  305.         fout.write("</script>\n")

  306.         fout.write("</div></div>{file_footer}".format(file_footer=file_footer.format(resources_prefix=resources_prefix)))

  307. 

  308. def write_all_days(restaurants, prefix, title, resources_prefix):

  309.     try:

  310.         os.mkdir(prefix)

  311.     except OSError as err:

  312.         pass # hope it already exists

  313.     last_day = find_last_day_with_foods(restaurants);

  314.     for day in range(7):

  315.         outfname = "{prefix}{day}.html".format(prefix=prefix, day=day+1)

  316.         if day > last_day:

  317.             try:

  318.                 os.unlink(outfname)

  319.             except OSError as err:

  320.                 pass # probably did not exist

  321.             continue

  322.         header = "{day_name} - {title} vko {max_week}{max_week_daterange}".format(day_name=day_names[day], title=title,

  323.                 max_week=max_week, max_week_daterange=max_week_daterange)

  324.         write_day(day, header, outfname, last_day, restaurants, prefix, resources_prefix)

  325.  

  326. def write_table(restaurants, prefix, title, resources_prefix):

  327.     last_day = find_last_day_with_foods(restaurants);

  328.     outfname = "{prefix}table.html".format(prefix=prefix)

  329.     with open(outfname, "w", encoding="utf-8") as fout:

  330.         header = "{title} vko {max_week}{max_week_daterange}".format(title=title, max_week=max_week, max_week_daterange=max_week_daterange)

  331.         fout.write(file_header.format(resources_prefix=resources_prefix))

  332.         fout.write("<h1>{header}</h1>\n".format(header=header))

  333.         fout.write("<div class=\"title\">\n")

  334. 

  335.         write_days_header(fout, -1, last_day)

  336.         write_prefix_header(fout, prefix, 0, resources_prefix)

  337.         fout.write("</div><table border=\"1\"><tr><th>Päivä</th>")

  338.         for r in restaurants:

  339.             (title, open_hours, week, week_foods, info) = r

  340.             (title2, url) = info[0:2]

  341.             url = re.sub("&", "&nbsp;", url)

  342.             fout.write("<th><a href=\"{url}\">{title}</a></th>".format(url=url, title=title))

  343.         fout.write("</tr>\n")

  344.         for day in range(last_day):

  345.             fout.write("<tr><td>{day_name}</td>\n".format(day_name=day_names[day]))

  346.             for r in restaurants:

  347.                 (title, open_hours, week, week_foods, info) = r

  348.                 if day in week_foods and (week == "" or week == max_week):

  349.                     fout.write("<td><ul>\n")

  350.                     for food in week_foods[day]:

  351.                         fout.write("<li>{food}</li>".format(food=food))

  352.                     fout.write("</ul></td>\n")

  353.                 else:

  354.                     fout.write("<td></td>\n")

  355.             fout.write("</tr\n")

  356.         fout.write("</table>{file_footer}".format(file_footer=file_footer.format(resources_prefix=resources_prefix)))

  357. 

  358. def get_restaurants_sorted(restaurants):

  359.     # consider writing comparator

  360.     out = []

  361.     for r in restaurants:

  362.         if r[4][3] == "left":

  363.             out.append(r)

  364.     for r in restaurants:

  365.         if r[4][3] == "right":

  366.             out.append(r)

  367.     for r in restaurants:

  368.         if r[4][3] == "middle" and not re.search("TAMK", r[4][0]):

  369.             out.append(r)

  370.     for r in restaurants:

  371.         if r[4][3] == "middle" and re.search("TAMK", r[4][0]):

  372.             out.append(r)

  373.     return out

  374. 

  375. def get_restaurants_with_prefix(prefix, restaurants):

  376.     out = []

  377.     for r in restaurants:

  378.         if re.search(prefix, r[4][0]):

  379.             out.append(r)

  380.     return get_restaurants_sorted(out)

  381. 

  382. tty_title = "TTY:n ruokalistat"

  383. tty = get_restaurants_with_prefix("TTY", unordered)

  384. write_all_days(tty, "tty/", tty_title, "../")

  385. write_table(tty, "tty/", tty_title, "../")

  386. 

  387. tay_title = "Tampereen yliopiston ruokalistat"

  388. tay = get_restaurants_with_prefix("TaY", unordered)

  389. write_all_days(tay, "tay/", tay_title, "../")

  390. write_table(tay, "tay/", tay_title, "../")

  391. 

  392. tays_title = "TAYS:n ruokalistat"

  393. tays = get_restaurants_with_prefix("TAYS", unordered)

  394. write_all_days(tays, "tays/", tays_title, "../")

  395. write_table(tays, "tays/", tays_title, "../")

  396. 

  397. for r in unordered:

  398.     if re.search(r"^\(TaY\)", r[0]):

  399.         r[4][3] = "left"

  400.     if re.search(r"^\(TTY\)", r[0]):

  401.         r[4][3] = "right"

  402.     if re.search(r"^\(TAYS\)", r[0]):

  403.         r[4][3] = "middle"

  404. 

  405. all_title = "Tampereen yliopistojen ruokalistat";

  406. all_restaurants = get_restaurants_sorted(unordered);

  407. # move fusion kitchen last

  408. #fusion = splice(@all_restaurants, 1, 1);

  409. #splice(@all_restaurants, 4, 0, @fusion);

  410. 

  411. write_all_days(all_restaurants, "", all_title, "");

  412. write_table(all_restaurants, "", all_title, "");