spider_baidutongji/auth.py

from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from pyquery import PyQuery as pq
from bs4 import BeautifulSoup
from time import sleep
import time
import traceback
import json
import os
import config
import util

chrome_path = "C:\Program Files (x86)\Google\Chrome\Application\chrome.exe"


def login(url):
    # browser = webdriver.Chrome()
    # browser = webdriver.Firefox()
    options = webdriver.ChromeOptions()
    # 此步骤很重要，设置为开发者模式，防止被各大网站识别出来使用了Selenium
    options.add_experimental_option('excludeSwitches', ['enable-automation'])
    # options.add_argument(r'--headless')
    # 启用缓存，同本机的浏览器一样的的数据
    options.add_argument(r'user-data-dir=C:\Users\Administrator\AppData\Local\Google\Chrome')

    browser = webdriver.Chrome(options=options)
    wait = WebDriverWait(browser, 15)
    browser.maximize_window()
    browser.get(url=url)

    sleep(10)
    # print(a)
    # iframe = browser.find_element_by_tag_name("iframe")
    # browser.switch_to.frame(iframe)
    try:
        # login_button = browser.find_element_by_id('login')
        login_button = browser.find_element_by_class_name('login')
        # browser.execute_script("arguments[0].scrollIntoView(false);", login_button)
        login_button.click()
        # browser.find_element_by_class_name("J_Quick2Static").click()
        # WebDriverWait(browser, 10).until(EC.element_to_be_clickable((By.PARTIAL_LINK_TEXT, u'密码登录'))).click()
        sleep(5)
        browser.find_element_by_id("uc-common-account").clear()
        browser.find_element_by_id("uc-common-account").send_keys("kfzsshop")
        sleep(3)
        browser.find_element_by_id("ucsl-password-edit").clear()
        browser.find_element_by_id("ucsl-password-edit").send_keys("Kfzs654")
        sleep(5)

        browser.find_element_by_id("submit-form").click()
        # browser.find_element_by_id("J_SubmitStatic").click()
        # browser.switch_to.default_content()
        sleep(30)
        cookies = browser.get_cookies()
        print(cookies)
        with open('data.json', 'w') as f:
            json.dump(cookies, f)
        # util.store_file("cookies.txt", cookies)
    # browser.close()
    except:
        print("no_login_id")

    browser.quit()
    return