selenium_action/wechat_action/login_ad.py

from selenium import webdriver
from selenium.webdriver import ActionChains
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait
from wechat_action.create_ad_plan_idea import IdeaAction
from wechat_action.human_ad import HumanAd
from selenium.webdriver import ChromeOptions
from selenium.webdriver.common.keys import Keys
from sqlalchemy import Table
from wechat_action import sql_tools
import time
import pickle
from settings import using_config
from wechat_action.sql_models import DB
import requests
import json
import logging


class LogIn:
    # TODO:整体运行使用逻辑,需要修改几次
    def __init__(self, user_id):
        # 获取到单独服务商下的独立公众号页面
        # TODO:之后的落地页创建,推广计划创建,改动service_name,wechat_name通过,login函数主动改动
        self.user_id = user_id
        self.driver = self.get_driver()
        self.db = DB(config=using_config)
        self.wechat_cookies_table = Table('wechat_cookies', self.db.metadata,
                                          autoload=True, autoload_with=self.db.engine)
        self.sql_session = self.db.DBSession()
        # self.log_in()
        # self.select_ad_master()

    def get_driver(self):
        options = ChromeOptions()
        # 防止selenium快速崩坏
        options.add_argument("--disable-dev-shm-usage")
        options.add_experimental_option('excludeSwitches', ['enable-automation'])
        # prefs = {"profile.managed_default_content_settings.images": 2, 'permissions.default.stylesheet': 2}
        # options.add_experimental_option("prefs", prefs)

        # driver = webdriver.Remote(
        #     # command_executor='http://192.168.1.101:4444/wd/hub',
        #     command_executor='http://118.31.53.105:4555/wd/hub',
        #     options=options)
        driver = webdriver.Chrome(options=options)
        driver.maximize_window()
        return driver

    def log_in(self):
        # self.wechat_cookie_use()
        logging.info('开始登录')
        self.driver.get('https://a.weixin.qq.com/index.html')

        img_selector = 'body > div.old-template > div > div > div.waiting.panelContent > div.wrp_code > img'
        frame_login = self.driver.find_element_by_xpath('//*[@id="login_container"]/iframe')
        self.driver.switch_to.frame(frame_login)
        # WebDriverWait(driver, 3).until(EC.invisibility_of_element_located((By.CSS_SELECTOR, img_selector)))
        # time.sleep(3)
        img_url = self.driver.find_element_by_css_selector(img_selector)
        print('img', img_url.get_attribute('src'))
        return img_url.get_attribute('src')

    def log_in_wait(self):
        # 默认等待6分钟
        WebDriverWait(self.driver, 6 * 60).until(lambda driver: self.driver.find_elements_by_link_text('广告投放'))
        logging.info('登录成功')

    def select_ad_master(self, service_name, wechat_name):
        # TODO:每次切换服务商,cookie进行数据库保存
        logging.info('开始切换服务商')
        time.sleep(5)
        self.driver.execute_script('''
                    window.scroll(0,1000000);
                    ''')
        self.driver.find_element_by_css_selector(
            '#root > div > header > div > div.CoreLayout__account-2lIr0 > div').click()
        self.driver.find_element_by_css_selector(
            '#root > div > div.CoreLayout__headerDropdown-3xWkD > div > div:nth-child(1) > button').click()
        service_names = self.driver.find_elements_by_class_name('CoreLayout__headerDropdownItem-X4S98')
        choice_service = None
        for _ in service_names:
            if service_name in _.text:
                choice_service = _
        choice_service.click()
        # 挑选广告投放位置
        time.sleep(3)
        input_wechat_name = self.driver.find_element_by_class_name('TextInput_new__iconRight-pekjS')
        input_wechat_name.click()
        input_wechat_name.send_keys(wechat_name)
        input_wechat_name.send_keys(Keys.RETURN)
        self.driver.execute_script('''
            window.scroll(100000,1000000);
            var e_one=document.getElementsByClassName('Table_new__wrapper-1cpZN')[0];
            e_one.scroll(10000,100000);
            ''')
        time.sleep(5)
        elements = self.driver.find_elements_by_link_text('广告投放')
        elements[0].click()
        time.sleep(1)
        logging.info('切换服务商成功')
        # 切换窗口,点击创建广告,切到广告页面

        self.driver.switch_to.window(self.driver.window_handles[-1])
        WebDriverWait(self.driver, 100).until(lambda driver: driver.find_element_by_class_name(
            'adui-button-hasLeftIcon'))

        # cookie 进行数据库保存
        wechat_cookies = self.wechat_cookie_pickle()
        update_res = self.wechat_cookies_table.update() \
            .where(self.wechat_cookies_table.c.user_id == self.user_id) \
            .values(cookies=wechat_cookies)
        self.sql_session.execute(update_res)
        self.sql_session.commit()
        if update_res.rowcount == 0:
            wechat_cookies_info = {'user_id': self.user_id, 'cookies': wechat_cookies}
            wechat_insert = sql_tools.save_wechat_cookies(wechat_cookies_info=wechat_cookies_info,
                                                          table_wechat_cookies=self.wechat_cookies_table)
            self.sql_session.execute(wechat_insert)
            self.sql_session.commit()

    @staticmethod
    def get_cookie(driver, login_cookie=True):
        if login_cookie:
            WebDriverWait(driver, 100).until(
                lambda x: [True for _ in driver.get_cookies() if 'token_ticket' == _['name']]
            )
        cookies = driver.get_cookies()
        cookie_dict = {}
        for _ in cookies:
            print(_)
            cookie_dict[_['name']] = _['value']
        return cookie_dict

    def wechat_cookie_pickle(self):
        self.driver.get('https://a.weixin.qq.com/client')
        WebDriverWait(self.driver, 100).until(
            lambda x: [True for _ in self.driver.get_cookies() if 'token_ticket' == _['name']]
        )
        cookies = self.driver.get_cookies()
        cookies_obj = pickle.dumps(cookies)
        return cookies_obj

    def wechat_cookies_check_alive(self, driver_cookies):
        # wechat 检查cookies 是否可用
        # 可用返回ture
        check_url = 'https://a.weixin.qq.com/cgi-bin/agency/get_delivery_metrics?page=1&page_size=10&search_key=&order_by=&ascending=1&only_collect=0&g_tk=5381&_={}'.format(
            int(time.time() * 1000))
        # cookie_dict = {}
        # for _ in driver_cookies:
        #     cookie_dict[_['name']] = _['value']
        # rsp_json = requests.get(url=check_url, cookies=cookie_dict).json()

        self.driver.get('https://www.baidu.com')
        self.driver.get('https://a.weixin.qq.com/client')
        for _ in driver_cookies:
            self.driver.add_cookie(_)
        self.driver.get('https://a.weixin.qq.com/client')
        self.driver.get(check_url)

        print('检查cookies 结果', self.driver.page_source)
        if '4101' in self.driver.page_source:
            return False
        else:
            return True

    def upadte_user_info(self):
        # TODO: 更新 用户相关信息
        #  每次登录就更新一次相关数据,------公众号相关数据,人群报相关数据

        pass

    def get_driver_loged(self):
        return self.driver

    def refresh_driver(self):
        err_num = 0
        logging.info('开始刷新chrome')
        while True:
            try:
                if len(self.driver.window_handles) > 1:
                    self.driver.switch_to.window(self.driver.window_handles[-1])
                    self.driver.execute_script('window.close();')
                    time.sleep(1)
                    # 规避有弹窗的情况
                    self.driver.switch_to.alert.accept()
                else:
                    self.driver.switch_to.window(self.driver.window_handles[-1])
                    self.driver.get('https://a.weixin.qq.com')
                    break
            except Exception as e:
                print(e)
                err_num = err_num + 1
                if err_num > 3:
                    break
        logging.info('刷新chrome 结束')