#coding=utf-8
#匯入之前登陸判斷方法
from find_element.kyb_login import driver

driver.find_element_by_id('com.tal.kaoyan:id/login_email_edittext').clear()
driver.find_element_by_id('com.tal.kaoyan:id/login_email_edittext').send_keys('12018')
driver.find_element_by_id('com.tal.kaoyan:id/login_password_edittext').clear()
driver.find_element_by_id('com.tal.kaoyan:id/login_password_edittext'
 
).send_keys('zxw2018')
driver.find_element_by_id('com.tal.kaoyan:id/login_login_btn').click()

#儲存截圖到當前資料夾下
driver.save_screenshot('login.png')

#儲存截圖到指定資料夾images下
driver.get_screenshot_as_file('./images/login1.png')