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
