import time
import logging

from appium import webdriver
from ding_talk import send_msg
from handle_file import EnToken
from conf.dbr import RedisClient
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from config import *

LOG_FORMAT = "%(asctime)s - %(levelname)s - line:%(lineno)s - msg:%(message)s"
logging.basicConfig(level=logging.INFO, format=LOG_FORMAT)
# logging.FileHandler(filename='app.log', encoding='utf-8')


# 微信獲取en token
class WeChat(object):
    def __init__(self):
        """
        初始化
        """
        # 驅(qū)動(dòng)配置
        self.desired_caps = {
            'platformName': PLATFORM,
            'deviceName': DEVICE_NAME,
            'appPackage': APP_PACKAGE,
            'appActivity': APP_ACTIVITY,
            'noReset': True
        }
        self.driver = webdriver.Remote(DRIVER_SERVER, self.desired_caps)
        self.wait = WebDriverWait(self.driver, TIMEOUT)
        self.hours_en = 60 * 60 * 1.1  # en控制1.1小時(shí)模擬點(diǎn)擊一次
        self.date_start_en = time.time()  # en開始時(shí)間
        self.date_end_en = 0  # en超過(guò)此時(shí)間后再次運(yùn)行
        # self.date_end_en = self.date_start_en + self.hours_en  # en超過(guò)此時(shí)間后再次運(yùn)行
        self.week = 60 * 60 * 24 * 7  # 按照周的頻率對(duì)xd進(jìn)行token更新
        self.week_start_xd = time.time()  # xd的開始時(shí)間
        self.week_end_xd = 0  # 根據(jù)周控制頻率控制再次開啟時(shí)間
        self.week_start_xiu = time.time()  # xd的開始時(shí)間
        self.week_end_xiu = 0  # 根據(jù)周控制頻率控制再次開啟時(shí)間

    def login(self):
        """
        登錄微信
        :return:
        """
        # 登錄按鈕
        a = time.time()
        try:
            login = self.wait.until(EC.presence_of_element_located((By.ID, 'com.tencent.mm:id/f34')))
            login.click()
        except Exception as e:
            # print(e)
            logging.info(f'failed login {e}')
        b = time.time() - a
        # print('點(diǎn)擊登錄', b)
        logging.info(f'click login,use time ')
        # 手機(jī)輸入
        try:
            phone = self.wait.until(EC.presence_of_element_located((By.ID, 'com.tencent.mm:id/bem')))
            phone.set_text(USERNAME)
        except Exception as e:
            # print(e)
            logging.info(f'something wrong{e}')
        c = time.time() - a - b
        # print('手機(jī)號(hào)輸入', c)
        logging.info(f'send keys phone nums use time {c}')
        # 下一步
        try:
            next = self.wait.until(EC.element_to_be_clickable((By.ID, 'com.tencent.mm:id/dw1')))
            next.click()
        except Exception as e:
            logging.info(f'something wrong{e}')
        d = time.time() - a - b - c
        logging.info(f'click next bottom use time {c}')
        # 密碼
        password = self.wait.until(EC.presence_of_element_located((By.XPATH, '//*[@text="請(qǐng)?zhí)顚懳⑿琶艽a"]')))
        password.set_text(PASSWORD)
        e = time.time() - a - b - c - d
        logging.info(f'send keys password use time {e}')
        # 提交
        # submit = self.wait.until(EC.element_to_be_clickable((By.ID, 'com.tencent.mm:id/dw1')))
        submit = self.wait.until(EC.element_to_be_clickable((By.XPATH, '//*[@text="登錄"]')))
        submit.click()
        f = time.time() - a - b - c - d - e
        logging.info(f'commit password use time {f}')

    def run(self):
        """
        入口
        :return:
        """
        # 滑動(dòng)之后等待出現(xiàn)en小程序
        self.slide_down()
        time.sleep(10)
        # 點(diǎn)擊進(jìn)入en小程序

        self.touch_en()
        if self.week_end_xd < self.week_start_xd:
            self.week_start_xd = time.time()
            self.week_end_xd = self.week_start_xd + self.week
            print('xd點(diǎn)擊')
            self.touch_xd()

        elif self.week_end_xiu < self.week_start_xiu:
            self.week_end_xiu = time.time() + self.week
            print('xiu')
            self.touch_xiu()

        time.sleep(10)
        # 退出小程序
        self.driver_closed()
        print('driver closed')
        emt = EnToken()
        token_res = emt.token_2_redis()
        if not token_res:
            print('需要發(fā)送失敗消息')
            return False
        return True

    def slide_down(self):
        """
        滑動(dòng)微信屏幕之后點(diǎn)擊小程序
        :return:
        """
        window_size_phone = self.driver.get_window_size()
        # print(window_size_phone)
        phone_width = window_size_phone.get('width')
        phone_height = window_size_phone.get('height')
        # print(phone_width, phone_height)
        time.sleep(15)
        x1 = phone_width * 0.5
        y1 = phone_height * 0.7
        y2 = phone_height * 0.26
        # print('準(zhǔn)備向下滑動(dòng)')
        logging.info(f'prepare slide down')
        a = time.time()
        self.driver.swipe(x1, y2, x1, y1, 2050)
        # print('向下滑動(dòng)完成', time.time() - a)
        logging.info(f'slide down success use time {time.time() - a}')

    def touch_en(self):
        """
        每次進(jìn)來(lái)之后都需要判斷是否到了時(shí)間，若時(shí)間到了之后才可執(zhí)行點(diǎn)擊操作
        :param : en 代表en; xd 代表xd; xiu 代表xiu.
        :return: None 無(wú)返回值
        """
        print(self.date_end_en, time.time())
        if self.date_end_en < time.time():  # 此時(shí)的時(shí)候已經(jīng)超時(shí)，需要再次從新進(jìn)行點(diǎn)擊
            print('en模擬點(diǎn)擊')
            # 從新定義開始結(jié)束時(shí)間
            print(self.date_end_en, time.time())
            self.date_end_en = time.time() + self.hours_en  # 再次更改end time為n小時(shí)后
            print(self.date_end_en, time.time())
            try:
                # print('id定位en')
                en_app = self.wait.until(
                    EC.presence_of_element_located((By.XPATH, f"http://android.widget.TextView[@text='textname…']")))
                # en_master = self.wait.until(EC.presence_of_element_located((By.ID, 'com.tencent.mm:id/hu')))
                # en_master = self.wait.until(
                # EC.presence_of_element_located((By.XPATH, "http://android.widget.TextView[@text='textname']")))
                en_app.click()
                logging.info(f'located by app_name en')
            except Exception as error:
                # print(e, 'id定位失敗')
                logging.info(f'failed located by id:{error}')
            time.sleep(20)
            # 關(guān)閉小程序按鈕點(diǎn)擊
            print('close the en app')
            close_button = self.wait.until(EC.presence_of_element_located((By.XPATH, f"http://android.widget.FrameLayout[2]/android.widget.ImageButton")))
            close_button.click()
            print('點(diǎn)擊了關(guān)閉小程序')

    def touch_xd(self):
        """
        需要考慮是否已經(jīng)登錄狀態(tài)還是需要再次登錄
        :return:
        """
        # 點(diǎn)擊后進(jìn)入到小程序
        logging.info('click app xd')
        xd_app = self.wait.until(EC.presence_of_element_located((By.XPATH, "http://android.widget.TextView[@text='textname']")))
        xd_app.click()
        time.sleep(20)
        # 頁(yè)面出現(xiàn)需要獲取到你的定位的時(shí)候需要點(diǎn)擊允許
        print('點(diǎn)擊確認(rèn)獲取當(dāng)前位置')
        self.driver.tap([(510, 679)], 500)

        # 點(diǎn)擊進(jìn)入到個(gè)人中心
        time.sleep(10)
        logging.info('click personal xd')
        self.driver.tap([(540, 1154)], 500)
        # 點(diǎn)擊快速登錄進(jìn)行登錄
        time.sleep(10)
        logging.info('click login xd')
        self.driver.tap([(270, 1030)], 500)
        # 點(diǎn)擊同意獲取頭像信息
        time.sleep(10)
        logging.info('同意獲取頭像等相關(guān)信息')
        self.driver.tap([(510, 775)], 500)
        time.sleep(20)
        # 關(guān)閉小程序按鈕點(diǎn)擊
        print('close the guaishou app')
        close_button = self.wait.until(
            EC.presence_of_element_located((By.XPATH, f"http://android.widget.FrameLayout[2]/android.widget.ImageButton")))
        close_button.click()
        print('結(jié)束')
        time.sleep(30)

    def touch_xiu(self):
        """
        xiu模擬點(diǎn)擊，需要考慮是否需要登錄狀態(tài)下
        :return:
        """
        # 點(diǎn)擊后進(jìn)入到小程序
        logging.info('click app xiu')
        xiu_app = self.wait.until(EC.presence_of_element_located((By.XPATH, "http://android.widget.TextView[@text='xiu']")))
        xiu_app.click()
        # 若頁(yè)面顯示需要確認(rèn)獲取當(dāng)前位置的話需要點(diǎn)擊確認(rèn)
        logging.info('click confirm xiu')
        time.sleep(15)
        confirm_loc = self.wait.until(
            EC.presence_of_element_located((By.XPATH, "http://android.widget.Button[@text='確定']")))
        confirm_loc.click()
        # 點(diǎn)擊個(gè)人中心
        logging.info('click personal xiu')
        time.sleep(5)
        try:
            personal = self.wait.until(
                EC.presence_of_element_located((By.XPATH, "http://android.view.View[@content-desc='個(gè)人中心']")))
            personal.click()
        except Exception as e:
            print(e)
        # 點(diǎn)擊快速登錄進(jìn)行登錄
        logging.info('click login xiu')
        time.sleep(5)
        try:
            login = self.wait.until(EC.presence_of_element_located((By.XPATH, "http://android.view.View[@content-desc='立即登錄']")))
            login.click()
        except Exception as e:
            print('xiu已經(jīng)登錄，不需要再次點(diǎn)擊確認(rèn)登錄')
        time.sleep(30)

    def driver_closed(self):
        self.driver.quit()


if __name__ == '__main__':
    conn_r = RedisClient(db=10)
    count_1 = 0
    # start_time = time.time()
    # end_time = time.time() + 60 * 60 * 1
    we_chat = WeChat()
    try:
        while 1:
            if conn_r.r_size() < 3:  # 監(jiān)控Redis情況，當(dāng)Redis中無(wú)數(shù)據(jù)后開始運(yùn)行一次
                res = we_chat.run()  # 操作微信做操作點(diǎn)擊en小程序生成token
                if not res:
                    count_1 += 1
                    if count_1 > 10:
                        break  # 當(dāng)失敗十次之后跳出循環(huán)
                # 此處增加限制，每次生成token之后一個(gè)小時(shí)后才會(huì)產(chǎn)生新的token，防止一個(gè)token多次使用導(dǎo)致被封號(hào)
                time.sleep(60*60)
            else:
                time.sleep(60*60)  # 當(dāng)有數(shù)據(jù)的時(shí)候等待五分鐘
            we_chat.driver = webdriver.Remote(DRIVER_SERVER, we_chat.desired_caps)
            we_chat.wait = WebDriverWait(we_chat.driver, TIMEOUT)
    except Exception as e:
        msg = f'業(yè)務(wù)報(bào)警：' \
            f'\n en獲取token出現(xiàn)問(wèn)題' \
            f'\n{e}'
        send_msg(msg)
        # print(e, type(e))

 import os

# 平臺(tái)
PLATFORM = 'Android'

# 設(shè)備名稱 通過(guò) adb devices -l 獲取
DEVICE_NAME = 'MI_9'

# APP路徑
APP = os.path.abspath('.') + '/weixin.apk'

# APP包名
APP_PACKAGE = 'com.tencent.mm'

# 入口類名
APP_ACTIVITY = '.ui.LauncherUI'

# Appium地址
DRIVER_SERVER = 'http://localhost:4723/wd/hub'
# 等待元素加載時(shí)間
TIMEOUT = 10

# 微信手機(jī)號(hào)密碼
USERNAME = 'wechatname'
PASSWORD = 'wechatpwd'

# 滑動(dòng)點(diǎn)
FLICK_START_X = 300
FLICK_START_Y = 300
FLICK_DISTANCE = 700

import re
import os
import logging

from conf.dbr import RedisClient

LOG_FORMAT = "%(asctime)s - %(levelname)s - line:%(lineno)s - msg:%(message)s"
logging.basicConfig(level=logging.DEBUG, format=LOG_FORMAT)


# 處理en token到Redis
class EnToken(object):
    def __init__(self):
        # self.token_path = 'F:\\en.txt'
        # self.token_path = 'F:\\xiu.txt'
        # self.token_path = 'F:\\xd.txt'
        self.conn = RedisClient(db=10)  # 解析日維度價(jià)格
        self.conn_en = RedisClient(db=9)  # 解析當(dāng)前經(jīng)緯度范圍內(nèi)店鋪點(diǎn)位

    # 處理en token文件，從文件中讀取到token之后只取最后一個(gè)，取到之后刪除本地文件
    @staticmethod
    def handle_en_txt():
        token_dict = {}
        path_token_list = [
            ('en', '>(e.*?)-->'),
            ('xd', 'headers-->(.*?)-->'),
            ('xiu', r'>(\d+)-->'),
                      ]
        for i in path_token_list:
            token_path = f'F:\\{i[0]}.txt'
            token_re = i[-1]
            if os.path.exists(token_path):
                with open(token_path, mode='r', encoding='utf-8') as f:
                    token_str = f.read()
                    # print(token_str)
                    # token_list = re.findall('>(e.*?)-->', token_str)
                    # token_list = re.findall('>(Q.*?)-->', token_str)
                    # token_list = re.findall('>(\d+)-->', token_str)
                    token_list = re.findall(token_re, token_str)
                    print(token_list)
                    if token_list:
                        token = token_list[-1]
                        print(token)
                        token_dict[i[0]] = token
                os.remove(token_path)  # 刪除掉
            # return token
        else:
            # print('file_en_dont_exit')
            logging.info('file_en_dont_exit')
        return token_dict

    # 將token放到Redis中
    def token_2_redis(self):
        """
        假如token存在的話 則根據(jù)token的最后幾位做key放入到Redis中
        :return:
        """
        token_dict = self.handle_en_txt()
        print(token_dict)
        if token_dict:
            for token_items in token_dict.items():
                token_key = token_items[0]
                token_val = token_items[-1]
                self.conn.set(token_key, token_val, over_time=None)
                # self.conn.set(token_key, token, over_time=60*65)  # 設(shè)置有效時(shí)長(zhǎng)65分鐘之后失效
                # self.conn_en.set(token_key, token, over_time=60*65)  # 設(shè)置有效時(shí)長(zhǎng)65分鐘之后失效
                logging.info(f'token success {token_key,token_val}')
            return True
        else:
            logging.info('token dons"t exist')
        self.conn.close()
        self.conn_en.close()


if __name__ == '__main__':
    en = EnToken()
    en.token_2_redis()

 if (oSession.oRequest["Host"]=="這里是請(qǐng)求的host") {
                    var filename = "F:\en.txt";
                    var curDate = new Date();
                    var logContent = 'en' + "[" + curDate.toLocaleString() +  "]";
                    var sw : System.IO.StreamWriter;
                    if (System.IO.File.Exists(filename)){
                          sw = System.IO.File.AppendText(filename);
                          sw.Write(logContent + 'oSession.oRequest.headers-->'  + oSession.oRequest.headers['x-wx-token'] + '-->' + oSession.oRequest.headers  +'\n');
                          // sw.Write("Request header:" + "\n" +  oSession.oRequest.headers);
                          // sw.Write(wap_s + '\n\n')
                    }
                    else{
                          sw = System.IO.File.CreateText(filename);
                          sw.Write(logContent + 'oSession.oRequest.headers-->'  + oSession.oRequest.headers['x-wx-token'] + '-->' + '\n');
                          // sw.Write("Request header:" + "\n" +  oSession.oRequest.headers);
                          // sw.Write(wap_s + '\n\n')
                    }
                    sw.Close();
                    sw.Dispose();
             }