Test_dada 发表于 2020-5-22 15:05

升级版!绕过验证码自动化登录网址,添加判断过期机制

本帖最后由 Test_dada 于 2020-7-30 10:51 编辑

{:301_997:} 又到新一期的发帖时间
最近开发小哥在登录页面加上了过期机制,一段时间就会登录失效{:1_909:}
but,,这并不能阻挡我解放双手的脚步,额... ...{:1_904:}手步?
之前发过一期自动化登录,虽然绕过了验证码,但是缺点是每过一段时间cookie就会失效,导致代码阻塞
这次加入过期机制,优化:参数脚本分离;提高代码维护性

2.0升级版入口:https://www.52pojie.cn/thread-1233956-1-1.html

话不多说,发!代!码!,觉得有用麻烦点个赞哟{:301_993:}

# ******************
# 通过cookie登录网站
# ******************

from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
import json
from time import sleep
from datetime import datetime
import os

class Test_loggin(object):

    def __init__(self, log_url, home_url, cookie_path, cookie_name = 'cookie.txt', expiration_time = 30):
      '''
      :param log_url: 登录网址
      :param home_url: 首页网址
      :param cookie_path: cookie文件存放路径
      :param cookie_path: 文件命名
      :param expiration_time: cookie过期时间,默认30分钟
      '''
      self.log_url = log_url
      self.home_url = home_url
      self.cookie_path = cookie_path
      self.cookie_name = cookie_name
      self.expiration_time = expiration_time

    def get_cookie(self):
      '''手动登录获取cookie'''
      driver = webdriver.Chrome()
      driver.get(self.log_url)
      driver.maximize_window()

      # 显性等待1,当页面某个元素消失时就执行下列的操作
      WebDriverWait(driver, 30, 2).until_not(lambda x: x.find_element_by_xpath("//button[@id='btnLogin']"))
      print("登录成功")
      # 显性等待2,当页面出现某个元素时就执行下列的操作
      WebDriverWait(driver, 30, 2).until(lambda x: x.find_element_by_xpath("//div[@class='c_title']"))
      print("进入主页验证1成功")
      # 显性等待3,EC.presence_of_element_located会抛出异常
      WebDriverWait(driver, 30, 2).until(EC.presence_of_element_located((By.XPATH, "//div[@class='c_title']")))
      print("进入主页验证2成功")

      with open(os.path.join(self.cookie_path, self.cookie_name), 'w') as cookief:    # 创建文本覆盖保存cookie
            # 将cookies保存为json格式
            cookief.write(json.dumps(driver.get_cookies()))
      print("cookie保存成功")
      driver.close()
      print("浏览器退出")

    def pd_Cookie(self):
      '''获取最新的cookie文件,判断是否过期'''
      cookie_list = os.listdir(self.cookie_path)# 获取目录下所有文件
      if not cookie_list:# 判断文件为空时,直接执行手动登录
            self.get_cookie()
      else:
            cookie_list2 = sorted(cookie_list)# 升序排序文件,返回新列表;sort是对原列表进行排列
            new_cookie = os.path.join(cookie_path, cookie_list2[-1])    # 获取最新cookie文件的全路径

            file_time = os.path.getmtime(new_cookie)# 获取最新文件的修改时间,返回为时间戳1590113596.768411
            t = datetime.fromtimestamp(file_time)# 时间戳转化为字符串日期时间
            print('当前时间:', datetime.now().strftime("%H:%M:%S"))
            print('最新cookie文件修改时间:', t.strftime("%H:%M:%S"))
            date = (datetime.now() - t).seconds // 60# 时间之差,seconds返回相距秒数//60,返回分数
            print('相距分钟:{0}分钟'.format(date))
            if date > self.expiration_time:# 默认判断大于30分钟,即重新手动登录获取cookie
                print("cookie已经过期,请重新手动登录获取")
                return self.get_cookie()
            else:
                print("cookie未过期")

    def cookie_loggin(self):
      '''自动登录操作'''
      self.pd_Cookie()# 首先判断cookie是否已获取,是否过期
      driver = webdriver.Chrome()
      driver.get(self.log_url)
      driver.delete_all_cookies()   # 清除旧cookies

      # with可以上下文管理上文进行设置部署,下文进行处理,然后把处理的结果赋值给变量(cookie)
      with open(os.path.join(self.cookie_path, self.cookie_name),'r') as cookief:
            #使用json读取cookies 注意读取的是文件 所以用load而不是loads
            cookieslist = json.load(cookief)

            for cookie in cookieslist:
                #该字段有问题所以删除就可以浏览器打开后记得刷新页面 有的网页注入cookie后仍需要刷新一下
                if 'expiry' in cookie:
                  del cookie['expiry']
                driver.add_cookie(cookie)

            driver.maximize_window()
            driver.get(self.home_url)
            sleep(4)
            driver.close()


if __name__ == "__main__":
    # 注意:网址要改为自己的网址哟,不然运行不了的
    log_url = 'http://www.test/Login.html'      # 登录页面url
    home_url = 'http://www.test/UserInfo'      # 首页url
    cookie_path = 'D:\python_file\python自动化用例\cookies'   # 存放cookie的路径

    test_loggin = Test_loggin(log_url, home_url, cookie_path)
    test_loggin.cookie_loggin()

Test_dada 发表于 2020-5-22 15:51

显性等待的元素,要改,才能判断登录是否成功

Test_dada 发表于 2020-5-22 15:54

要是手动登录操作慢导致获取cookie失败的话,把显性等待那个30,改大一点,单位是秒

Test_dada 发表于 2020-9-21 09:05

2513002960 发表于 2020-9-18 16:21
我需要获取token,想写个慕课的

if __name__ == "__main__":
    log_url = 'https://www.imooc.com/course/list'         # 登录页面
    home_url = 'https://www.imooc.com/course/list'   # 主页
    cookie_path = 'D:\python_file\cookies'                     # cookies保存绝对路径
    yuanshu = "//a[@id='header-avator']//img"            # 登录后元素验证

    test_loggin = Test_loggin(log_url, home_url, cookie_path, yuanshu)
    test_loggin.cookie_loggin()

框架里面的不用改,就是参数改下就可以自动登录了

Test_dada 发表于 2020-7-29 10:14

本帖最后由 Test_dada 于 2020-7-29 10:15 编辑

玩的天性 发表于 2020-7-28 21:19
求教
http://4a.chinatowercom.cn/uac_oa这个登录要双重验证,账号密码登录后需要接收验证码 ...
安装pymssql包
pip install pymssql

import pymssql

def sql_send(phone):
    # 连接数据库
    con = pymssql.connect(server='数据库ip地址', user='用户名', password='密码', database='库名')
    # 实例化
    cor = con.cursor()
    # 查询语句
    cor.execute("sql查询语句",phone)
    # 返回查询结果(元组)
    return cor.fetchone()
    # # 返回所有元组,为一个list
    # return cor.fetchall()

print(sql_send('号码'))

xz1997 发表于 2020-5-22 15:22

谢谢楼主分享

大兵马元帅 发表于 2020-5-22 15:24

你为什么这么优秀{:1_911:}

hj170520 发表于 2020-5-22 15:26

{:301_1001:} 这个代码具有普适性吗?

Test_dada 发表于 2020-5-22 15:28

hj170520 发表于 2020-5-22 15:26
这个代码具有普适性吗?

之前试过华为官网,淘宝都是可以的,只要改下参数地址就可以测试了

极地企鹅 发表于 2020-5-22 15:32

cookie期限太短

hj170520 发表于 2020-5-22 15:33

Test_dada 发表于 2020-5-22 15:28
之前试过华为官网,淘宝都是可以的,只要改下参数地址就可以测试了

{:301_992:}还有一个问题,就是我之前写代码用过selenium包,这个论坛的大佬推荐我用request 以及lxml,这个也应该可以用requests爬吗?
我发现很多人看得到selenium 以及webdriver() 就没人看帖子了(估计门槛高一些,哈哈{:301_997:})

Test_dada 发表于 2020-5-22 15:33

对了。输入新网址,记得把cookie删掉哦,因为更换网址,旧的cookie用在新网址一定会登录失败的

Test_dada 发表于 2020-5-22 15:38

hj170520 发表于 2020-5-22 15:33
还有一个问题,就是我之前写代码用过selenium包,这个论坛的大佬推荐我用request 以及lxml, ...

找到适合自己项目的方法就好

Test_dada 发表于 2020-5-22 15:39

极地企鹅 发表于 2020-5-22 15:32
cookie期限太短

可以设置expiration_time参数,判断过期时间长短,具体看实际情况
页: [1] 2 3 4 5 6
查看完整版本: 升级版!绕过验证码自动化登录网址,添加判断过期机制