1 # -*- coding:utf-8 -*-
 2 
 3 from selenium import webdriver
 4 from selenium.webdriver.common.by import By
 5 from selenium.webdriver.common.keys import Keys
 6 from selenium.webdriver.support.ui import Select
 7 from selenium.common.exceptions import NoSuchElementException
 8 from selenium.common.exceptions import
 
 NoAlertPresentException
 9 import unittest
10 import time
11 import re
12 import httplib
13 
14 
15 class XiaoMi(unittest.TestCase):
16     def setUp(self):
17         self.driver = webdriver.Chrome()
18         self.base_url = ‘http://item.mi.com/‘
19         self.verificationErrors = []
20         self.accept_next_alert = True
 
21 
22     def login():
23         """
24         用戶登陸
25         """
26         driver = webdriver.Chrome()
27         driver.get("http://item.mi.com/product/10000041.html")
28         driver.find_element_by_xpath(".//*[@id=‘J_userInfo‘]/a[1]").click()
29         time.sleep(2)
30         driver.find_element_by_xpath("
 
.//*[@id=‘username‘]").clear()
31         driver.find_element_by_xpath(".//*[@id=‘username‘]").send_keys(‘用戶名‘)
32         driver.find_element_by_xpath(".//*[@id=‘pwd‘]‘]").clear()
33         driver.find_element_by_xpath(".//*[@id=‘pwd‘]‘]").send_keys(‘密碼‘)
34         dirver.find_element_by_id(‘login-button‘).click()
35         assertEqual(u‘登陸失敗‘, u‘1253785032‘, driver.find_element_by_xpath(".//*[@id=‘J_userInfo‘]/span[1]/a/span").span)
36         
37 
38     def get_xiaomi6(self):
39         ‘‘‘
40         設置9：59：55開始瀏覽器模擬用戶行為不停的點擊加入購物車
41         ‘‘‘
42         def getSysTime():
43             sys_time = time.time()
44             return sys_time
45 
46         def set_stamp():
47             set_time = ‘2017-05-12 09:59:55‘   #設置搶購時間，最好提前幾秒
48             # 將其轉換為時間數組
49             timeArray = time.strptime(set_time, ‘%Y-%m-%d %H:%M:%S‘)
50             # 轉換為時間戳
51             timeStamp = int(time.mktime(timeArray))
52             return timeStamp
53 
54         if getSysTime() >= set_stamp():
55             while True:
56                 self.driver.find_element_by_class_name(‘btn btn-primary btn-biglarge J_proBuyBtn add‘).click()
57 
58                 if self.assertTrue(u‘你來晚了‘, driver.title) == True:
59                     print ‘又悲劇了，默默的問候小米~‘
60                     break
61 
62     def is_element_present(self):
63         try:self.driver.find_element(by=how, value=what)
64         except NoSuchElementException as e:return False
65         return True
66 
67     def is_alert_present(self):
68         try:self.driver.switch_to_alert()
69         except NoAlertPresentException as e:return False
70         return True
71 
72     def close_alert_and_get_its_text(self):
73         try:
74             alert = self.driver.switch_to_alert()
75             alert_text = alert.text
76             if self.accept_next_alert:
77                 alert.accept()
78             else:
79                 alert.dismiss()
80             return alert_text
81         finally:
82             self.accept_next_alert = True
83 
84     def tearDown(self):
85         pass
86         #self.driver.quit()
87         #self.assertEqual([], self.verificationErrors)
88 
89 if __name__ == ‘__main__‘:
90     unittest.main()

python搶小米6自動化腳本