import os

def getCurPath():
    return os.getcwd()

def getAllFiles(data_path):
    # 返回data_path目录下的所有文件名
    return [f for f in os.listdir(data_path)]

def isUcExtension(file):
    # 判断是否是.uc文件
    if len(file) >= 4 and file[-4:] == '.uc!':
        return True
    else:
        return False

def ucToFlac(file):
    # 将当前文件按字节与0xA3进行异或，并对文件格式进行修改
    fSource = open(file, 'rb')
    fOut = open(file[:-4] + '.flac', 'wb')
    content = bytearray(fSource.read())
    for index in range(len(content)):
        content[index] ^= 0xA3
    fOut.write(content)
    fSource.close()
    fOut.close()

if __name__ == '__main__':
    origin_path = "Music"
    cur_path = getCurPath()
    total_origin_path = os.path.join(cur_path, origin_path)

    # 函数调用：将当前目录下所有的.uc文件自动解码成.flac文件
    files = getAllFiles(total_origin_path)
    for file in files:
        if isUcExtension(os.path.join(total_origin_path, file)):
            ucToFlac(os.path.join(total_origin_path, file))
            print(file[:-4] + '.flac' + '转换成功')