Script/autowrite_day.py

"""
日报自动化填写脚本

依赖安装：
1. 安装Python依赖包：
   pip install playwright
   pip install psutil
   pip install asyncio

2. 安装Playwright浏览器：
   playwright install chromium

注意事项：
1. 确保Chrome浏览器已安装
2. 脚本使用Chrome用户配置文件，路径为：C:\Users\zhens\AppData\Local\Google\Chrome\User Data
3. 运行前请确保没有其他Chrome进程在运行
4. 脚本会自动关闭所有Chrome进程，请确保没有重要工作未保存
5. 如果修改了Chrome用户配置路径，需要相应修改代码中的user_data_dir变量
6. 脚本运行时会打开Chrome浏览器窗口，请勿手动操作浏览器
7. 如果遇到网络问题，可以适当增加等待时间（asyncio.sleep的值）
8. 如果提交失败，可以检查网络连接或适当增加等待时间
"""

import asyncio
import os
import psutil
import time
from playwright.async_api import async_playwright

# 执行自动化浏览器操作的协程函数

def close_chrome():
    """关闭所有Chrome进程"""
    for proc in psutil.process_iter(['name']):
        try:
            if proc.info['name'] == 'chrome.exe':
                proc.kill()
        except (psutil.NoSuchProcess, psutil.AccessDenied):
            pass
    # 等待进程完全关闭
    time.sleep(2)

async def run(playwright):
    # 确保Chrome已关闭
    close_chrome()
    
    # 使用用户实际的Chrome配置，配置查询chrome://version
    user_data_dir = r'C:\Users\zhens\AppData\Local\Google\Chrome\User Data'
    
    def text_to_div_html(text):
        """将多行文本转换为<div>...</div>格式的HTML"""
        lines = [line.strip() for line in text.strip().splitlines() if line.strip()]
        return ''.join(f'<div>{line}</div>' for line in lines)

    # 你可以在这里自定义要填写的内容
    summary_text = """
    一、今日完成CRM脚本开发
    二、完成小程序测试工作
    三、修复了异常处理bug
    """
    plan_text = """
    一、明天继续完善脚本
    二、小程序工人付费工作完善
    三、增加更多异常场景测试
    """
    summary_html = text_to_div_html(summary_text)
    plan_html = text_to_div_html(plan_text)

    try:
        # 使用已有的Chrome配置文件启动浏览器
        context = await playwright.chromium.launch_persistent_context(
            user_data_dir,
            channel="chrome",
            headless=False,
            args=[
                '--start-maximized',
                '--disable-blink-features=AutomationControlled',
                '--profile-directory=Default',
                '--no-first-run',
                '--no-default-browser-check'
            ],
            ignore_default_args=['--enable-automation']
        )

        try:
            # 创建新页面
            page = await context.new_page()
            
            # 设置实际的User-Agent
            await page.set_extra_http_headers({
                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/135.0.0.0 Safari/537.36'
            })
            
            # 访问页面并等待网络空闲
            await page.goto('https://doc.weixin.qq.com/forms/j/AFIADwd9AA4AUcA6AbCADgJXbq19fUR0j_base?page=6', 
                            wait_until='networkidle')
            
            # 等待元素出现并点击
            await page.wait_for_selector('div.TitleBarBtn_title__dKAyV:text("填写")', timeout=60000)
            await page.click('div.TitleBarBtn_title__dKAyV:text("填写")', delay=100)

            # 等待操作完成，确保页面完全加载
            await asyncio.sleep(5)

            # 查找所有可编辑的输入框
            editors = await page.query_selector_all('.maileditor-editorview[contenteditable="true"]')
            print("找到输入框数量：", len(editors))

            # 打印每个输入框的父级HTML，帮助确认 data-qid，每次打开data-qid都会不同所以不准确
            # for i, editor in enumerate(editors):
            #     parent = await editor.evaluate_handle('el => el.closest("div.question")')
            #     parent_html = await parent.evaluate('el => el.outerHTML')
            #     print(f"输入框{i+1}父级HTML：\n{parent_html}\n")

            # if len(editors) >= 2:
            #     await editors[0].evaluate(f'el => el.innerHTML = `{summary_html}`')
            #     await editors[1].evaluate(f'el => el.innerHTML = `{plan_html}`')

            # 今日工作总结输入
            # 1. 先点击输入框，确保获得焦点
            await page.locator('div.question:has-text("今日工作总结") .maileditor-editorview[contenteditable="true"]').click()
            # 2. 使用键盘输入方式模拟真实输入，这样可以触发输入事件
            await page.keyboard.type(summary_text)
            # 3. 等待输入完成，确保内容被正确输入
            await asyncio.sleep(2)
            
            # 明日工作计划输入
            # 1. 先点击输入框，确保获得焦点
            await page.locator('div.question:has-text("明日工作计划") .maileditor-editorview[contenteditable="true"]').click()
            # 2. 使用键盘输入方式模拟真实输入，如果通过代码编辑，会导致输入栏没有任何信息
            await page.keyboard.type(plan_text)
            # 3. 等待输入完成
            await asyncio.sleep(2)

            # 等待内容保存
            # 给系统足够的时间来保存输入的内容
            await asyncio.sleep(5)

            # 点击提交按钮
            # 1. 等待提交按钮出现
            await page.wait_for_selector('button.FillFooter_confirm__0ClPl', timeout=60000)
            # 2. 点击提交按钮，添加延迟模拟真实点击
            await page.click('button.FillFooter_confirm__0ClPl', delay=100)
            
            # 等待提交完成
            # 给系统足够的时间来处理提交操作
            await asyncio.sleep(5)

        finally:
            # 确保浏览器正常关闭
            await context.close()
    except Exception as e:
        print(f"发生错误: {str(e)}")
        raise

# 主函数，用于启动 playwright 并调用 run 函数


async def main():
    async with async_playwright() as playwright:
        await run(playwright)

# 判断当前环境是否已经有事件循环在运行
if __name__ == "__main__":
    try:
        # 尝试获取正在运行的事件循环（某些 IDE/Jupyter 会预先启动）
        loop = asyncio.get_running_loop()
    except RuntimeError:
        loop = None

    # 如果事件循环存在且正在运行（比如在 Jupyter Notebook 中）
    if loop and loop.is_running():
        print("检测到事件循环正在运行，使用 create_task 启动协程")
        asyncio.create_task(main())  # 使用 create_task 异步运行
    else:
        # 否则，正常使用 asyncio.run 启动主协程
        asyncio.run(main())