joonvena

search_input = driver.find_element_by_name("q")

joonvena

search_input.send_keys(search_word + " site:stackoverflow.com")
search_input.send_keys(Keys.RETURN)

joonvena

results = wait.until(EC.presence_of_all_elements_located((By.XPATH, '//div[@class="r"]/a')))

joonvena

$x('//div[@class="r"]/a')

joonvena

results[0].click()

joonvena

# Check if page contains element
def element_is_present(xpath):
    try:
        driver.find_element(By.XPATH, xpath)
    except NoSuchElementException:
        return False
    return True

joonvena

# If page contains acccepted answer
if element_is_present('//div[@class="answer accepted-answer"]'):
    answer = driver.find_element(By.XPATH, '//div[@class="answer accepted-answer"]//pre//code')
    print("### Answer ###")
    print(answer.text)
    
# If there is no accepted answer get all the code blocks from the topic
else:
    code_blocks = wait.until(EC.presence_of_all_elements_located((By.XPATH, '//pre/code')))
    for i in range(0,len(code_blocks)):

joonvena

driver.quit()

joonvena

system('cls')

joonvena

system('clear')