init

2024-05-29 10:21:31 +08:00
commit 54ac29d27b
119 changed files with 6817 additions and 0 deletions
--- a/web/utils/login.py
+++ b/web/utils/login.py
@@ -0,0 +1,97 @@
+import time
+
+from selenium import webdriver
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support import expected_conditions
+from selenium.webdriver.support.wait import WebDriverWait
+
+HOME_PAGE_URL = 'https://fuzhou.mysteel.com/'
+USERNSAME = '13960925044'
+PASSWORD = 'mysteel336627'
+
+
+def load_element(browser, pattern, time=3600.0):
+    ''' 等待元素载入 '''
+    element = WebDriverWait(browser, time).until(expected_conditions.presence_of_element_located((By.XPATH, pattern)))
+    return element
+
+
+def login_mysteel():
+    options = webdriver.ChromeOptions()
+    browser = webdriver.Remote(
+        command_executor="http://localhost:4444/wd/hub",
+        options=options
+    )
+
+    browser.get(HOME_PAGE_URL)
+    print('Visit {0} ...'.format(HOME_PAGE_URL))
+    print(browser.title)
+    try:
+        # 打开登陆窗口
+        print('Click Login...')
+        time.sleep(1)
+        load_element(browser, '//*[@id="mysteel-topBar"]/div[3]/p').click()
+        time.sleep(1)
+
+        # 填写登陆资料
+        print('Insert data ...')
+        print('username: {0}'.format(USERNSAME))
+        browser.find_elements(By.XPATH, "//div[contains(text(),'账号登录')]")[0].click()
+        time.sleep(2)
+        browser.find_elements(By.XPATH, '/html/body/div/div/div[2]/div[5]/div[1]/input')[0].send_keys(USERNSAME)
+        browser.find_elements(By.XPATH, '/html/body/div/div/div[2]/div[5]/div[3]/input')[0].send_keys(PASSWORD)
+        time.sleep(1)
+        browser.find_elements(By.XPATH, '/html/body/div/div/div[2]/div[6]/div[1]/div[1]')[0].click()
+        time.sleep(5)
+
+        cookies = browser.get_cookies()
+        for i in range(10):
+            if cookies:
+                break
+            else:
+                browser.refresh()
+                time.sleep(5)
+                cookies = browser.get_cookies()
+
+        print(cookies)
+        print([i for i in cookies if i['name'] == '_login_token'])
+        print([i for i in cookies if i['name'] == '_MSPASS_SESSION'])
+        return cookies
+    finally:
+        browser.close()
+        browser.quit()
+
+
+def login_baiinfo(home_page_url='http://www.baiinfo.com/shiyou/liqing', usernsame='fjglsl', password='111111'):
+    options = webdriver.ChromeOptions()
+    browser = webdriver.Remote(
+        command_executor="http://localhost:4444/wd/hub",
+        options=options
+    )
+    browser.set_window_size(1280, 800)
+    browser.get('http://www.baiinfo.com/login')
+    print('Visit {0} ...'.format(home_page_url))
+    print(browser.title)
+    try:
+        # 填写登陆资料
+        print('Insert data ...')
+        print('username: {0}'.format(usernsame))
+        load_element(browser, '//*[@id="__nuxt"]')
+        print('ddd')
+        time.sleep(2)
+        browser.find_elements(By.XPATH, '//form/div[1]/input')[0].click()
+        browser.find_elements(By.XPATH, '//form/div[1]/input')[0].send_keys(usernsame)
+        browser.find_elements(By.XPATH, '//form/div[2]/input')[0].send_keys(password)
+        time.sleep(3)
+        browser.find_elements(By.XPATH, '//*[@id="__nuxt"]/div/div[5]/div/div[2]/div/div[2]/div[2]/div[2]/form/div[4]/div/img')[0].click()
+        time.sleep(4)
+        cookies = browser.get_cookies()
+        print(cookies)
+        return cookies
+    finally:
+        browser.close()
+        browser.quit()
+
+
+if __name__ == '__main__':
+    login_baiinfo()