update leetcode_generate for Leetcode change the login page

bonfy · bonfy · commit bb36021a9052 · 2018-09-27T09:22:42.000+08:00
diff --git a/leetcode_generate.py b/leetcode_generate.py
@@ -202,10 +202,18 @@ def login(self):
             chrome_options=options, executable_path=executable_path
         )
         driver.get(LOGIN_URL)
-        driver.find_element_by_id('id_login').send_keys(usr)
-        driver.find_element_by_id('id_password').send_keys(pwd)
+
+        # Wait for update
+        time.sleep(10)
+
+        driver.find_element_by_name('login').send_keys(usr)
+        driver.find_element_by_name('password').send_keys(pwd)
         # driver.find_element_by_id('id_remember').click()
-        driver.find_element_by_xpath('//button[@type="submit"]').click()
+        btns = driver.find_elements_by_tag_name('button')
+        # print(btns)
+        submit_btn = btns[1]
+        submit_btn.click()
+
         time.sleep(5)
         webdriver_cookies = driver.get_cookies()
         driver.close()
