forked from devmanorg/wine
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.py
62 lines (50 loc) · 2.02 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
import os
from http.server import HTTPServer, SimpleHTTPRequestHandler
import collections
import datetime
from dotenv import load_dotenv
from jinja2 import Environment, FileSystemLoader, select_autoescape
import pandas
import argparse
def decline_years(n):
if 11 <= n % 100 <= 19:
return 'лет'
else:
remainde = n % 10
if remainde == 1:
return 'год'
elif 2 <= remainde <= 4:
return 'года'
else:
return 'лет'
if __name__ == '__main__':
load_dotenv()
excel_path = os.environ['EXCEL_PATH']
env = Environment(
loader=FileSystemLoader('.'),
autoescape=select_autoescape(['html', 'xml'])
)
template = env.get_template('template.html')
now_date = datetime.date.today()
start_work = 1920
years_work = now_date.year - start_work
parser = argparse.ArgumentParser(description='Если ваш файл находится не в папке с кодом, то вы можете ввести путь к нему дополнительным аргументом')
parser.add_argument('--path', default=excel_path, help='Путь')
args = parser.parse_args()
wines_excel = pandas.read_excel(args.path,
sheet_name='Лист1',
na_values=['N/A', 'NA'],
keep_default_na=False).to_dict(orient='records')
wine_collection = collections.defaultdict(list)
for wine in wines_excel:
wine_collection[wine['Категория']].append(wine)
rendered_page = template.render(
current_year=years_work,
word_year=decline_years(years_work),
wines=wine_collection,
collection_names=wine_collection.keys()
)
with open('index.html', 'w', encoding="utf8") as file:
file.write(rendered_page)
server = HTTPServer(('0.0.0.0', 8000), SimpleHTTPRequestHandler)
server.serve_forever()