forked from chinese-soup/zradlo
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathmichelska.py
executable file
·81 lines (66 loc) · 2.1 KB
/
michelska.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#!/usr/bin/env python3
# coding=utf-8
import requests, sys, re
from datetime import datetime
from bs4 import BeautifulSoup
def get_url():
return "http://www.michelska.cz/dnes.htm"
def get_name():
return "Michelská"
def get_file():
user_agent = {'User-agent': 'Mozilla/5.0'}
kantyna = requests.get(get_url(), headers = user_agent)
kantyna.encoding = 'Windows-1250'
return kantyna
def prepare_bs(kantyna):
if kantyna is not None and kantyna.status_code == 200:
html = kantyna.text
soup = BeautifulSoup(html, 'html.parser')
return soup
else:
return "Error"
def return_menu(soup):
date = soup.find_all("p", {"class": "MsoNormal"})[1].text.strip().strip('Nabídka na')
a = soup.find("table").find_all("tr")
previous = ''
items = []
for item in a:
# print(item)
# print(previous)
grid = item.find_all("td")
jidlo = grid[1].text.strip()
# print(grid[2].text)
predcena = re.match("([0-9]{2,3}[\s]*,-).*", grid[2].text.strip())
if predcena:
cena = predcena.group(1).strip()
if jidlo and cena:
if previous == 'Salát':
jidlo = jidlo + ' salát'
arr = [jidlo, cena]
items.append(arr)
else:
continue
else:
previous = item.text.replace(':', '').strip()
continue
return(items, date)
def result():
try:
file = get_file()
bs = prepare_bs(file)
nazev = get_name()
url = get_url()
lokalita = "brumlovka"
menu_list, date = return_menu(bs)
return (nazev, url, date, menu_list, lokalita)
except Exception as e:
print(e)
nazev = get_name()
url = get_url()
lokalita = "brumlovka"
return (nazev, url, "Menu nenalezeno", [], lokalita)
if __name__ == "__main__":
file = get_file()
bs = prepare_bs(file)
menu_list, date = return_menu(bs)
print (date, menu_list)