ai-yunying/douyin/chat_test.py

import os
import sys
sys.path.append(os.path.dirname(os.path.dirname(__file__)))

from conf.config import logger
from DrissionPage import ChromiumPage, ChromiumOptions

options = ChromiumOptions().set_paths(
    browser_path=r"C:\Program Files\Google\Chrome\Application\chrome.exe",
    local_port=9226,
    user_data_path=r'C:\Users\mrh\AppData\Local\Google\Chrome\User Data')
mg_page = ChromiumPage(options)
# page.new_tab("chrome://version/")
# page.get("https://www.douyin.com/user/self")

class Chat:
    homepage = 'https://www.douyin.com/user/self'
    def __init__(self, ) -> None:
        self.tab_home = mg_page.tab
        # logger.info(f"{self.tab_home.url}") 
        self.check_home_page()
    
    def send_msg(self, msg):
        ele_popShadowAnimation = self.tab_home.ele('xpath://div[@class="qbjZBApl popShadowAnimation"]')



        # 找到输入框
        # data-e2e="msg-input"
        ele_input = ele_popShadowAnimation.ele('xpath://div[@data-e2e="msg-input"]')
        ele_input.click()
        ele_input.input(msg)

        # 找到发送按钮
        # span class="e2e-send-msg-btn"
        ele_send = ele_popShadowAnimation.ele('xpath://span[contains(@class, "e2e-send-msg-btn")]')
        ele_send.click()

        logger.info(f"回复成功")


    def run(self):
        stranger_msg_count = 0
        while True:
            self.check_home_page()

    def check_home_page(self):
        if "/user/self" not in self.tab_home.url:
            self.tab_home.get(self.homepage)

def main():
    mg = Chat()
    mg.send_msg("测试")

if __name__ == "__main__":
    main()