用selenium自动登录虎牙账号

费小牛 · 发表于 2024-9-24 16:28:58

from random import choice
from time import sleep
from selenium import webdriver
from selenium.webdriver import ActionChains
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.service import Service


def parse_data(wd):
    sleep(1)

    room_list = wd.find_elements(By.XPATH,"/html/body/div[1]/div/div[2]/div/div/div[3]/ul/li")
    print(len(room_list))

    temp_list = []

    for room in room_list:
        temp = {}
        try:
            # temp["title"] = room.find_element(By.XPATH,"./a[2]/li[1]/a[2]").get_attribute("title")
            temp["url"] = room.find_element(By.XPATH,"./a[1]").get_attribute("href")
            temp["owner"] = room.find_element(By.XPATH,"./span[1]/span[1]/i").text
            temp["type"] = room.find_element(By.XPATH,"./span[1]/span[2]/i[2]").text
            # temp["num"] = room.find_element(By.XPATH,"./span[1]/span[3]/i[2]").text
            temp_list.append(temp)
        except:
            continue
        
    return temp_list


def save_data(temp_list):
    count = 1
    for temp in temp_list:
        print(count,temp)
        count += 1
def HuyaDanmu(address,ua):
    options = webdriver.ChromeOptions()            # 定义Chrome配置参数
    options.add_argument('--ignore-certificate-errors')           # 忽略https证书错误
    options.add_argument('user-agent=' + ua)                      # 设置user-agent，取消浏览器被识别出来
    # options.add_argument('--headless')                            # 该‘浏览器不提供可视化页面’设置去掉
    # options.add_argument('blink-settings=imagesEnabled=false')  # 添加不加载图片设置，提升速度

    s = Service(address)                                            # 加载Chrome服务
    wd = webdriver.Chrome(service=s, options=options)                 # 创建浏览器实例
    wd.implicitly_wait(160)                                          # 让浏览器实例隐式等待60秒，方便浏览器加载完成后就可以获取页面的元素
    # 调用WebDriver 对象的get方法 可以让浏览器打开指定网址
    wd.get('https://www.huya.com')
    #点击登陆出现扫码登陆
    # wd.find_element(By.LINK_TEXT, "登录").click()
    #账号密码在记事本中，第一行账号，第二行密码
    with open(r"G:\TDDOWNLOAD\qq自动化登录虎牙\LoginHuyaByQ\HY账号密码.txt", "r", encoding="utf-8") as g:
        hypassword = g.readlines()
        print(hypassword[0])
        print(hypassword[1])
    # wd.switch_to.frame("UDBSdkLgn_iframe")                          #定位第一个弹窗并关闭
    # wd.find_element(By.ID, 'close-udbLogin').click()

    action = ActionChains(wd)
    denglu = wd.find_element(By.CLASS_NAME, "LoginHd--1Jf6S0CCU3DUkJdjVqVn3")   #定位悬停位置
    action.move_to_element(denglu).perform()                                      #鼠标悬停

    # wd.find_element(By.LINK_TEXT, "点我注册").click()
    # wd.switch_to.frame("UDBSdkLgn_iframe")
    # wd.find_element(By.CLASS_NAME, 'returnlogin-btn').click()
    # # wd.find_element(By.XPATH, "//*[@id='account-login-form']/div[1]/input").send_keys(hypassword[0])
    # wd.find_element(By.XPATH, "//*[@id=‘account-login-form’]/div[1]/input").send_keys(hypassword[0])
    #
    # # wd.find_element(By.XPATH, "//*[@id='account-login-form']/div[2]/input").send_keys(hypassword[1])
    # wd.find_element(By.XPATH, "//*[@id=‘account-login-form’]/div[2]/input").send_keys(hypassword[1])
    # wd.find_element(By.XPATH, "//*[@id='login-btn']").click()
    sleep(10)



#打印模拟器板块主播数量以及具体信息
    wd.get("https://www.huya.com/g/3203#cate-1-4507-0")
    temp_list_monq = parse_data(wd)
    save_data(temp_list_monq)

    wd.get("https://www.huya.com/g/3203")
    temp_list = parse_data(wd)
    save_data(temp_list)

    while True:

        for roomherf in temp_list:
            wd.get(roomherf['url'])

            f = open("G:\TDDOWNLOAD\qq自动化登录虎牙\LoginHuyaByQ\主播打招呼文本.txt", "r", encoding="UTF-8")  # 设置文件对象
            lines = f.readlines()  # 将txt文件的所有内容读入到字符串str中
            f.close()

            messageBox = wd.find_element(By.XPATH, '//*[@id="pub_msg_input"]')  # 找到页面上的输入框，用于发送弹幕
            messageBut = wd.find_element(By.XPATH, '//*[@id="msg_send_bt"]')  # 找到发送弹幕按钮
            danmu = choice(lines).strip('\n')  # 随机从文件中读取一个弹幕
            messageBox.send_keys(danmu)  # 在输入框中输入弹幕
            sleep(1)
            messageBut.click()  # 点击发送弹幕
            sleep(20)
        break
            # try:
            #     el_next = wd.find_element_by_xpath("//a[@class='laypage_next']")
            #     wd.execute_script("scrollTo(0, 100000)")
            #     el_next.click()
            # except:
            #     print("结束！")
            # break





if __name__ == '__main__':
    # f = open("settings.txt", "r",encoding='utf-8')  # 设置文件对象
    # lines = f.readlines()  # 将txt文件的所有内容读入到字符串str中
    address="G:\TDDOWNLOAD\qq自动化登录虎牙\LoginHuyaByQ\chromedriver.exe"
    ua="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36"
    # f.close()

    HuyaDanmu(address,ua)
    input("输入任意键结束")

现在这个网页怎么获取登录框信息，切换不到那个ifram框里，捣鼓半天没弄不好，想爬取每个网页信息现在最多获取7个，还请各位大佬帮忙，谢谢了，代码让我改的烂摊子了

FishC · 发表于 2024-9-24 16:35:33

要用 Selenium 自动登录虎牙账号并爬取房间信息，可以按照以下步骤进行修改和完善代码：

1. **安装相关库（如果未安装）：**

pip install selenium

2. **初始化 WebDriver 和登录功能：**

from selenium import webdriver
   from selenium.webdriver.common.by import By
   from selenium.webdriver.chrome.service import Service
   from time import sleep

   def login_huya(wd, username, password):
       # 打开虎牙登录页
       wd.get("https://www.huya.com")

       # 点击登录按钮
       login_button = wd.find_element(By.LINK_TEXT, "登录")
       login_button.click()

       # 切换到账号登录方式
       wd.switch_to.frame(wd.find_element(By.ID, 'UDBSdkLgn_iframe_content'))
       account_login = wd.find_element(By.CSS_SELECTOR, ".udbsdk_udblogin_tab-item:nth-child(2)")
       account_login.click()

       # 输入用户名和密码
       wd.find_element(By.ID, "account-login-accountName").send_keys(username)
       wd.find_element(By.ID, "account-login-password").send_keys(password)

       # 点击登录
       wd.find_element(By.ID, "login-btn-submit").click()

       # 确保登录成功，可以添加合适的等待时间
       sleep(5)

   def parse_data(wd):
       sleep(1)
       room_list = wd.find_elements(By.XPATH, "/html/body/div/div/div/div/div/div/ul/li")
       temp_list = []

       for room in room_list:
           temp = {}
           try:
               temp['url'] = room.find_element(By.XPATH, "./a").get_attribute("href")
               temp['title'] = room.find_element(By.XPATH, "./a").get_attribute("title")
               temp['info'] = room.find_element(By.XPATH, "./span/span/i").text
               temp_list.append(temp)
           except Exception as e:
               print(e)
               continue

       return temp_list

   def save_data(temp_list):
       count = 1
       for temp in temp_list:
           print(count, temp)
           count += 1

   def HuyaDanmu(address, ua, username, password):
       options = webdriver.ChromeOptions()
       options.add_argument('--ignore-certificate-errors')
       options.add_argument('user-agent=' + ua)

       s = Service(address)
       wd = webdriver.Chrome(service=s, options=options)
       wd.implicitly_wait(60)

       login_huya(wd, username, password)

       temp_list = parse_data(wd)
       save_data(temp_list)
       wd.quit()

   # 示例调用
   HuyaDanmu("path_to_chromedriver", "your_user_agent", "your_username", "your_password")

请将 `"path_to_chromedriver"` 修改为 `chromedriver` 的实际路径，将 `"your_user_agent"` 修改为实际的浏览器 UA（用户代理）， `"your_username"` 和 `"your_password"` 修改为实际的虎牙账号和密码。

注意：由于虎牙的登录方式较为复杂，有时可能会遇到验证码或其他验证流程。以上脚本使用了账号密码登录方式，如果虎牙过程中有任何修改，脚本可能需要相应更新。另外，请确保访问控制策略和节操。

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。

很cool的阳 · 发表于 2024-9-26 17:26:08

1. **等待元素可见**：使用`WebDriverWait`和`expected_conditions`来等待登录框的iframe可见。
2. **检查iframe的属性**：确认iframe的id和class属性是否正确，有时候这些属性可能会因为网站更新而改变。
3. **滚动到元素可见**：有时候iframe可能不在当前视图内，需要滚动到该元素。
4. **检查iframe是否有多层嵌套**：有时候登录框可能在一个嵌套的iframe中，需要逐层切换。

下面是一个更新后的代码示例，它使用了`WebDriverWait`来等待登录框的iframe可见，并尝试切换到该iframe：

```python
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.chrome.service import Service
import time

def HuyaDanmu(address, ua):
options = webdriver.ChromeOptions()
options.add_argument('--ignore-certificate-errors')
options.add_argument('user-agent=' + ua)
s = Service(address)
wd = webdriver.Chrome(service=s, options=options)
wd.implicitly_wait(10)

wd.get('https://www.huya.com')
time.sleep(5)  # 等待页面加载

try:
      # 等待登录iframe可见
      iframe = WebDriverWait(wd, 10).until(
         EC.presence_of_element_located((By.XPATH, '你的登录iframe的XPATH'))
      )
      wd.switch_to.frame(iframe)  # 切换到iframe
except TimeoutException:
      print("登录iframe未找到")
      return

# 现在你应该在登录iframe中，可以进行登录操作
# ...

if __name__ == '__main__':
address = "你的chromedriver路径"
ua = "你的user-agent字符串"
HuyaDanmu(address, ua)
input("输入任意键结束")
```

请注意，你需要将`你的登录iframe的XPATH`替换为实际的XPATH，可以通过检查页面元素来找到正确的XPATH。

费小牛 · 发表于 2024-9-27 18:26:59

FishC 发表于 2024-9-24 16:35
要用 Selenium 自动登录虎牙账号并爬取房间信息，可以按照以下步骤进行修改和完善代码：

1. **安装相关 ...

哥哥，还是没法切换到哪个登录iframe登录框

是我那个地方做错了吗

FishC · 发表于 2024-9-27 18:27:04

费小牛发表于 2024-09-27 18:26
哥哥，还是没法切换到哪个登录iframe登录框 ...

我明白了，可能是因为 iframe 切换出错，导致无法正确切换到登录框。我们可以通过检查和确定 iframe 的 ID 和位置来解决这个问题。以下是一个修正版本，确保按步骤替代和测试：

```python
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.service import Service
from time import sleep

def login_huya(wd, username, password):
# 打开虎牙主页
wd.get("https://www.huya.com")

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。

费小牛 · 发表于 2024-9-27 18:31:49

很cool的阳发表于 2024-9-26 17:26
1. **等待元素可见**：使用`WebDriverWait`和`expected_conditions`来等待登录框的iframe可见。
2. **检查 ...

哥，您这个IFRAME对话框我就切换成功进去一次再次运行再也不成功了

FishC · 发表于 2024-9-27 18:31:57

费小牛发表于 2024-09-27 18:31
哥，您这个IFRAME对话框我就切换成功进去一次再次 ...

要正确地切换到iframe，并确保操作后的元素找到正确，可以尝试下面的方法：

1. **等待iframe出现**：
确保iframe已经加载成功，之后再进行切换。

2. **明确地切换到iframe并操作**：
利用WebDriverWait等待元素，确保切换到iframe后再进行操作。

下面是一个示例代码，可以帮助你正确切换到iframe并进行登录操作：

```python
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.service import Service

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。

很cool的阳 · 发表于 2024-9-28 17:20:33

费小牛发表于 2024-9-27 18:26
哥哥，还是没法切换到哪个登录iframe登录框是我那个地方做错了吗

试下这样行不行
1. **定位iframe并切换**：首先，您需要确保您定位到了正确的iframe。您可以使用iframe的id、name或者WebElement对象来定位。例如，如果您知道iframe的id，您可以这样做：
```python
iframe = wd.find_element(By.ID, 'iframe_id')
wd.switch_to.frame(iframe)
```
如果没有id，您可以尝试使用其他属性或者XPath来定位iframe。例如：
```python
iframe = wd.find_element(By.XPATH, "//iframe[contains(@src, 'iframe_src')]")
wd.switch_to.frame(iframe)
```

2. **等待iframe加载**：有时候iframe可能还没有加载完成，您需要等待它变得可用。您可以使用`WebDriverWait`和`expected_conditions`来等待iframe的加载：
```python
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

iframe = WebDriverWait(wd, 10).until(
EC.frame_to_be_available_and_switch_to_it((By.ID, 'iframe_id'))
)
```

3. **处理嵌套iframe**：如果iframe中还有iframe，您可能需要逐层切换。每次切换到新的iframe后，都要重新定位元素。

4. **切换回主文档**：在iframe中操作完成后，您需要切换回主文档来继续其他操作：
```python
wd.switch_to.default_content()
```

5. **使用JavaScript切换**：如果Selenium的常规方法不起作用，您可以尝试使用JavaScript来切换iframe：
```python
wd.execute_script("window.frames['iframe_id'].focus();")
```

不二如是 · 发表于 2024-10-27 22:15:38

最终解决后的代码，记得发一下

账号		自动登录	找回密码
密码			立即注册