使用Selenium的Python脚本实现自动登录

使用Selenium的Python脚本实现自动登录

背景：连接WiFi时，会自动弹出登录页，其中包含一个用于输入用户名和密码的iframe，所以，笔者试图使用Selenium来实现自动化登录的流程。

什么是iframe？

在HTML中，</code>标签用来嵌入另一个HTML文档。iframe可以用来嵌入广告、地图、视频等独立内容。为了操作iframe中的内容，我们需要先切换到该iframe的上下文。 <h4><a name="t1"></a><a id="iframe_5"></a>为什么要切换到iframe？</h4> 当页面中有iframe时，Selenium默认在主文档上下文中操作。如果我们想操作iframe中的元素，必须先切换到iframe的上下文，否则Selenium无法找到这些元素。 <h4><a name="t2"></a><a id="iframe_9"></a>如何切换到iframe？</h4> 使用Selenium，我们可以通过多种方式定位iframe并切换到其中。常见的方法包括通过iframe的<code>id</code>、<code>name</code>、<code>索引</code>或者直接传递iframe的WebElement对象。 <h4><a name="t3"></a><a id="_13"></a>代码示例</h4> 以下是一个示例，演示如何在包含iframe的网页中操作iframe中的元素： <h5><a id="1_HTML_17"></a>1. 示例HTML页面</h5> 假设我们有一个包含iframe的HTML页面，如下所示： <pre data-index="0" class="set-code-show prettyprint"><code class="prism language-html has-numbering" onclick="mdcp.signin(event)" style="position: unset;"><!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Example with iframe</title> </head> <body> <h1>Main Page</h1> <iframe id="my_iframe" src="iframe_content.html" name="iFrame1"></iframe> </body> </html> <div class="hljs-button signin active" data-title="登录复制" data-report-click="{"spm":"1001.2101.3001.4334"}"></div></code></pre> <code>iframe_content.html</code>文件内容如下： <pre data-index="1" class="set-code-show prettyprint"><code class="prism language-html has-numbering" onclick="mdcp.signin(event)" style="position: unset;"><!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Iframe Content</title> </head> <body> <h1>Iframe Content</h1> <input type="text" id="username" placeholder="Enter username"> <input type="password" id="password" placeholder="Enter password"> <button id="login_button">Login</button> </body> </html> <div class="hljs-button signin active" data-title="登录复制" data-report-click="{"spm":"1001.2101.3001.4334"}"></div></code></pre> <h5><a id="2_Python_53"></a>2. Python脚本</h5> 下面是使用Selenium的Python脚本，通过iframe的<code>name</code>属性切换到iframe并操作其中的元素： <pre data-index="2" class="set-code-hide prettyprint"><code class="prism language-python has-numbering" onclick="mdcp.signin(event)" style="position: unset;">import time from selenium import webdriver from selenium.webdriver.common.by import By from selenium.webdriver.chrome.service import Service from webdriver_manager.chrome import ChromeDriverManager from selenium.webdriver.chrome.options import Options # 配置Chrome选项 chrome_options = Options() chrome_options.add_argument("--disable-gpu") chrome_options.add_argument("--no-sandbox") # 初始化Chrome WebDriver driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=chrome_options) try: # 打开主页面 driver.get("file:///path/to/your/main_page.html") # 替换为实际页面的路径 # 等待页面加载 time.sleep(2) # 切换到iframe driver.switch_to.frame(driver.find_element(By.NAME, "iFrame1")) # 现在可以操作iframe中的元素 username_input = driver.find_element(By.ID, "username") username_input.send_keys("my_username") password_input = driver.find_element(By.ID, "password") password_input.send_keys("my_password") login_button = driver.find_element(By.ID, "login_button") login_button.click() # 等待一段时间以确保操作完成（根据需要调整时间） time.sleep(5) finally: # 关闭浏览器 driver.quit() <div class="hljs-button signin active" data-title="登录复制" data-report-click="{"spm":"1001.2101.3001.4334"}"></div></code></pre> <h4><a name="t4"></a><a id="_101"></a>详细步骤解释</h4> <ol><li> 初始化WebDriver： <pre data-index="3" class="set-code-show prettyprint"><code class="prism language-python has-numbering" onclick="mdcp.signin(event)" style="position: unset;">driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=chrome_options) <div class="hljs-button signin active" data-title="登录复制" data-report-click="{"spm":"1001.2101.3001.4334"}"></div></code></pre> </li><li> 打开主页面： <pre data-index="4" class="set-code-show prettyprint"><code class="prism language-python has-numbering" onclick="mdcp.signin(event)" style="position: unset;">driver.get("file:///path/to/your/main_page.html") <div class="hljs-button signin active" data-title="登录复制" data-report-click="{"spm":"1001.2101.3001.4334"}"></div></code></pre> 这一步将打开包含iframe的主页面。 </li><li> 等待页面加载： <pre data-index="5" class="set-code-show prettyprint"><code class="prism language-python has-numbering" onclick="mdcp.signin(event)" style="position: unset;">time.sleep(2) <div class="hljs-button signin active" data-title="登录复制" data-report-click="{"spm":"1001.2101.3001.4334"}"></div></code></pre> 由于页面加载需要时间，使用<code>time.sleep</code>等待几秒钟。 </li><li> 切换到iframe： <pre data-index="6" class="set-code-show prettyprint"><code class="prism language-python has-numbering" onclick="mdcp.signin(event)" style="position: unset;">driver.switch_to.frame(driver.find_element(By.NAME, "iFrame1")) <div class="hljs-button signin active" data-title="登录复制" data-report-click="{"spm":"1001.2101.3001.4334"}"></div></code></pre> 使用<code>switch_to.frame</code>方法切换到iframe的上下文，方法接受一个WebElement对象。这里通过iframe的<code>name</code>属性来定位iframe元素。 </li><li> 操作iframe中的元素： <pre data-index="7" class="set-code-show prettyprint"><code class="prism language-python has-numbering" onclick="mdcp.signin(event)" style="position: unset;">username_input = driver.find_element(By.ID, "username") username_input.send_keys("my_username") password_input = driver.find_element(By.ID, "password") password_input.send_keys("my_password") login_button = driver.find_element(By.ID, "login_button") login_button.click() <div class="hljs-button signin active" data-title="登录复制" data-report-click="{"spm":"1001.2101.3001.4334"}"></div></code></pre> 一旦切换到iframe上下文，就可以像操作主文档中的元素一样操作iframe中的元素。 </li><li> 关闭浏览器： <pre data-index="8" class="set-code-show prettyprint"><code class="prism language-python has-numbering" onclick="mdcp.signin(event)" style="position: unset;">driver.quit() <div class="hljs-button signin active" data-title="登录复制" data-report-click="{"spm":"1001.2101.3001.4334"}"></div></code></pre> 在操作完成后，关闭浏览器。 </li></ol> <h4><a name="t5"></a><a id="_145"></a>小结</h4> 切换到iframe的过程主要包括定位iframe元素并使用<code>switch_to.frame</code>方法切换到iframe上下文。在此之后，Selenium的所有操作都会在iframe中进行，直到切换回主文档或另一个iframe。切换回主文档可以使用<code>driver.switch_to.default_content()</code>。这种方法适用于嵌套iframe的复杂页面结构。 </div> </div> </li> <li class="list-group-item ul-li"> 相关阅读: <a href="/Article/Index/1140683">三菱机床联网</a> <a href="/Article/Index/613035">程序员转业指南 - 当文员</a> <a href="/Article/Index/1123207">C++STL——string类与模拟实现</a> <a href="/Article/Index/766538">DevOps流程demo（实操记录）</a> <a href="/Article/Index/1219414">【TI毫米波雷达笔记】IWR6843AOPEVM-G+DCA1000EVM的mmWave Studio数据读取、配置及避坑</a> <a href="/Article/Index/1369712">【Java】查找jdk步骤</a> <a href="/Article/Index/1068611">新型氢氧型N,N,N′,N′-四甲基-N″-丁基胍离子液体[BTMG]OH离子液体</a> <a href="/Article/Index/1017236">2. 设计模式-详解UML类图</a> <a href="/Article/Index/1373006">普冉PY32系列(八) GPIO模拟和硬件SPI方式驱动无线收发芯片XN297LBW</a> <a href="/Article/Index/1005700">真心求问：拼夕夕的链接到底是怎么搞的？能用其他大厂域名？</a> </li> <li class="list-group-item from-a mb-2"> 原文地址：https://blog.csdn.net/weixin_43844521/article/details/139322101 </li> </ul> </div> <div class="col-lg-4 col-sm-12"> <ul class="list-group" style="word-break:break-all;"> <li class="list-group-item ul-li-bg" aria-current="true"> 最新文章 </li> <li class="list-group-item ul-li"> <a href="/Article/Index/2039405">沪漂五周年了：我越来越迷茫了</a> <a href="/Article/Index/2039406">Agentic Skill Routing 实战：别再把所有 Skill 塞进 AI Agent 上下文</a> <a href="/Article/Index/2039407">MySQL-Seconds_behind_master的精度误差</a> <a href="/Article/Index/2039408">[MAF预定义ChatClient中间件-03]CachingChatClient——利用缓存省钱省时间</a> <a href="/Article/Index/2039409">AI的至暗历史：从万众期待到被政府撤资，AI的两次死亡徘徊</a> <a href="/Article/Index/2039410">Agent OS ：五种驯服不确定性的范式</a> <a href="/Article/Index/2039411">PortSwigger SQL注入LAB11</a> <a href="/Article/Index/2039412">数据库即时编译JIT</a> <a href="/Article/Index/2039413">[Begin]AI Learn Data Day 0</a> <a href="/Article/Index/2039414">深度学习进阶（二十七）现代 LLM 的核心架构设计其二：SwiGLU</a> </li> </ul> <ul class="list-group pt-2" style="word-break:break-all;"> <li class="list-group-item ul-li-bg" aria-current="true"> 热门文章 </li> <li class="list-group-item ul-li"> <a href="/Article/Index/888177">十款代码表白小特效一个比一个浪漫赶紧收藏起来吧！！！</a> <a href="/Article/Index/797680">奉劝各位学弟学妹们，该打造你的技术影响力了！</a> <a href="/Article/Index/888183">五年了，我在 CSDN 的两个一百万。</a> <a href="/Article/Index/888179">Java俄罗斯方块，老程序员花了一个周末，连接中学年代！</a> <a href="/Article/Index/797730">面试官都震惊，你这网络基础可以啊！</a> <a href="/Article/Index/797725">你真的会用百度吗？我不信 — 那些不为人知的搜索引擎语法</a> <a href="/Article/Index/797702">心情不好的时候，用 Python 画棵樱花树送给自己吧</a> <a href="/Article/Index/797709">通宵一晚做出来的一款类似CS的第一人称射击游戏Demo！原来做游戏也不是很难，连憨憨学妹都学会了！</a> <a href="/Article/Index/797716">13 万字 C 语言从入门到精通保姆级教程2021 年版</a> <a href="/Article/Index/888192">10行代码集2000张美女图，Python爬虫120例，再上征途</a> </li> </ul> </div> </div> </div>     <nav class="navbar navbar-inverse navbar-fixed-bottom"> <div class="container"> <div class="row"> <div class="col-md-12 text-center"> <div style="padding:0px 5px;"> <a href="/tools">小工具</a> <a href="https://games.1000bd.com">小游戏</a> </div> <div class="text-muted center foot-height"> Copyright © 2022 侵权请联系<a href="mailto:2656653265@qq.com">2656653265@qq.com</a>    <a href="https://beian.miit.gov.cn/" target="_blank">京ICP备2022015340号-1</a> </div> <div style=" padding:5px 0;"> <a target="_blank" href="http://www.beian.gov.cn/portal/registerSystemInfo?recordcode=11010502049817" style="display:inline-block;text-decoration:none;height:20px;line-height:20px;"> <img src="" style="float:left;" />京公网安备 11010502049817号</a> </div> </div> </div> </div> </nav>   <script src="/js/plugins/jquery/jquery.js"></script> <script src="/js/bootstrap.min.js"></script>   </body> </html>

什么是iframe？