import os, threading, datetime
 
# 獲取當(dāng)前文件所在目錄，拼接出LOG路徑
LOG_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), "log")
 
# 配置需要監(jiān)控的關(guān)鍵字
KEYWORDS = ["ANR ", "NullPointerException", "CRASH", "Force Closed"]
 
# 控制開啟和關(guān)閉
STOP_LOGCAT = True
 
# 指令對(duì)應(yīng)具體操作
INSTRUCTION = {
    "1": "filter_keywords",
    "2": "stop_filter_keywords",
    "3": "exit"
    }
 
def filter_keywords():
 global STOP_LOGCAT
 STOP_LOGCAT = False
 devices = get_devices() # 先獲取所有連接的設(shè)備
 print("開始監(jiān)控關(guān)鍵字")
 for device in devices:
  t = threading.Thread(target=filter_keyword, args=(device,))
  t.start()
 
def stop_filter_keywords():
 global STOP_LOGCAT
 if STOP_LOGCAT:
  print("沒有正在執(zhí)行的任務(wù)\n")
 else:
  STOP_LOGCAT = True
  print("正在停止關(guān)鍵字監(jiān)控\n")

def filter_keyword(device):
 print("設(shè)備%s關(guān)鍵字監(jiān)控已開啟" % str(device))
 sub = logcat(device)
 with sub:
  for line in sub.stdout: # 子進(jìn)程會(huì)持續(xù)輸出日志，對(duì)子進(jìn)程對(duì)象.stdout進(jìn)行循環(huán)讀取
   for key in KEYWORDS:
    if line.decode("utf-8").find(key) != -1: # stdout輸出為字節(jié)類型，需要轉(zhuǎn)碼
     message = "設(shè)備：%s 檢測(cè)到：%s\n" % (device, key)# 設(shè)備：192.168.56.104:5555 檢測(cè)到：ANR
     path = get_log_path("bugreport") # 根據(jù)時(shí)間創(chuàng)建文件夾
     bugreport(device, path)# 拉取完整日志壓縮包到創(chuàng)建的文件夾內(nèi)
     send_message(message) # 這里可以換成自己要做的事情，比如發(fā)送郵件或釘釘通知
   if STOP_LOGCAT:
    break
  print("設(shè)備%s關(guān)鍵字監(jiān)控已停止" % str(device))
  sub.kill()

# logcat持續(xù)輸出日志
def logcat(device):
 command = "adb -s " + str(device) + " logcat -v time"
 sub = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
 return sub

# 獲取所有device
def get_devices():
 command = "adb devices"
 res = os.popen(command).read()
 devices = []
 res = res.split("\n")
 for i in res:
  if i.endswith("device"):
   devices.append(i.split('\t')[0])
 return devices

# 打包下載所有日志到當(dāng)前目錄
def bugreport(device, path):
 os.chdir(path)# bugreport會(huì)下載日志到當(dāng)前文件夾，所以需要先切換到已經(jīng)創(chuàng)建的目錄
 command = "adb -s " + str(device) + " bugreport"
 subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, bufsize=-1)
 print("設(shè)備：%s 日志路徑：%s" % (str(device), path))

# 獲取日志存放路徑，如果不存在則按日期創(chuàng)建
def get_log_path(tag):
 year = datetime.datetime.now().strftime('%Y')
 month = datetime.datetime.now().strftime('%m')
 day = datetime.datetime.now().strftime('%d')
 path = os.path.join(LOG_PATH, tag, year, month, day)
 if not os.path.exists(path):
  os.makedirs(path)
 return path

def main():
 while True:
  print("-" * 100)
  print("1：開啟關(guān)鍵字監(jiān)控\n2：停止關(guān)鍵字監(jiān)控\n3：退出")
  print("-" * 100)
  instruction = str(input("\n\n請(qǐng)輸入要進(jìn)行的操作號(hào)：\n"))
  print("-" * 100)
  while instruction not in INSTRUCTION.keys():
   instruction = str(input("\n\n輸入無效，請(qǐng)重新輸入:"))
  if int(instruction) == 9:
   exit() # TODO 退出前需要判斷是否有正在執(zhí)行的monkey任務(wù)和關(guān)鍵字監(jiān)控任務(wù)
  eval(INSTRUCTION[str(instruction)] + "()")
 
if __name__ == '__main__':
 main()