/
suggestible.py
159 lines (120 loc) · 4.99 KB
/
suggestible.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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
import sys
from PyQt5.QtCore import *
from PyQt5.QtWidgets import (QApplication, QWidget, QPushButton, QDesktopWidget,
QSlider, QLabel, QDateTimeEdit, QVBoxLayout, QGridLayout)
from PyQt5.QtGui import QIcon
import jinja2
from datetime import timedelta
from dateutil.parser import parse
import webbrowser
from Main import main
import tablegui
class Main(QWidget):
def __init__(self):
super().__init__()
self.initUI()
def initUI(self):
btn_gen = QPushButton('Generate', self)
btn_gen.setToolTip("Generate meal suggestions")
btn_gen.setStyleSheet('QPushButton {background-color : rgb(255,0,0); color : black}')
btn_edit = QPushButton('Edit List', self)
btn_edit.setToolTip("Edit Meals")
sld = QSlider(Qt.Horizontal, self)
sld.setToolTip("The ratio of vegetarian dishes in the next list of suggestions")
sld.setMinimum(0)
sld.setMaximum(100)
sld.setValue(70)
sld_lab = QLabel("Veg Dishes " + str(sld.value()) + "%", self)
def onChanged():
text = "Veg Dishes " + str(sld.value()) + "%"
sld_lab.setText(text)
sld.valueChanged.connect(onChanged)
start_cal = QDateTimeEdit(self)
start_cal.setCalendarPopup(True)
start_cal.setMinimumDate(QDate.currentDate())
start_cal.setDisplayFormat("ddd dd.MM")
start_cal.setToolTip("Select a starting date")
start_cal_lab = QLabel("Start:", self)
end_cal = QDateTimeEdit(self)
end_cal.setCalendarPopup(True)
end_cal.setDate(QDate.currentDate().addDays(5))
end_cal.setMinimumDate(QDate.currentDate().addDays(0))
end_cal.setDisplayFormat("ddd dd.MM")
end_cal.setToolTip("Select an end date")
end_cal_lab = QLabel("End:", self)
def startrange():
if end_cal.date() < start_cal.date():
start_cal.setDate(end_cal.date())
end_cal.dateChanged.connect(startrange)
self.setFixedSize(300, 150)
self.center()
self.setWindowTitle('suggestible')
self.setWindowIcon(QIcon('icon.png'))
def createGridLayout(self):
self.horizontalGroupBox = QWidget()
layout = QGridLayout()
layout.setColumnStretch(0, 5)
layout.setColumnStretch(1, 5)
layout.setRowStretch(5, 1)
layout.setVerticalSpacing(8)
layout.addWidget(start_cal_lab, 0, 0)
layout.addWidget(end_cal_lab, 0, 1)
layout.addWidget(start_cal, 1, 0)
layout.addWidget(end_cal, 1, 1)
layout.addWidget(btn_edit, 2, 0)
layout.addWidget(sld, 2, 1)
layout.addWidget(sld_lab, 2, 1, 3, 1, alignment=Qt.AlignCenter)
layout.addWidget(btn_gen, 4, 0, 2, 2)
layout.setRowMinimumHeight(4, 15)
layout.setRowMinimumHeight(5, 15)
self.horizontalGroupBox.setLayout(layout)
createGridLayout(self)
windowLayout = QVBoxLayout()
windowLayout.addWidget(self.horizontalGroupBox)
self.setLayout(windowLayout)
self.show()
def dates(starting, ending):
end = ending.date()
start = starting.date()
en = end.toString("yyyy-MM-dd")
en = parse(en)
st = start.toString("yyyy.MM.dd")
st = parse(st)
def daterange(st, en):
for n in range(int((en - st).days) + 1):
yield st + timedelta(n)
fin_dates = []
for dt in daterange(st, en):
fin_dates.append(dt.strftime("%A %d.%m"))
return fin_dates
def dates_header(starting, ending):
end = ending.date()
start = starting.date()
en = end.toString("dd.MM")
st = start.toString("dd.MM")
return st + "-" + en
def html_creator(starting, ending, ratio0):
templateLoader = jinja2.FileSystemLoader(searchpath="")
templateEnv = jinja2.Environment(loader=templateLoader)
TEMPLATE_FILE = "template_suggestions.html"
template = templateEnv.get_template(TEMPLATE_FILE)
outputText = template.render(range=dates_header(starting, ending), table=main(dates(starting, ending), ratio0))
html_file = open("suggestions" + '.html', 'w')
html_file.write(outputText)
html_file.close()
def generate():
html_creator(start_cal, end_cal, sld.value() / 100)
webbrowser.open("suggestions.html", new=1)
def popupwindow():
self.editor = tablegui.App()
btn_gen.clicked.connect(generate)
btn_edit.clicked.connect(popupwindow)
def center(self):
qr = self.frameGeometry()
cp = QDesktopWidget().availableGeometry().center()
qr.moveCenter(cp)
self.move(qr.topLeft())
if __name__ == '__main__':
app = QApplication(sys.argv)
ex = Main()
sys.exit(app.exec_())