forked from devmanorg/wine
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.py
66 lines (50 loc) · 1.95 KB
/
main.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
from http.server import HTTPServer, SimpleHTTPRequestHandler
from jinja2 import Environment, FileSystemLoader, select_autoescape
import datetime
import pandas
import collections
import os
from dotenv import load_dotenv
def get_str_years(years):
if len(str(years)) == 2 and (years // 10) == 1:
return 'лет'
if len(str(years)) > 1:
return get_str_years(years % (10 ** (len(str(years)) - 1)))
if years == 1:
return 'год'
elif years in (2, 3, 4):
return 'года'
else:
return 'лет'
def main():
load_dotenv()
foundation_year = int(os.environ['FOUNDATION_YEAR'])
excel_filepath = os.environ['EXCEL_FILE']
now_year = datetime.datetime.now().year
existence_years = now_year - foundation_year
ru_years = get_str_years(existence_years)
env = Environment(
loader=FileSystemLoader('.'),
autoescape=select_autoescape(['html', 'xml'])
)
template = env.get_template('template.html')
excel_wines = pandas.read_excel(excel_filepath, na_values=' ', keep_default_na=False).to_dict(orient='record')
wines = collections.defaultdict(list)
for wine in excel_wines:
wines[wine['Категория']].append({'Название': wine['Название'],
'Сорт': wine['Сорт'],
'Цена': wine['Цена'],
'Картинка': wine['Картинка'],
'Акция': wine['Акция'],
})
rendered_page = template.render(
existence_years=existence_years,
ru_years=ru_years,
wines=wines,
)
with open('index.html', 'w', encoding="utf8") as file:
file.write(rendered_page)
server = HTTPServer(('0.0.0.0', 8000), SimpleHTTPRequestHandler)
server.serve_forever()
if __name__ == "__main__":
main()