From e09fb064771f60f9b7ebf1bc9c47a78187b1e2c2 Mon Sep 17 00:00:00 2001 From: "pookz@stme" Date: Wed, 14 Aug 2024 10:44:17 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E6=8A=96=E9=9F=B3=E5=9B=A0=E4=B8=BA?= =?UTF-8?q?=E8=AE=BE=E7=BD=AE=E5=9C=B0=E7=90=86=E4=BD=8D=E7=BD=AE=E5=8F=AF?= =?UTF-8?q?=E8=83=BD=E9=80=A0=E6=88=90=E7=9A=84=E9=98=BB=E5=A1=9E?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- uploader/douyin_uploader/main.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/uploader/douyin_uploader/main.py b/uploader/douyin_uploader/main.py index cccee41..26efe96 100644 --- a/uploader/douyin_uploader/main.py +++ b/uploader/douyin_uploader/main.py @@ -1,7 +1,7 @@ # -*- coding: utf-8 -*- from datetime import datetime -from playwright.async_api import Playwright, async_playwright +from playwright.async_api import Playwright, async_playwright, Page import os import asyncio @@ -160,15 +160,7 @@ class DouYinVideo(object): await asyncio.sleep(2) # 更换可见元素 - await page.locator('div.semi-select span:has-text("输入地理位置")').click() - await page.keyboard.press("Backspace") - await page.wait_for_timeout(2000) - # await page.keyboard.press("Control+KeyA") - # await page.keyboard.press("Delete") - await page.keyboard.type("杭州市") - # await asyncio.sleep(1) - await page.wait_for_timeout(1000) - await page.locator('div[role="listbox"] [role="option"]').first.click() + await self.set_location(page, "杭州市") # 頭條/西瓜 third_part_element = '[class^="info"] > [class^="first-part"] div div.semi-switch' @@ -204,6 +196,14 @@ class DouYinVideo(object): await context.close() await browser.close() + async def set_location(self, page: Page, location: str = "杭州市"): + await page.locator('div.semi-select span:has-text("输入地理位置")').click() + await page.keyboard.press("Backspace") + await page.wait_for_timeout(2000) + await page.keyboard.type(location) + await page.wait_for_selector('div[role="listbox"] [role="option"]', timeout=5000) + await page.locator('div[role="listbox"] [role="option"]').first.click() + async def main(self): async with async_playwright() as playwright: await self.upload(playwright)