from DrissionPage import ChromiumPage

# 初始化瀏覽器
page = ChromiumPage()
page.get('https://example.com/upload')  # 替換為實(shí)際上傳頁面

# 直接定位file input并上傳
file_input = page.e('input[type="file"]')  # CSS選擇器
file_input.send_keys(r'C:\test\sample.jpg')  # 發(fā)送文件路徑

def upload_file(page, url, file_path):
    """
    通用文件上傳函數(shù)
    :param page: DrissionPage對(duì)象
    :param url: 上傳頁面URL
    :param file_path: 本地文件路徑
    :return: 是否上傳成功
    """
    try:
        page.get(url)
        # 等待上傳元素加載（顯式等待更可靠）
        upload_input = page.wait('@input[type="file"]', timeout=10)
        upload_input.send_keys(file_path)
        return True
    except Exception as e:
        print(f"上傳失敗: {str(e)}")
        return False

import os
import re
import time
import random
from DrissionPage import ChromiumPage

# 配置參數(shù)
TARGET_DIR = r'G:\驗(yàn)證碼項(xiàng)目\抖音九宮格語義點(diǎn)選驗(yàn)證碼\圖片\temp改名文件夾/'
SOURCE_DIR = r'G:\驗(yàn)證碼項(xiàng)目\抖音九宮格語義點(diǎn)選驗(yàn)證碼\圖片\3裁剪后的圖片\'
UPLOAD_URL = 'https://chat.baidu.com/search'

def process_images():
    # 初始化瀏覽器
    page = ChromiumPage()
    
    # 確保目標(biāo)目錄存在
    os.makedirs(TARGET_DIR, exist_ok=True)
    
    # 獲取所有圖片
    image_files = [f for f in os.listdir(SOURCE_DIR) if f.lower().endswith(('.jpg', '.png'))]
    
    for img_name in image_files:
        img_path = os.path.join(SOURCE_DIR, img_name)
        
        try:
            # 上傳流程
            page.get(UPLOAD_URL)
            
            # 點(diǎn)擊上傳按鈕（根據(jù)實(shí)際頁面結(jié)構(gòu)調(diào)整）
            upload_btn = page.ele('xpath://div[contains(@class,"upload-btn")]')
            upload_btn.click()
            
            # 操作隱藏的file input
            file_input = page.ele('input[type="file"]')
            file_input.send_keys(img_path)
            
            # 填寫表單（示例）
            textarea = page.ele('textarea[id="chat-textarea"]')
            textarea.input("請(qǐng)識(shí)別圖片中的內(nèi)容，用【】標(biāo)注結(jié)果")
            
            # 提交
            submit_btn = page.ele('img[class*="submit-button"]')
            submit_btn.click()
            
            # 等待結(jié)果（根據(jù)實(shí)際頁面調(diào)整等待條件）
            time.sleep(5)
            
            # 提取識(shí)別結(jié)果
            result_div = page.ele('div[class*="marklang"]')
            result_text = result_div.text
            print(f"識(shí)別結(jié)果: {result_text}")
            
            # 解析結(jié)果
            matches = re.findall(r'【(.*?)】', result_text)
            if matches:
                # 生成新文件名
                new_name = f"{matches[0]}_{int(time.time()*1000)}_{random.randint(1000,9999)}.jpg"
                new_path = os.path.join(TARGET_DIR, new_name)
                
                # 移動(dòng)文件（先刪除已存在的同名文件）
                if os.path.exists(new_path):
                    os.remove(new_path)
                os.rename(img_path, new_path)
                print(f"文件已重命名為: {new_name}")
                
        except Exception as e:
            print(f"處理圖片 {img_name} 時(shí)出錯(cuò): {str(e)}")
            continue

if __name__ == '__main__':
    process_images()

# 方法1：先點(diǎn)擊觸發(fā)按鈕再操作input
trigger_btn = page.ele('button[data-action="upload"]')
trigger_btn.click()
file_input = page.ele('input[type="file"]')  # 現(xiàn)在應(yīng)該可見
file_input.send_keys(file_path)

# 方法2：直接執(zhí)行JS觸發(fā)點(diǎn)擊
page.run_js('document.querySelector("input[type=file]").click()')

# 支持多選的上傳（需頁面允許）
file_input = page.ele('input[type="file"][multiple]')
file_input.send_keys([
    r'C:\test\file1.jpg',
    r'C:\test\file2.png'
])

# 更可靠的等待方式
from DrissionPage.common import Wait

# 等待元素出現(xiàn)（最多10秒）
upload_input = Wait(page).until(lambda p: p.ele('input[type="file"]'))

# 或者使用顯式等待
page.wait('@input[type="file"]', timeout=10)