Python logcat 키워드 모니터링

import os, threading, datetime

# 获取当前文件所在目录，拼接出LOG路径
LOG_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), "log")

# 配置需要监控的关键字
KEYWORDS = ["ANR ", "NullPointerException", "CRASH", "Force Closed"]

# 控制开启和关闭
STOP_LOGCAT = True

# 指令对应具体操作
INSTRUCTION = {
               "1": "filter_keywords",
               "2": "stop_filter_keywords",
               "3": "exit"
               }


def filter_keywords():
    global STOP_LOGCAT
    STOP_LOGCAT = False
    devices = get_devices()  # 先获取所有连接的设备
    print("开始监控关键字")
    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("没有正在执行的任务\n")
    else:
        STOP_LOGCAT = True
        print("正在停止关键字监控\n")

def filter_keyword(device):
    print("设备%s关键字监控已开启" % str(device))
    sub = logcat(device)
    with sub:
        for line in sub.stdout: # 子进程会持续输出日志，对子进程对象.stdout进行循环读取
            for key in KEYWORDS:
                if line.decode("utf-8").find(key) != -1: # stdout输出为字节类型，需要转码
                    message = "设备：%s 检测到：%s\n" % (device, key)# 设备：192.168.56.104:5555 检测到：ANR
                    path = get_log_path("bugreport") # 根据时间创建文件夹
                    bugreport(device, path)# 拉取完整日志压缩包到创建的文件夹内
                    send_message(message) # 这里可以换成自己要做的事情，比如发送邮件或钉钉通知
            if STOP_LOGCAT:
                break
        print("设备%s关键字监控已停止" % str(device))
        sub.kill()

# logcat持续输出日志
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

# 打包下载所有日志到当前目录
def bugreport(device, path):
    os.chdir(path)# bugreport会下载日志到当前文件夹，所以需要先切换到已经创建的目录
    command = "adb -s " + str(device) + " bugreport"
    subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, bufsize=-1)
    print("设备：%s 日志路径：%s" % (str(device), path))

# 获取日志存放路径，如果不存在则按日期创建
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：开启关键字监控\n2：停止关键字监控\n3：退出")
        print("-" * 100)
        instruction = str(input("\n\n请输入要进行的操作号：\n"))
        print("-" * 100)
        while instruction not in INSTRUCTION.keys():
            instruction = str(input("\n\n输入无效，请重新输入:"))
        if int(instruction) == 9:
            exit()  # TODO 退出前需要判断是否有正在执行的monkey任务和关键字监控任务
        eval(INSTRUCTION[str(instruction)] + "()")


if __name__ == '__main__':
    main()