import os
import re
from time import sleep

import librosa

# 获取桌面路径
desktop_path = os.path.join(os.path.expanduser("~"), "Desktop\\")
regular = re.compile("2.*.ogg")
while True:
    # 获取tmp目录下所有的文件名
    list_tmp = os.popen("adb shell ls /tmp").read()
    # 判断是否有ogg文件
    if "ogg" in list_tmp:
        # 使用正则表达式将ogg文件名从所有字符串中剥离出来
        filename = regular.findall(list_tmp)[0]
        # 获取文件大小（结果中包含文件名所以后续还需要进一步的处理）
        filesize = os.popen("adb shell du -H /tmp/{0}".format(filename)).read()
        # 为了防止程序过早或者过晚抓取ogg文件导致文件不完整或丢失，每次循环时都会进行两次文件大小读取进行比较。这里是第一次读取
        filesize1_result = filesize[:filesize.find("/")]
        if filesize1_result == "du: ":
            continue
        else:
            filesize1_result = int(filesize1_result)

        # 睡眠2秒
        sleep(2)

        # 第二次读取文件大小
        filesize = os.popen("adb shell du -H /tmp/{0}".format(filename)).read()
        filesize2_result = filesize[:filesize.find("/")]
        if filesize2_result == "du: ":
            continue
        else:
            filesize2_result = int(filesize2_result)
        # 将第二次和第一次读取的文件大小进行比较，如果第二次大于第一次（不存在第一次比第二次大的情况），则跳过下面的环节，直接进行新一轮的循环
        if filesize2_result > filesize1_result:
            continue
        # 如果第二次等于第一次，说明文件写入已经结束，此时文件是完整的，可以抓取
        elif filesize2_result == filesize1_result:
            # 判断文件路径是否存在,没有就创建
            if not os.path.exists(desktop_path+"newtmpogg"):
                os.mkdir(desktop_path+"newtmpogg")
            # 判断ogg音频文件是否存在，如果存在就不写入
            if os.path.exists(desktop_path+r"newtmpogg\{0}".format(filename)):
                pass
            else:
                os.system(r"adb pull /tmp/{0} {1}newtmpogg".format(filename,desktop_path))
                filepath = desktop_path+r"newtmpogg\{0}".format(filename)
                try:
                    duration = librosa.get_duration(filename=filepath)
                    # 创建日志文件，将文件名和时长记录在内
                    with open(desktop_path+"voicelog.txt", "a", encoding="utf-8") as f:
                        f.write("{0}       时长：        {1}\n".format(filename, duration))
                except Exception as e:
                    print(e)