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
60 Commits
1 Branches
Tree: 4f81e1137e
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '4f81e1137e'
${ noResults }
pna/food.py

food.py 18KB
History Raw

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412 
  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.   <link rel="stylesheet" type="text/css" href="{resources_prefix}ruoka.css" />

  83. </head>

  84. <body>

  85. 

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

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

  88. </div>

  89. 

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

  91. '''

  92. 

  93. file_footer = """

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

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

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

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

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

  99.       </div>

  100.     </form>

  101.   </body>

  102. </html>

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

  104. 

  105. def find_last_day_with_foods(restaurants):

  106.     last_day = 0

  107.     for r in restaurants:

  108.         for day in range(7):

  109.             if day in r[3]:

  110.                 last_day = day if day > last_day else last_day

  111.     return last_day

  112. 

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

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

  115.     for i in range(last_day):

  116.         if i == day:

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

  118.         else:

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

  120.   

  121.     if day < 0:

  122.         fout.write("Taulukko")

  123.     else:

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

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

  126. 

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

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

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

  130.     if prefix == "":

  131.         fout.write("Kaikki ")

  132.     else:

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

  134.     if prefix == "tay/":

  135.         fout.write("TaY ")

  136.     else:

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

  138.     if prefix == "tays/":

  139.         fout.write("TAYS ")

  140.     else:

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

  142.     if prefix == "tty/":

  143.         fout.write("TTY ")

  144.     else:

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

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

  147. 

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

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

  150. 

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

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

  153.         # print weekday links

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

  155.         write_days_header(fout, day, last_day)

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

  157.         for a in allergies:

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

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

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

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

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

  163. 

  164.         # print foods

  165.         foodnum = 0

  166.         eatable_food_numbers = {}

  167.         maybe_eatable_food_numbers = {}

  168.         for a in allergies:

  169.             eatable_food_numbers[a] = []

  170.             maybe_eatable_food_numbers[a] = []

  171.         css_class = "left"

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

  173.         for r in restaurants:

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

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

  176.             if day in week_foods or day < 5:

  177.                 if info_class != css_class:

  178.                     css_class = info_class

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

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

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

  182.                 if not day in week_foods:

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

  184.                     continue

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

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

  187.                         # early..

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

  189.                     else:

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

  191.                         continue

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

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

  194.                     continue

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

  196.                 for food in week_foods[day]:

  197.                     output = []

  198.                     total_allergies = {}

  199.                     maybe_allergies = {}

  200.                     for a in allergies:

  201.                         total_allergies[a] = 0

  202.                         maybe_allergies[a] = 0

  203.                     part_count = 0

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

  205.                         # who cares?

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

  207.                             continue

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

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

  210. 	                # waste visible space

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

  212.                         fries = fries > 0

  213.                         part_count += 1

  214. 

  215.                         # add missing () around allergies

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

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

  218.                         if match:

  219.                             # fix allergy issues

  220.                             food = match.group(1)

  221.                             allergy = match.group(2)

  222.                             # standardization

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

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

  225.                             # spaces to commas

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

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

  228.                             # remove double commas

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

  230.                             # eriks: standardization

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

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

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

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

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

  236. 

  237.                         if output and not fries:

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

  239. 

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

  241.                         if match:

  242.                             alt = match.group(1)

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

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

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

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

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

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

  249. 

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

  251.                         if match:

  252.                             text = match.group(1)

  253.                             allergy = match.group(2)

  254.                             if fries:

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

  256.                             else:

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

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

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

  260.                             this_allergies = set()

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

  262.                                 for al in allergies:

  263.                                     if a == al:

  264.                                         this_allergies.add(a)

  265.                                         break

  266.                             if "L" in this_allergies:

  267.                                 this_allergies.add("VL")

  268.                             for a in this_allergies:

  269.                                 if a in total_allergies:

  270.                                     total_allergies[a] += 1

  271.                                 if a in maybe_allergies:

  272.                                     maybe_allergies[a] += 1

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

  274.                             if match:

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

  276.                                     maybe_allergies["M"] += 1

  277.                         else:

  278.                             if lazy_allergies == "all":

  279.                                 for a in allergies:

  280.                                     maybe_allergies[a] += 1

  281. 

  282.                             output.append(part)

  283.                     allergy_output = ""

  284.                     for a in allergies:

  285.                         if total_allergies[a] == part_count:

  286.                             eatable_food_numbers[a].append(foodnum)

  287.                         elif maybe_allergies[a]== part_count:

  288.                             maybe_eatable_food_numbers[a].append(foodnum)

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

  290.                     foodnum += 1

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

  292.         # write allergy scripts

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

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

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

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

  297.         for a in allergies:

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

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

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

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

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

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

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

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

  306. 

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

  308.     try:

  309.         os.mkdir(prefix)

  310.     except OSError as err:

  311.         pass # hope it already exists

  312.     last_day = find_last_day_with_foods(restaurants);

  313.     for day in range(7):

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

  315.         if day > last_day:

  316.             try:

  317.                 os.unlink(outfname)

  318.             except OSError as err:

  319.                 pass # probably did not exist

  320.             continue

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

  322.                 max_week=max_week, max_week_daterange=max_week_daterange)

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

  324.  

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

  326.     last_day = find_last_day_with_foods(restaurants);

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

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

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

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

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

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

  333. 

  334.         write_days_header(fout, -1, last_day)

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

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

  337.         for r in restaurants:

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

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

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

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

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

  343.         for day in range(last_day):

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

  345.             for r in restaurants:

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

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

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

  349.                     for food in week_foods[day]:

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

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

  352.                 else:

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

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

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

  356. 

  357. def get_restaurants_sorted(restaurants):

  358.     # consider writing comparator

  359.     out = []

  360.     for r in restaurants:

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

  362.             out.append(r)

  363.     for r in restaurants:

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

  365.             out.append(r)

  366.     for r in restaurants:

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

  368.             out.append(r)

  369.     for r in restaurants:

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

  371.             out.append(r)

  372.     return out

  373. 

  374. def get_restaurants_with_prefix(prefix, restaurants):

  375.     out = []

  376.     for r in restaurants:

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

  378.             out.append(r)

  379.     return get_restaurants_sorted(out)

  380. 

  381. tty_title = "TTY:n ruokalistat"

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

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

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

  385. 

  386. tay_title = "Tampereen yliopiston ruokalistat"

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

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

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

  390. 

  391. tays_title = "TAYS:n ruokalistat"

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

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

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

  395. 

  396. for r in unordered:

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

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

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

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

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

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

  403. 

  404. all_title = "Tampereen yliopistojen ruokalistat";

  405. all_restaurants = get_restaurants_sorted(unordered);

  406. # move fusion kitchen last

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

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

  409. 

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

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