selenium_action/wechat_action/send_ad_idea.py

import time
import random
from selenium.webdriver.support.wait import WebDriverWait


class IdeaAction:
    def __init__(self, driver):
        self.driver = driver

    def push_page(self):
        self.driver.find_element_by_class_name('adui-button-small').click()
        select_input = self.driver.find_element_by_class_name('adui-input-base')
        select_input.send_keys('原生')
        self.driver.find_element_by_id('select_canvas_ok_btn').click()
        time.sleep(random.uniform(1, 2))

    def idea_banner_gzh_bottom(self):
        self.driver.find_element_by_css_selector('#test_material_type_135 > span.inputWrapper-3fBModc9zM > i').click()
        wen_input = self.driver.find_element_by_class_name('input-3MbKvywHL2')
        wen_input.send_keys('nihaonihao')

    def idea_pic_gzh_bottom(self):
        # TODO:添加try,catch因为图片上传可能慢
        WebDriverWait(self.driver, 10).until(lambda driver: driver.find_element_by_class_name('adui-button-small'))

        self.driver.find_element_by_class_name("webuploader-element-invisible").send_keys(
            "/usr/share/wallpapers/deepin/Beach_by_Samuel_Scrimshaw.jpg")
        WebDriverWait(self.driver, 5).until(lambda driver: driver.find_element_by_class_name('btnFist-uueBS6DQFa'))
        self.driver.find_element_by_class_name('btnFist-uueBS6DQFa').click()
        WebDriverWait(self.driver, 10).until(lambda driver: driver.find_element_by_id('test_progress_confirm'))
        # 等待上传完毕,因为完成按钮本身可以
        time.sleep(10)
        self.driver.find_element_by_id('test_progress_confirm').click()
        time.sleep(random.uniform(1, 2))

        self.driver.save_screenshot('set_page_pic.png')
        self.push_page()
        self.driver.save_screenshot('set_page_idea.png')

    def idea_movie_gzh_movie(self):
        WebDriverWait(self.driver, 10).until(
            lambda driver: driver.find_element_by_class_name('webuploader-element-invisible'))

        self.driver.find_element_by_class_name('webuploader-element-invisible').send_keys(
            '/home/cxyu/Downloads/上门龙婿-竖3(1).mp4')
        suc_button = self.driver.find_element_by_id('test_progress_confirm')
        WebDriverWait(self.driver, 1000).until(lambda  x:(suc_button.is_displayed() and suc_button.is_enabled()))

        self.driver.find_element_by_id('test_progress_confirm').click()

        self.driver.save_screenshot('set_page_pic.png')
        self.push_page()
        text_input = self.driver.find_element_by_css_selector('#reward_video_desc > div > input')
        text_input.send_keys('asdf')