def start(self):
app = QtGui.QApplication(sys.argv)
self.main_window = QtGui.QWidget()
self.main_window.resize(WINDOW_HEIGHT, WINDOW_WIDTH)
self.main_window.setWindowTitle('Hockey probability app')
app.setWindowIcon(QtGui.QIcon(join(dirname(sys.argv[0]), 'hokey.png')))
self.main_layout = QtGui.QVBoxLayout()
raw_data = get_raw_data(players_data_url, matches_data_url, headers)
raw_data_with_tendencies = get_players_tendencies(raw_data)
dump_player_stats(raw_data_with_tendencies)
team_names = raw_data_with_tendencies.get('Tým').unique()
self.data = FormulasRegistry.get_processed_data_for_all_formulas(raw_data_with_tendencies)
team_combo_box_1 = teams_combo_box(team_names)
team_combo_box_2 = teams_combo_box(team_names)
team_combo_box_1.currentIndexChanged['QString'].connect(self.team_combo_box_callback('team1'))
team_combo_box_2.currentIndexChanged['QString'].connect(self.team_combo_box_callback('team2'))
buttons_layout = QtGui.QGridLayout()
buttons_layout.addWidget(team_combo_box_1, 0, 0)
buttons_layout.addWidget(team_combo_box_2, 0, 1)
calculate_btn = QtGui.QPushButton('Calculate probabilities')
calculate_btn.clicked.connect(self.calculate_btn_callback)
self.tabs = construct_tabs({})
self.main_layout.addWidget(self.tabs)
self.main_layout.addLayout(buttons_layout)
self.main_layout.addWidget(calculate_btn)
self.team1 = team_combo_box_1.currentText()
self.team2 = team_combo_box_2.currentText()
self.main_window.setLayout(self.main_layout)
self.main_window.show()
sys.exit(app.exec_())
def calculate_btn_callback(self):
data_to_show = self.filter_data_to_show()
self.tabs.setParent(None)
self.tabs = construct_tabs(data_to_show)
self.main_layout.insertWidget(0, self.tabs)
