from selenium import webdriver
import time

browser = webdriver.Chrome(executable_path=r'E:\Python\python爬虫\chromedriver.exe')

base_url = 'http://www.baidu.com'
browser.get(base_url)
time.sleep(1)

browser.find_element_by_id('kw').send_keys('贸易战')
browser.find_element_by_id('su').click()

time.sleep(1)

# 截图
browser.save_screenshot('baidu.png')

# 打印 当前浏览器状态的代码
print(browser.page_source)

browser.quit()