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ū)動配置
  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小時模擬點擊一次
  self.date_start_en = time.time()  # en開始時間
  self.date_end_en = 0  # en超過此時間后再次運行
  # self.date_end_en = self.date_start_en + self.hours_en  # en超過此時間后再次運行
  self.week = 60 * 60 * 24 * 7  # 按照周的頻率對xd進行token更新
  self.week_start_xd = time.time()  # xd的開始時間
  self.week_end_xd = 0  # 根據(jù)周控制頻率控制再次開啟時間
  self.week_start_xiu = time.time()  # xd的開始時間
  self.week_end_xiu = 0  # 根據(jù)周控制頻率控制再次開啟時間
 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('點擊登錄', b)
  logging.info(f'click login,use time ')
  # 手機輸入
  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('手機號輸入', 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="請?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:
  """
  # 滑動之后等待出現(xiàn)en小程序
  self.slide_down()
  time.sleep(10)
  # 點擊進入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點擊')
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):
  """
  滑動微信屏幕之后點擊小程序
  :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('準備向下滑動')
  logging.info(f'prepare slide down')
  a = time.time()
  self.driver.swipe(x1, y2, x1, y1, 2050)
  # print('向下滑動完成', time.time() - a)
  logging.info(f'slide down success use time {time.time() - a}')
 def touch_en(self):
  """
  每次進來之后都需要判斷是否到了時間，若時間到了之后才可執(zhí)行點擊操作
  :param : en 代表en; xd 代表xd; xiu 代表xiu.
  :return: None 無返回值
  """
  print(self.date_end_en, time.time())
  if self.date_end_en < time.time():  # 此時的時候已經(jīng)超時，需要再次從新進行點擊
print('en模擬點擊')
# 從新定義開始結(jié)束時間
print(self.date_end_en, time.time())
self.date_end_en = time.time() + self.hours_en  # 再次更改end time為n小時后
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)閉小程序按鈕點擊
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('點擊了關(guān)閉小程序')
 def touch_xd(self):
  """
  需要考慮是否已經(jīng)登錄狀態(tài)還是需要再次登錄
  :return:
  """
  # 點擊后進入到小程序
  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)
  # 頁面出現(xiàn)需要獲取到你的定位的時候需要點擊允許
  print('點擊確認獲取當前位置')
  self.driver.tap([(510, 679)], 500)
  # 點擊進入到個人中心
  time.sleep(10)
  logging.info('click personal xd')
  self.driver.tap([(540, 1154)], 500)
  # 點擊快速登錄進行登錄
  time.sleep(10)
  logging.info('click login xd')
  self.driver.tap([(270, 1030)], 500)
  # 點擊同意獲取頭像信息
  time.sleep(10)
  logging.info('同意獲取頭像等相關(guān)信息')
  self.driver.tap([(510, 775)], 500)
  time.sleep(20)
  # 關(guā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模擬點擊，需要考慮是否需要登錄狀態(tài)下
  :return:
  """
  # 點擊后進入到小程序
  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()
  # 若頁面顯示需要確認獲取當前位置的話需要點擊確認
  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()
  # 點擊個人中心
  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='個人中心']")))
personal.click()
  except Exception as e:
print(e)
  # 點擊快速登錄進行登錄
  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)登錄，不需要再次點擊確認登錄')
  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情況，當Redis中無數(shù)據(jù)后開始運行一次
 res = we_chat.run()  # 操作微信做操作點擊en小程序生成token
 if not res:
  count_1 += 1
  if count_1 > 10:break  # 當失敗十次之后跳出循環(huán)
 # 此處增加限制，每次生成token之后一個小時后才會產(chǎn)生新的token，防止一個token多次使用導(dǎo)致被封號
 time.sleep(60*60)
else:
 time.sleep(60*60)  # 當有數(shù)據(jù)的時候等待五分鐘
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ù)報警：' \
f'\n en獲取token出現(xiàn)問題' \
f'\n{e}'
  send_msg(msg)
  # print(e, type(e))

 import os
# 平臺
PLATFORM = 'Android'
# 設(shè)備名稱 通過 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'
# 等待元素加載時間
TIMEOUT = 10
# 微信手機號密碼
USERNAME = 'wechatname'
PASSWORD = 'wechatpwd'
# 滑動點
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)  # 解析日維度價格
  self.conn_en = RedisClient(db=9)  # 解析當前經(jīng)緯度范圍內(nèi)店鋪點位
 # 處理en token文件，從文件中讀取到token之后只取最后一個，取到之后刪除本地文件
 @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è)置有效時長65分鐘之后失效
 # self.conn_en.set(token_key, token, over_time=60*65)  # 設(shè)置有效時長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"]=="這里是請求的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();
 }