selenium_action/wechat_action/create_ad_layout.py

from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.common.keys import Keys
from selenium.webdriver import ActionChains
from logging import handlers
from datetime import datetime
import logging
import requests
import random
import shutil
import time
import re
import os


class CreateAd:
    def __init__(self, login_ad, service_name, wechat_name):
        self.log_ad = login_ad
        self.service_name = service_name
        self.wechat_name = wechat_name
        self.driver = login_ad.get_driver_loged()
        self.get_into_create_page()
        self.send_file_limit_num = 8
        self.img_dir = './img'

    def get_into_create_page(self):
        # 进入创建页面
        logging.info('开始进入创建页面')
        self.driver.find_element_by_id('material').click()
        for i in range(10):
            try:
                WebDriverWait(self.driver, 10).until(lambda driver: self.driver.find_element_by_xpath(
                    '//*[@class="ui-fl-r adui-button-base adui-button-primary adui-button-small"]'))
                create_element = self.driver.find_element_by_xpath(
                    '//*[@class="ui-fl-r adui-button-base adui-button-primary adui-button-small"]')
                WebDriverWait(self.driver, 10).until(
                    lambda driver: create_element.is_displayed() and create_element.is_enabled())
                create_element.click()
                break
            except Exception as e:
                logging.info(str(e))
                pass
            if i == 10:
                raise ValueError('点击新建推广页 出错')
        logging.info('点击新建推广页 结束')
        WebDriverWait(self.driver, 5).until(lambda driver: driver.find_element_by_css_selector(
            '#wxadcontainer > div:nth-child(1) > div:nth-child(2) > div.dialog-1fj_N480ZT > div > div > div:nth-child(1) > div.dialogCardFooter-17KpBD1lgN > button'))
        self.driver.find_element_by_css_selector(
            '#wxadcontainer > div:nth-child(1) > div:nth-child(2) > div.dialog-1fj_N480ZT > div > div > div:nth-child(1) > div.dialogCardFooter-17KpBD1lgN > button').click()
        self.driver.switch_to.window(self.driver.window_handles[-1])
        WebDriverWait(self.driver, 100).until(lambda driver: driver.find_element_by_class_name('addContent-8pexaaAGYy'))
        self.driver.find_element_by_class_name('addContent-8pexaaAGYy').click()
        WebDriverWait(self.driver, 100).until(lambda driver: driver.find_element_by_class_name('topArea-qOwEAeNuIn'))
        logging.info('进入到推广页,编辑界面')

    def set_advertisement_sign(self, layout_name):
        # 设置广告标记
        self.driver.find_element_by_xpath('//*[@class="icon-edQB0KK2VG"]').click()
        input_element = self.driver.find_element_by_xpath('//input[@class="input-2lFnByGCRh"]')
        input_element.send_keys(Keys.BACKSPACE)
        input_element.send_keys(layout_name)

    def set_background_color(self, color):
        color_buttons = self.driver.find_elements_by_class_name('adui-cp-picker')
        c_buttons_can_use = []
        for _ in color_buttons:
            if _.is_displayed() and _.is_enabled():
                c_buttons_can_use.append(_)
        c_buttons_can_use[0].click()
        time.sleep(random.uniform(0.2, 0.3))
        input_elements = self.driver.find_elements_by_xpath('//input[@class="adui-input-base"]')
        for _ in input_elements:
            if _.is_enabled() and _.is_displayed() and _.get_attribute("value") == 'FFFFFF':
                _.click()
                _.send_keys(color)

    def set_head_assemb(self, info):
        logging.info('开始设置头版首页')

        def single_page_set():
            self.driver.find_element_by_xpath('//*[@id="stage-sidebar"]/div[1]/div/div[1]/div/div').click()
            time.sleep(random.uniform(0.1, 0.2))

            # 设置图片格式
            if info['content']['adSite'] != '朋友圈信息流':
                self.driver.find_elements_by_xpath('//*[@class="adui-radio-indicator"]')[1].click()
            elif info['content']['outerStyle'] != '常规广告':
                select_e = self.driver.find_elements_by_xpath('//*[@class="adui-select-selection-item"]')
                for _ in select_e:
                    if _.text == '常规广告':
                        _.click()
                time.sleep(0.1)
                select_e = self.driver.find_elements_by_xpath('//*[@class="adui-select-item-option-content"]')
                for _ in select_e:
                    if _.text == '卡片广告':
                        _.click()

            # 上传图片
            file_name = re.split('\/', info['content']['url'])[-1]
            self.driver.find_element_by_class_name('upload-img-item-inner-2gsg7NjaZ8').click()
            WebDriverWait(self.driver, 10).until(
                lambda x: len([_ for _ in self.driver.find_elements_by_xpath(
                    '//*[@class="base-1H7btNX4v9 primary-2diSJcTI_7 small-ZbOBQ_8kVy iconBtn-NYPRTZ5yJB webuploader-container"]')
                               if _.is_displayed() and _.is_enabled()]) > 0)

            turn_page_buttons = self.driver.find_elements_by_class_name('paginationIcon-1EfoH0sNRF')
            can_use_button = []
            for _ in turn_page_buttons:
                if _.is_enabled() and _.is_displayed():
                    can_use_button.append(_)
            # 不断翻页获取到元素为止
            chose_over = False
            while True:
                WebDriverWait(self.driver, 10).until(
                    lambda driver: self.driver.find_elements_by_class_name('title-29sncpKgTl'))
                #TODO:翻页后,有时没有加载出来
                page_elements = self.driver.find_elements_by_class_name('title-29sncpKgTl')

                for _ in page_elements:
                    if _.is_displayed() and _.is_enabled():
                        if file_name in _.text:
                            _.click()
                            chose_over = True
                            break
                if chose_over or len(can_use_button) == 0:
                    break
                # 翻到最后一页时停止
                page_text = self.driver.find_element_by_class_name('count_small-37CcvfzoTl').text
                page_nums = re.findall('\d+', page_text)
                if int(page_nums[0].strip()) == int(page_nums[1].strip()):
                    break
                # 翻页
                can_use_button[2].click()
                # 确保翻页成功
                WebDriverWait(self.driver, 10).until(
                    lambda x: page_text != self.driver.find_element_by_class_name('count_small-37CcvfzoTl').text)

            self.driver.find_element_by_xpath('//*[@id="test_material_container_confirm"]').click()
            try:
                WebDriverWait(self.driver, 4).until(
                    lambda driver: driver.find_element_by_class_name(
                        'btnFist-uueBS6DQFa'))
                _ = self.driver.find_element_by_class_name(
                    'btnFist-uueBS6DQFa')
                WebDriverWait(self.driver, 4).until(
                    lambda x: (_.is_displayed() and _.is_enabled()))
                time.sleep(1)
                _.click()

                WebDriverWait(self.driver, 100).until(
                    lambda driver: driver.find_element_by_class_name(
                        'btn-3E823IXt3m'))
                _ = self.driver.find_element_by_class_name(
                    'btn-3E823IXt3m')
                WebDriverWait(self.driver, 100).until(
                    lambda x: (_.is_displayed() and _.is_enabled()))
                time.sleep(1)
                _.click()


            except Exception as e:
                pass
            logging.info('头版首页单图上传结束')

        def multi_page_set():
            # TODO:现在图片设置为素材库,需要对应下载然后存储
            self.driver.find_element_by_css_selector(
                '#stage-sidebar > div.topArea-qOwEAeNuIn > div > div:nth-child(2)').click()
            time.sleep(random.uniform(0.1, 0.2))
            page_size = len(info_v)
            if page_size == 3:
                pass
            if page_size == 4:
                self.driver.find_element_by_css_selector(
                    '#stage-settings > div:nth-child(1) > div > div > div:nth-child(3) > div.adui-form-item > div.adui-form-control > div > button:nth-child(2)').click()

            if page_size == 6:
                self.driver.find_element_by_css_selector(
                    '#stage-settings > div:nth-child(1) > div > div > div:nth-child(3) > div.adui-form-item > div.adui-form-control > div > button:nth-child(3)').click()

            time.sleep(random.uniform(0.1, 0.2))
            self.driver.find_element_by_class_name('imageUploadItem-tA9JX0RWua').click()
            WebDriverWait(self.driver, 1000).until(
                lambda driver: self.driver.find_element_by_class_name('adui-tabs-tab')
            )
            input_elements = self.driver.find_elements_by_tag_name('input')
            input_find_element = None
            for _ in input_elements:
                if '输入关键词搜索素材' in _.get_attribute('placeholder'):
                    input_find_element = _

            for _ in info_v:
                file_name = os.path.basename(_)
                logging.info(file_name)
                input_find_element.send_keys(file_name)
                input_find_element.send_keys(Keys.RETURN)
                WebDriverWait(self.driver, 1000).until(
                    lambda driver: driver.find_element_by_class_name('img-2HvhMmpnzP'))
                file_element = self.driver.find_element_by_class_name('img-2HvhMmpnzP')
                WebDriverWait(self.driver, 1000).until(
                    lambda x: (file_element.is_displayed() and file_element.is_enabled()))
                ActionChains(self.driver).move_to_element(file_element).perform()
                time.sleep(random.uniform(0.5, 1))
                file_element.click()

                for i in range(len(file_name) + 10):
                    input_find_element.send_keys(Keys.BACKSPACE)
                time.sleep(random.uniform(0.5, 1))
            self.driver.find_element_by_xpath('/html/body/div[12]/div/div/div[2]/div/div[3]/button[2]').click()

            # 切图操作如果有的话进行对应操作.
            for i in range(len(info_v) + 1):
                try:
                    WebDriverWait(self.driver, 6).until(
                        lambda driver: driver.find_element_by_xpath(
                            '/html/body/div[13]/div/div/div[2]/div/div[3]/button[2]'))
                    self.driver.find_element_by_xpath('/html/body/div[13]/div/div/div[2]/div/div[3]/button[2]').click()
                except:
                    pass

            logging.info('头版多图选择 结束')

        def movie_set():
            # TODO:视频转为链接,
            file_name = os.path.basename(info_v)
            self.driver.find_element_by_xpath('//*[@id="stage-sidebar"]/div[1]/div/div[3]/div/div').click()
            self.driver.find_element_by_xpath('//*[@class="comptEditButton-2JsnAFdOGZ"]').click()

            WebDriverWait(self.driver, 100).until(
                lambda x: len([_ for _ in self.driver.find_elements_by_class_name('title-29sncpKgTl') if
                               _.is_displayed() and _.is_enabled()]) > 0)
            #
            turn_page_buttons = self.driver.find_elements_by_class_name('paginationIcon-1EfoH0sNRF')
            can_use_button = []
            for _ in turn_page_buttons:
                if _.is_enabled() and _.is_displayed():
                    can_use_button.append(_)
            # 不断翻页获取到元素为止
            chose_over = False
            while True:
                page_elements = self.driver.find_elements_by_class_name('title-29sncpKgTl')
                for _ in page_elements:
                    if _.is_displayed() and _.is_enabled():
                        if file_name in _.text:
                            _.click()
                            chose_over = True
                            break
                if chose_over or len(can_use_button) == 0:
                    # can_use_button 数量说明是否可以翻页,如果不可以翻页,则直接停止
                    break
                # 翻到最后一页时停止
                page_text_elements = self.driver.find_elements_by_class_name('count_small-37CcvfzoTl')
                for _ in page_text_elements:
                    if _.is_enabled() and _.is_displayed():
                        page_text = _.text
                        page_text_element = _

                page_nums = re.findall('\d+', page_text)
                if int(page_nums[0].strip()) == int(page_nums[1].strip()):
                    break
                # 翻页
                can_use_button[2].click()
                # 确保翻页成功
                WebDriverWait(self.driver, 10).until(
                    lambda x: page_text != page_text_element.text)
            sc_buttons = self.driver.find_elements_by_id('test_material_container_confirm')
            for _ in sc_buttons:
                if _.is_enabled() and _.is_displayed():
                    _.click()

        info_type = info['type']
        info_v = None  # TODO: 暂时存在,之后修改好multi_page_set 和movie_set.就删除

        if info['type'] == 'topImg':
            single_page_set()
        if info['type'] == 'topSlider':
            multi_page_set()
        if info['type'] == 'topVideo':
            movie_set()

        # 设置头部组件
        pass

    def set_page(self, config_info):
        logging.info('开始设置图片模块')
        page_link = config_info['url']
        # 设置图片模块
        self.driver.find_element_by_xpath('//*[@id="stage-sidebar"]/section/div[2]/div[1]/div/div').click()
        time.sleep(random.uniform(0.1, 0.2))

        file_name = re.split('\/', page_link)[-1]

        upload_elements = self.driver.find_elements_by_class_name('upload-img-item-inner-2gsg7NjaZ8')
        for _ in upload_elements:
            if _.is_displayed() and _.is_enabled():
                _.click()
        WebDriverWait(self.driver, 10).until(
            lambda x: len([_ for _ in self.driver.find_elements_by_xpath(
                '//*[@class="base-1H7btNX4v9 primary-2diSJcTI_7 small-ZbOBQ_8kVy iconBtn-NYPRTZ5yJB webuploader-container"]')
                           if _.is_displayed() and _.is_enabled()]) > 0)

        # 不断翻页获取到元素为止
        chose_over = False
        logging.info('翻页开始')
        while True:
            WebDriverWait(self.driver, 10).until(
                lambda driver: self.driver.find_elements_by_class_name('title-29sncpKgTl'))
            # TODO:翻页后,有时没有加载出来
            page_elements = self.driver.find_elements_by_class_name('title-29sncpKgTl')
            for _ in page_elements:
                if _.is_displayed() and _.is_enabled():
                    if file_name in _.text:
                        _.click()
                        chose_over = True
                        break
            turn_page_buttons = self.driver.find_elements_by_class_name('paginationIcon-1EfoH0sNRF')
            can_use_button = []
            for _ in turn_page_buttons:
                if _.is_enabled() and _.is_displayed():
                    can_use_button.append(_)
            if chose_over or len(can_use_button) == 0:
                # can_use_button 数量说明是否可以翻页,如果不可以翻页,则直接停止
                break
            # 翻到最后一页时停止
            page_text_elements = self.driver.find_elements_by_class_name('count_small-37CcvfzoTl')
            for _ in page_text_elements:
                if _.is_enabled() and _.is_displayed():
                    page_text = _.text
                    page_text_element = _

            page_nums = re.findall('\d+', page_text)
            if int(page_nums[0].strip()) == int(page_nums[1].strip()):
                break
            # 翻页
            can_use_button[2].click()
            # 确保翻页成功
            WebDriverWait(self.driver, 10).until(
                lambda x: page_text != page_text_element.text)
        logging.info('翻页结束')
        sc_buttons = self.driver.find_elements_by_id('test_material_container_confirm')
        for _ in sc_buttons:
            if _.is_enabled() and _.is_displayed():
                _.click()
        logging.info('是否截图开始')
        try:
            WebDriverWait(self.driver, 4).until(
                lambda driver: driver.find_element_by_class_name(
                    'btnFist-uueBS6DQFa'))
            _ = self.driver.find_element_by_class_name(
                'btnFist-uueBS6DQFa')
            WebDriverWait(self.driver, 4).until(
                lambda x: (_.is_displayed() and _.is_enabled()))
            # 已经出现但任需要等待
            time.sleep(1)
            _.click()

            WebDriverWait(self.driver, 100).until(
                lambda x: len([_ for _ in self.driver.find_elements_by_class_name('btn-3E823IXt3m') if
                               _.is_displayed() and _.is_enabled()]) > 0)
            for _ in self.driver.find_elements_by_class_name('btn-3E823IXt3m'):
                if _.is_displayed() and _.is_enabled():
                    _.click()
        except Exception as e:
            logging.info(e)
        logging.info('是否截图结束')
        # 设置文本边距
        if config_info['marginTop'] != 0 or config_info['marginBottom'] != 0:
            distance_buttons = self.driver.find_elements_by_xpath(
                '//div[@class="adui-numeric-input adui-input-wrapper adui-input-small adui-input-normal"]/input')
            distance_buttons_can_use = []
            for _ in distance_buttons:
                if _.is_enabled() and _.is_displayed() and _.get_attribute("value") == '0':
                    distance_buttons_can_use.append(_)
            distance_buttons_can_use[0].click()
            for i in range(4):
                distance_buttons_can_use[0].send_keys(Keys.BACKSPACE)
            distance_buttons_can_use[0].send_keys(config_info['marginTop'])
            distance_buttons_can_use[1].click()
            for i in range(4):
                distance_buttons_can_use[1].send_keys(Keys.BACKSPACE)
            distance_buttons_can_use[1].send_keys(config_info['marginBottom'])
        logging.info('图片模块设置结束')

    def set_content(self, config_info):
        logging.info(' 开始设置文本模块')
        # 设置文本模块
        # 1.文本内容2.文本颜色3.是否加粗4.文本位置设置5.文本大小

        # 设置文本内容
        self.driver.find_element_by_xpath('//*[@id="stage-sidebar"]/section/div[2]/div[5]/div/div').click()
        time.sleep(random.uniform(0.2, 0.3))
        input_elements = self.driver.find_elements_by_xpath('//textarea[@class="adui-input-base"]')
        # pyperclip.copy(config_info['text'])
        js_base = '''
        function copyToClipboard(text) {
            var dummy = document.createElement("textarea");
            // to avoid breaking orgain page when copying more words
            // cant copy when adding below this code
            // dummy.style.display = 'none'
            document.body.appendChild(dummy);
            //Be careful if you use texarea. setAttribute('value', value), which works with "input" does not work with "textarea". – Eduard
            dummy.value = text;
            dummy.select();
            document.execCommand("copy");
            document.body.removeChild(dummy);
        }
        '''
        content_change = config_info['text']

        content_change = content_change.replace("\\", "\\\\")
        content_change = content_change.replace('\n', '\\n')
        content_change = content_change.replace('\n', '\\n')
        content_change = content_change.replace("'", "\'")

        js_base = js_base + '''copyToClipboard('{}');'''.format(content_change)
        self.driver.execute_script(js_base)
        for _ in input_elements:
            if _.is_enabled() and _.is_displayed():
                _.send_keys(Keys.CONTROL, 'v')
        logging.info('文本内容设置完毕')
        # 设置颜色
        if config_info['color'] != '#595959':
            color_buttons = self.driver.find_elements_by_class_name('adui-cp-picker')
            c_buttons_can_use = []
            for _ in color_buttons:
                if _.is_displayed() and _.is_enabled():
                    c_buttons_can_use.append(_)
            c_buttons_can_use[1].click()
            time.sleep(random.uniform(0.2, 0.3))

            input_elements = self.driver.find_elements_by_xpath(
                '//div[@class="adui-cp-input adui-input-wrapper adui-input-small adui-input-light adui-input-normal"]/input')
            for _ in input_elements:
                if _.is_enabled() and _.is_displayed() and _.get_attribute("value") == '595959':
                    _.click()
                    _.send_keys(config_info['color'])

        # 设置是否加粗
        if config_info['fontWeight'] != 'normal':
            big_elements = self.driver.find_elements_by_xpath(
                '//button[@class="adui-button-base adui-button-normal adui-button-mini"]/span')
            for _ in big_elements:
                if _.text == '加粗' and _.is_displayed() and _.is_enabled():
                    _.click()

        # 设置文本位置
        if config_info['textAlign'] != 'left':
            loc_buttons = self.driver.find_elements_by_xpath(
                '//button[@class="adui-button-base adui-button-normal adui-button-mini adui-button-hasLeftIcon adui-button-hasRightIcon"]')
            loc_buttons_can_use = []
            for _ in loc_buttons:
                if _.is_enabled() and _.is_displayed():
                    loc_buttons_can_use.append(_)
            if config_info['textAlign'] == 'center':
                loc_buttons_can_use[0].click()
            if config_info['textAlign'] == 'right':
                loc_buttons_can_use[1].click()

        # 设置文本大小
        if config_info['fontSize'] != 15:
            str_buttons = self.driver.find_elements_by_xpath("//*[@class='adui-select-selection-search']")
            str_buttons_can_use = []
            for _ in str_buttons:
                if _.is_displayed() and _.is_enabled():
                    str_buttons_can_use.append(_)
            str_buttons_can_use[1].click()
            time.sleep(random.uniform(0.5, 1))

            str_num_buttons = self.driver.find_elements_by_xpath("//*[@class='adui-select-item-option-content']")
            str_num_can_use = []
            for _ in str_num_buttons:
                if _.is_displayed() and _.is_enabled():
                    str_num_can_use.append(_)
            # str_num_map = {14: 0, 15: 1, 16: 2, 18: 3, 20: 4, 24: 5, 36: 6}
            # str_num_can_use[str_num_map[config_info['fontSize']]].click()

            for _ in str_num_can_use:
                if _.text == str(config_info['fontSize']):
                    _.click()

        # 设置文本边距
        if config_info['marginTop'] != 22 or config_info['marginBottom'] != 22:
            distance_buttons = self.driver.find_elements_by_xpath(
                '//div[@class="adui-numeric-input adui-input-wrapper adui-input-small adui-input-normal"]/input')
            distance_buttons_can_use = []
            for _ in distance_buttons:
                if _.is_enabled() and _.is_displayed() and _.get_attribute("value") == '22':
                    distance_buttons_can_use.append(_)
            distance_buttons_can_use[0].click()
            for i in range(4):
                distance_buttons_can_use[0].send_keys(Keys.BACKSPACE)
            distance_buttons_can_use[0].send_keys(config_info['marginTop'])
            distance_buttons_can_use[1].click()
            for i in range(4):
                distance_buttons_can_use[1].send_keys(Keys.BACKSPACE)
            distance_buttons_can_use[1].send_keys(config_info['marginBottom'])

    def set_follow_button(self, config_info):
        logging.info('设置一键关注')
        # 设置关注按钮
        # 0.是否一键关注 1.设置button文本内容 2.是否加粗 3.设置字体颜色,边框颜色,填充色 4.设置边距
        self.driver.find_element_by_xpath('//*[@id="stage-sidebar"]/section/div[4]/div[2]/div/div').click()

        # 开始设置一键关注
        if 'follow' in config_info.keys():
            if not config_info['follow']:
                time.sleep(0.1)
                follow_bs = self.driver.find_elements_by_xpath(
                    '//*[@class="form-caption-3Xp3o6Drwf"]/div/span')

                for _ in follow_bs:
                    if _.is_displayed() and _.is_enabled():
                        _.click()

        # 文本设置
        input_elements = self.driver.find_elements_by_xpath(
            '//div[@class="adui-input-wrapper adui-input-mini adui-input-limited adui-input-normal"]/input')
        for _ in input_elements:
            if _.is_enabled() and _.is_displayed() and _.get_attribute("value") == '关注公众号':
                _.click()
                # for i in range(10):
                #     _.send_keys(Keys.BACKSPACE)
                _.send_keys(Keys.CONTROL + 'a')
                _.send_keys(Keys.BACKSPACE)
                _.send_keys(config_info['text'])

        # 文本是否加粗
        if config_info['fontWeight'] != 'normal':
            big_elements = self.driver.find_elements_by_xpath(
                '//button[@class="adui-button-base adui-button-normal adui-button-mini"]/span')
            for _ in big_elements:
                if _.text == '加粗' and _.is_displayed() and _.is_enabled():
                    _.click()

        # 颜色设置
        input_elements = self.driver.find_elements_by_xpath(
            '//div[@class="adui-cp-input adui-input-wrapper adui-input-mini adui-input-normal"]/input')
        input_elements_can_use = []
        for _ in input_elements:
            if _.is_enabled() and _.is_displayed() and (
                    _.get_attribute("value") == '07C160' or _.get_attribute("value") == 'FFFFFF'):
                input_elements_can_use.append(_)

        if config_info['textColor'] != '#FFFFFF':
            input_elements_can_use[0].click()
            input_elements_can_use[0].send_keys(config_info['textColor'])
            time.sleep(random.uniform(0.1, 0.5))
        if config_info['borderColor'] != '#FFFFFF':
            input_elements_can_use[1].click()
            input_elements_can_use[1].send_keys(config_info['borderColor'])
            time.sleep(random.uniform(0.1, 0.5))
        input_elements_can_use[2].click()
        input_elements_can_use[2].send_keys(config_info['backColor'])
        time.sleep(random.uniform(0.1, 0.5))

        # 间隙设置
        size_buttons = self.driver.find_elements_by_xpath(
            '//div[@class="adui-numeric-input adui-input-wrapper adui-input-small adui-input-normal"]/input')
        size_buttons_can_use = []
        for _ in size_buttons:
            if _.is_enabled() and _.is_displayed() and _.get_attribute("value") == '28':
                size_buttons_can_use.append(_)
        size_buttons_can_use[0].click()
        for i in range(4):
            size_buttons_can_use[0].send_keys(Keys.BACKSPACE)
        size_buttons_can_use[0].send_keys(config_info['marginTop'])
        size_buttons_can_use[1].click()
        for i in range(4):
            size_buttons_can_use[1].send_keys(Keys.BACKSPACE)
        size_buttons_can_use[1].send_keys(config_info['marginBottom'])
        logging.info('设置一键关注,结束')

    def set_text_button(self):
        # 设置图文按钮
        pass

    def send_file_multi(self, layout, err_num=0):
        # 有问题,暂时不使用
        # 上传文件,文件上传与整体流程切割开
        token = re.findall('token=(\d+)', self.driver.current_url)
        url_ = 'https://mp.weixin.qq.com/promotion/frame?t=ad_system/common_frame&t1=material_std/material_library&token={}'.format(
            token[0])
        js = "window.open('{}')".format(url_)
        self.driver.execute_script(js)
        time.sleep(random.uniform(1, 2))
        self.driver.switch_to.window(self.driver.window_handles[-1])
        file_set = set()
        for v in layout.values():
            if isinstance(v, dict):
                if 'multi_page' in v.keys() or 'page' in v.keys() or 'movie' in v.keys():
                    v_name = list(v.keys())[0]
                    if isinstance(v[v_name], list):
                        for _ in v[v_name]:
                            file_set.add(_)
                    else:
                        file_set.add(v[v_name])
        file_list = list(file_set)
        # print(file_list)
        # print('file list size', len(file_list))
        send_file_sign = self.send_file_limit_num
        for _ in file_list:
            # print(_)
            self.driver.find_element_by_xpath('//*[@id="wxadcontainer"]/div[1]/div[2]/div[4]/input').send_keys(_)

        time.sleep(100)
        while True:
            # 失败,失败直接退出重跑
            # 正在上传数量,如果是正在上传小于限定数量,就不断输入
            # 上传完毕,小于限定数量,不断输入,.....已经上传好了,就关闭
            time.sleep(1)
            send_reslut_content = self.driver.find_element_by_xpath(
                '//*[@id="wxadcontainer"]/div[1]/div[2]/div[4]/div/strong').text
            if '失败' in send_reslut_content:
                if err_num < 3:
                    self.send_file(layout, err_num=err_num + 1)
                else:
                    exit()
            if '正在上传' in send_reslut_content:
                if send_file_sign < len(file_list):
                    send_num = re.findall('(\d+)', send_reslut_content)[0]
                    if int(send_num) < self.send_file_limit_num:
                        # 还可以继续上传
                        x = send_file_sign
                        can_send_sign = send_file_sign + (self.send_file_limit_num - send_num)
                        y = can_send_sign if can_send_sign < len(file_list) else len(file_list)
                        for i in range(x, y):
                            send_file_sign = send_file_sign + 1
                            self.driver.find_element_by_xpath(
                                '//*[@id="wxadcontainer"]/div[1]/div[2]/div[4]/input').send_keys(file_list[i])

                else:
                    pass
            if '上传成功' in send_reslut_content:
                if send_file_sign < len(file_list):
                    send_num = re.findall('(\d+)', send_reslut_content)[0]
                    if int(send_num) < self.send_file_limit_num:
                        # 还可以继续上传
                        x = send_file_sign
                        can_send_sign = send_file_sign + (self.send_file_limit_num - send_num)
                        y = can_send_sign if can_send_sign < len(file_list) else len(file_list)
                        for i in range(x, y):
                            send_file_sign = send_file_sign + 1
                            self.driver.find_element_by_xpath(
                                '//*[@id="wxadcontainer"]/div[1]/div[2]/div[4]/input').send_keys(file_list[i])
                else:
                    break

        time.sleep(1000)
        self.driver.execute_script('window.close();')

    def send_file(self, file_link, err_num=0):
        # 下载图片,存储于
        try:
            if not os.path.exists(self.img_dir):
                os.makedirs(self.img_dir)
            link_pra = re.split('\/', file_link)
            file_path = os.getcwd() + '/' + self.img_dir + '/' + link_pra[-1]
            rsp = requests.get(file_link, timeout=0.5)
            with open(file_path, 'wb') as f:
                f.write(rsp.content)
            # 有问题,暂时不使用
            self.driver.execute_script('location.reload();')
            self.driver.find_element_by_xpath('//*[@id="wxadcontainer"]/div[1]/div[2]/div[4]/input').send_keys(
                file_path)
        except:
            if err_num < 20:
                self.send_file(file_link, err_num=err_num + 1)

    def send_file_alone(self, layout):
        def check_file(file_set, err_num=0):
            try:
                logging.info('start check file')
                token = re.findall('token=(\d+)', self.driver.current_url)
                token = str(token[0])
                cookies = self.driver.get_cookies()
                cookie_dict = {}
                for _ in cookies:
                    cookie_dict[_['name']] = _['value']

                wechat_url = 'https://mp.weixin.qq.com/promotion/material/material_library'
                headers = {
                    'content-type': 'application/x-www-form-urlencoded',
                    'referer': 'https://mp.weixin.qq.com/promotion/frame?t=ad_system/common_simple_frame&t1=rdcanvas/canvas_edit_v2&token={}'.format(
                        token),
                    'user-agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.182 Safari/537.36'}
                data = 'action=get_material_data&args={"condition_list":[{"min_width":0,"min_height":0}],"type":1,"page":1,"page_size":8}&token=' + token + '&appid=&spid=&_=' + str(
                    int(time.time() * 1000))
                rsp = requests.post(wechat_url, data=data, headers=headers, cookies=cookie_dict, timeout=1)
                total_page = rsp.json()['data']['page_info']['total_page']
                import copy
                tmp_file_set = copy.deepcopy(file_set)
                for file_url in tmp_file_set:
                    link_pra = re.split('\/', file_url)[-1]
                    for page_info in rsp.json()['data']['list']:
                        if link_pra == page_info['name']:
                            if file_url in file_set:
                                file_set.remove(file_url)

                for i in range(2, total_page + 1):
                    data = 'action=get_material_data&args={"condition_list":[{"min_width":0,"min_height":0}],"type":1,"page":' + str(
                        i) + ',"page_size":8}&token=' + token + '&appid=&spid=&_=' + str(
                        int(time.time() * 1000))
                    rsp = requests.post(wechat_url, data=data, headers=headers, cookies=cookie_dict, timeout=1)
                    tmp_file_set = copy.deepcopy(file_set)
                    for file_url in tmp_file_set:
                        link_pra = re.split('\/', file_url)[-1]
                        for page_info in rsp.json()['data']['list']:
                            if link_pra == page_info['name']:
                                file_set.remove(file_url)
            except Exception as e:
                if err_num < 10:
                    check_file(file_set, err_num + 1)

        def get_url(v_info):
            if 'url' in v_info.keys():
                file_set.add(v_info['url'])
            for v in v_info.values():
                if isinstance(v, dict):
                    get_url(v)
                if isinstance(v, list):
                    for _ in v:
                        if isinstance(_, dict):
                            get_url(_)

        # 上传文件,单线程
        # 上传文件,文件上传与整体流程切割开
        logging.info('开始传送文件')
        file_set = set()
        get_url(layout)
        check_file(file_set)
        if not file_set:
            logging.info('文件都已传送,无需重新上传')
            return

        WebDriverWait(self.driver, 10).until(
            lambda x: len(re.findall('token=(\d+)', self.driver.current_url)))
        token = re.findall('token=(\d+)', self.driver.current_url)
        url_ = 'https://mp.weixin.qq.com/promotion/frame?t=ad_system/common_frame&t1=material_std/material_library&token={}'.format(
            token[0])
        js = "window.open('{}')".format(url_)
        self.driver.execute_script(js)
        time.sleep(random.uniform(1, 2))
        self.driver.switch_to.window(self.driver.window_handles[-1])
        # TODO:之后还有multi_page 和 movie
        # for v in layout.values():
        #     if isinstance(v, dict):
        #         if 'multi_page' in v.keys() or 'page' in v.keys() or 'movie' in v.keys():
        #             v_name = list(v.keys())[0]
        #             if isinstance(v[v_name], list):
        #                 for _ in v[v_name]:
        #                     file_set.add(_)
        #             else:
        #                 file_set.add(v[v_name])

        for _ in file_set:
            self.send_file(_)
            err_num = 0
            while True:
                time.sleep(1)
                send_reslut_content = self.driver.find_element_by_xpath(
                    '//*[@id="wxadcontainer"]/div[1]/div[2]/div[4]/div/strong').text
                if '上传成功' in send_reslut_content:
                    break
                if '失败' in send_reslut_content:
                    err_num = err_num + 1
                    self.send_file(_)
                    if err_num > 3:
                        raise ValueError("图片上传失败")

        self.driver.execute_script('window.close();')
        self.driver.switch_to.window(self.driver.window_handles[-1])
        logging.info('传送文件,结束')

    def set_share_content(self, title_content, des_content):
        self.driver.find_element_by_xpath('//*[@id="wxadcontainer"]/div[1]/div/section/header/div[2]/button[2]').click()
        WebDriverWait(self.driver, 50).until(lambda driver: driver.find_element_by_xpath(
            '//*[@id="wxadcontainer"]/div[1]/div/div/div/div[1]/section/section/div[2]/div[1]/div/div/input'))
        title_input = self.driver.find_element_by_xpath(
            '//*[@id="wxadcontainer"]/div[1]/div/div/div/div[1]/section/section/div[2]/div[1]/div/div/input')
        title_input.click()
        title_input.send_keys(title_content)
        des_input = self.driver.find_element_by_xpath(
            '//*[@id="wxadcontainer"]/div[1]/div/div/div/div[1]/section/section/div[2]/div[2]/div/div/input')
        des_input.click()
        des_input.send_keys(des_content)
        self.driver.find_element_by_xpath('//*[@id="wxadcontainer"]/div[1]/div/header/div[2]/button').click()
        WebDriverWait(self.driver, 10).until(
            lambda driver: driver.find_element_by_xpath('/html/body/div[6]/div/div/div[2]/div/div[3]/button[2]'))
        self.driver.find_element_by_xpath('/html/body/div[6]/div/div/div[2]/div/div[3]/button[2]').click()
        time.sleep(1)

    @staticmethod
    def check_sucess_api(log_ad, layout_name):
        # 默认在进入公众号初始页面
        logging.info('开始检查落地页是否创建成功')
        WebDriverWait(log_ad.driver, 100).until(
            lambda driver: True if 'token' in log_ad.driver.current_url else False)
        cookie_dict = log_ad.get_cookie(log_ad.driver, login_cookie=False)
        token_id = re.findall('token=(\d+)', log_ad.driver.current_url)[0]
        requests.session()
        layout_url = 'https://mp.weixin.qq.com/promotion/landingpage_manager?action=list_pages&page=1&page_size=10&canvas_name={layout_name}&login_from=&is_grant=0&owner_uid=&token={wechat_token}&appid=&spid=&_={time_}'.format(
            layout_name=layout_name, wechat_token=token_id, time_=int(time.time() * 1000))
        session = requests.session()
        rsp = session.get(url=layout_url, cookies=cookie_dict)
        result_json = rsp.json()
        if 'landing_page_list' in result_json.keys():
            if len(result_json['landing_page_list']):
                logging.info('对应落地页存在于微信后台')
                return True
        else:
            logging.info('对应落地页不存在于微信后台')

    def remove_all_file(self):
        try:
            shutil.rmtree(os.getcwd() + '/' + self.img_dir)
        except Exception as e:
            logging.info(str(e))

    def create_layout(self, layout, sql_session, err_num=0):
        try:
            self.send_file_alone(layout)
            self.set_advertisement_sign(layout_name=layout['layoutName'])
            self.set_head_assemb(layout['topContent'])
            self.set_background_color(layout['pageBackColor'])

            for _ in layout['content']:

                if _['type'] == 'img':
                    self.set_page(_['content'])
                if _['type'] == 'txt':
                    self.set_content(_['content'])
                if _['type'] == 'followAcc':
                    self.set_follow_button(_['content'])

            self.set_share_content(title_content=layout['shareTittle'], des_content=layout['shareDesc'])

            self.remove_all_file()
            if self.check_sucess_api(layout_name=layout['layoutName'], log_ad=self.log_ad):
                return {'sucess': True, 'result_info': ''}
            else:
                if err_num > 3:
                    return {'sucess': False, 'result_info': '过程中全程无错误,失败'}
                else:

                    self.log_ad.refresh_driver()
                    self.log_ad.select_ad_master(service_name=self.service_name, wechat_name=self.wechat_name,
                                                 sql_session=sql_session)
                    self.get_into_create_page()
                    return self.create_layout(layout, sql_session, err_num=err_num + 1)
        except Exception as e:
            # raise
            logging.error(e)
            self.log_ad.driver.save_screenshot(
                'user_id:{}_time_{}_layout_name:{}_layout_error.png'.format(self.log_ad.user_id,
                                                                            datetime.now().strftime(
                                                                                "%Y-%m-%d, %H:%M:%S"),
                                                                            layout['layoutName']))
            # TODO:有空时讲 e 内容设置为原始内容
            if err_num > 3:
                return {'sucess': False, 'result_info': str(e)}
            else:
                self.log_ad.refresh_driver()
                self.log_ad.select_ad_master(service_name=self.service_name, wechat_name=self.wechat_name,
                                             sql_session=sql_session)
                self.get_into_create_page()
                return self.create_layout(layout, sql_session, err_num=err_num + 1)


if __name__ == '__main__':
    logging.basicConfig(
        handlers=[
            logging.handlers.RotatingFileHandler('./create_ad_layout.log',
                                                 maxBytes=10 * 1024 * 1024,
                                                 backupCount=5,
                                                 encoding='utf-8')
            , logging.StreamHandler()  # 供输出使用
        ],
        level=logging.INFO,
        format="%(asctime)s - %(levelname)s %(filename)s %(funcName)s %(lineno)s - %(message)s"
    )