java使用Selenium实现获取cookie并自动登录

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档

使用步骤

public class LzqUtils {

    public static void main(String[] args) throws IOException, InterruptedException {
        System.setProperty("webdriver.chrome.driver", "D:\\chromedriver_win32/chromedriver.exe");
        WebDriver webDriver = new ChromeDriver();
        if(!cookieLogin(webDriver)) {
            pLogin(webDriver);
        }
        return;
    }

    private static boolean cookieLogin(WebDriver webDriver) {
        BufferedReader bufferedReader;
        webDriver.get("网站");
        try {
            File cookieFile = new File("D:\\vue\\token/cookie.txt");
            FileReader fileReader = new FileReader(cookieFile);
            bufferedReader = new BufferedReader(fileReader);
            String line;
            while ((line = bufferedReader.readLine()) != null){
                StringTokenizer stringTokenizer = new StringTokenizer(line, ";");
                while (stringTokenizer.hasMoreTokens()){
                    String name = stringTokenizer.nextToken();
                    String value = stringTokenizer.nextToken();
                    Cookie cookie = new Cookie(name, value);
                    webDriver.manage().addCookie(cookie);
                }
            }
            webDriver.navigate().refresh();
            WebElement element =  webDriver.findElement(By.xpath("//button[@class='btn-md btn-theme']"));
            if(element!=null){
                return false;
            }
            return true;
        } catch (Exception e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            return false;
        }
    }

    private static void pLogin(WebDriver webDriver) throws InterruptedException {
        File cookieFile = new File("D:/vue/token/cookie.txt");
        try {
            if(cookieFile.exists()) {
                cookieFile.delete();
            }
            cookieFile.createNewFile();
            FileWriter fileWriter = new FileWriter(cookieFile);
            BufferedWriter bufferedWriter = new BufferedWriter(fileWriter);

            for (Cookie cookie:webDriver.manage().getCookies()){
                bufferedWriter.write((cookie.getName()+";"+
                        cookie.getValue()));
                bufferedWriter.newLine();
            }
            bufferedWriter.flush();
            bufferedWriter.close();
            fileWriter.close();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }
  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
1. 获取cookie 首先,我们需要使用Selenium启动浏览器,并打开我们要自动登录的网站。 ```python from selenium import webdriver # 启动浏览器 driver = webdriver.Chrome() # 打开网站 driver.get('https://www.example.com/') ``` 然后,我们可以使用`driver.get_cookies()`方法获取当前页面的所有cookie。 ```python # 获取所有cookie cookies = driver.get_cookies() # 输出所有cookie for cookie in cookies: print(cookie) ``` 输出的结果类似于: ``` {'domain': 'www.example.com', 'expiry': 1641947537, 'httpOnly': False, 'name': 'SESSIONID', 'path': '/', 'secure': False, 'value': '1234567890'} {'domain': 'www.example.com', 'expiry': 1641947537, 'httpOnly': False, 'name': 'username', 'path': '/', 'secure': False, 'value': 'testuser'} {'domain': 'www.example.com', 'expiry': 1641947537, 'httpOnly': False, 'name': 'password', 'path': '/', 'secure': False, 'value': 'testpassword'} ``` 其中,每个cookie都是一个字典,包含以下键值对: - `domain`:cookie所属的域名。 - `expiry`:cookie的过期时间(以Unix时间戳表示)。 - `httpOnly`:指示浏览器是否只能通过HTTP或HTTPS协议访问cookie。 - `name`:cookie的名称。 - `path`:cookie的路径。 - `secure`:指示浏览器是否只能通过HTTPS协议访问cookie。 - `value`:cookie的值。 2. 使用cookie完成自动登录 获取cookie后,我们可以使用`driver.add_cookie()`方法将cookie添加到浏览器中。然后,我们可以重新打开网站,浏览器会自动使用我们添加的cookie进行登录。 ```python # 添加cookie driver.add_cookie({'name': 'SESSIONID', 'value': '1234567890'}) driver.add_cookie({'name': 'username', 'value': 'testuser'}) driver.add_cookie({'name': 'password', 'value': 'testpassword'}) # 重新打开网站 driver.get('https://www.example.com/') ``` 这样,我们就完成了自动登录功能。 完整代码如下: ```python from selenium import webdriver # 启动浏览器 driver = webdriver.Chrome() # 打开网站 driver.get('https://www.example.com/') # 获取所有cookie cookies = driver.get_cookies() # 输出所有cookie for cookie in cookies: print(cookie) # 添加cookie driver.add_cookie({'name': 'SESSIONID', 'value': '1234567890'}) driver.add_cookie({'name': 'username', 'value': 'testuser'}) driver.add_cookie({'name': 'password', 'value': 'testpassword'}) # 重新打开网站 driver.get('https://www.example.com/') ``` 注意事项: - 获取使用cookie的代码必须在同一个浏览器实例中执行。 - 添加cookie时,必须指定cookie的名称和值。其他键值对是可选的,但建议尽量指定。 - 如果添加的cookie与网站当前存在的cookie冲突,可能会导致自动登录失败。此时,可以尝试清除浏览器缓存或使用不同的浏览器实例。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值