添加cookie绕过登陆

添加cookie绕过登陆

#!/usr/bin/env python

# -*- coding: utf-8 -*-

from selenium import webdriver

import time

import pickle

class SkipLogin(object):

    def __init__(self):

        self.driver = webdriver.Chrome()

        self.driver.implicitly_wait(15)

        self.driver.maximize_window()

    def add_have_login_state(self):

        try:

            cookies = pickle.load(open("cookies.pkl", "rb"))

            for cookie in cookies:

                cookie_dict = {

                    "domain": ".yaozh.com",  # 火狐浏览器不用填写domain，谷歌要需要

                    'name': cookie.get('name'),

                    'value': cookie.get('value'),

                    "expires": "",

                    'path': '/',

                    'httpOnly': False,

                    'HostOnly': False,

                    'Secure': False}

                self.driver.add_cookie(cookie_dict)

        except Exception as e:

            print(e)

    # 打开个人中心

    def open_personal_center(self):

        self.driver.get("https://www.yaozh.com/member/")

    def is_login(self):

        return self.driver.current_url is not "https://www.yaozh.com/member/"

    # 添加cookies到文件中

    def write_cookies(self):

        self.driver.get("https://www.yaozh.com/login/")

        user_name = self.driver.find_element_by_id("username")

        pwd = self.driver.find_element_by_id("pwd")

        login_button = self.driver.find_element_by_id("button")

        user_name.send_keys("")

        pwd.send_keys("ybeuofnw")

        login_button.click()

        time.sleep(5)

        pickle.dump(self.driver.get_cookies(), open("cookies.pkl", "wb"))

        time.sleep(5)

        self.driver.close()

    def main(self):

        self.open_personal_center()

        if self.is_login():

            self.add_have_login_state()

            # 这里要加一个判断cookie是否有效判断，可以通过页面判断

        time.sleep(5)

        self.open_personal_center()

        time.sleep(5)

        self.driver.close()

if __name__ == '__main__':

    sl = SkipLogin()

    # sl.write_cookies()

    sl.main()