def set_setup(self, qualifying): weather = self.get_weather(qualifying) temp = self.get_temp(qualifying) if qualifying == 'Q2' or 'Race': weather_select_element = browser.find_element_by_name('weatherr') weather_select = Select(weather_select_element) new_weather = weather if weather == 'Rain': new_weather = 'Wet' weather_select.select_by_visible_text(new_weather) weather_input_element = browser.find_element_by_name('tempr') weather_input_element.clear() weather_input_element.send_keys(temp) btn_recalc_element = browser.find_element_by_class_name('SubmitButton') btn_recalc_element.click() setup_elements = browser.find_elements_by_xpath('//th[@class="Calc Blue"]/input') setup = Setup() setup.original_front_wing = setup_elements[0].get_attribute('value') setup.original_rear_wing = setup_elements[1].get_attribute('value') setup.engine = setup_elements[2].get_attribute('value') setup.brakes = setup_elements[3].get_attribute('value') setup.gear = setup_elements[4].get_attribute('value') setup.suspension = setup_elements[5].get_attribute('value') setup.qualifying = qualifying setup.temperature = temp setup.weather = weather return setup