from selenium import webdriverfrom selenium.webdriver.common.by import Byfrom selenium.webdriver.common.keys import Keysfrom selenium.webdriver.support.ui import WebDriverWaitfrom selenium.webdriver.support import expected_conditions as ECclass JTWZ():def __init__(self,carAuthCode,carNum):"""carAuthCode:車輛識別碼carNum:車牌號"""self.driver = webdriver.Chrome()self.url = 'http://xxcx.hbsjg.gov.cn:8087/hbjj/'self.carAuthCode=carAuthCodeself.carNum=carNumdef get_content(self):self.driver.get(self.url)try:element = WebDriverWait(self.driver, 10).until(EC.presence_of_element_located((By.ID, "checkCode1")))print(u'開始登錄...')except Exception as e:print(e) self.carNum1 = self.driver.find_element_by_id('carNum1')self.carNum1.send_keys(self.carNum)self.carAuthCode1 = self.driver.find_element_by_id('carAuthCode1')self.carAuthCode1.send_keys(self.carAuthCode)captcha1=self.driver.find_element_by_id('captcha1')#從cookies找尋驗證碼for n in self.driver.get_cookies():if n.get('name')!=None and n['name']=='RANDOMVALIDATECODEKEY1':checkCode1=n['value']captcha1.send_keys(checkCode1)sub=self.driver.find_element_by_xpath("http://input[@value='開始查詢']")sub.click()try:element = WebDriverWait(self.driver, 10).until(EC.presence_of_element_located((By.CLASS_NAME, "fsmiddle")))print(u'獲取違章內(nèi)容成功，保存為:wz.jpg...')self.driver.save_screenshot('wz.jpg')return 0except:print(u'獲取失敗...') return 1finally:self.driver.quit()if __name__ == '__main__':jtwz=JTWZ(carAuthCode=000,carNum='')jtwz.get_content()