from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
import time
import os

def screenshot(driver,file_path = None):
#用户没有传参数
if file_path == None:
project_path = os.path.dirname(os.getcwd())
print(project_path)
file_path = project_path +"\images"
if not os.path.exists(file_path):
os.mkdir(file_path)
images_name = time.strftime("%Y-%m-%d %H:%M:%S",time.localtime())
file_path = file_path+images_name+".png"
print(file_path)
driver.save_screenshot(file_path)

try:
driver = webdriver.Chrome()
driver.maximize_window()
url = "https://www.cnblogs.com/"
driver.get(url)

ele = driver.find_element_by_xpath(’//*[@id=“cate_item_2”]’)

ActionChains(driver).move_to_element(ele).perform()

driver.find_element_by_partial_link_text(‘python’).click()

time.sleep(3)

#浏览器句柄切换
#先获取浏览器所有句柄
handles = driver.window_handles
#当前浏览器句柄
current_handles = driver.current_window_handle
#for循环句柄
for handle in handles:
#如果handle不是当前handle
if handle != current_handles:
#handle就句柄切换一下
driver.switch_to.window(handle)
screenshot(driver)#自己写的screenshot

finally:
time.sleep(3)
driver.quit()