files.py

"""数据集整理"""

import os, shutil

def remove(file_path: str) -> None:
    """批量删除指定路径下所有非 `.wav` 文件"""
    for root, dirs, files in os.walk(file_path):
        for item in files:
            if not item.endswith('.wav'):
                try:
                    print("Delete file: ", os.path.join(root, item))
                    os.remove(os.path.join(root, item))
                except:
                    continue

def rename(file_path: str) -> None:
    """批量按指定格式改名（不然把相同情感的音频整理到同一个文件夹时会重名）"""
    for root, dirs, files in os.walk(file_path):
        for item in files:
            if item.endswith('.wav'):
                people_name = root.split('/')[-2]
                emotion_name = root.split('/')[-1]
                item_name = item[:-4] # 音频原名（去掉.wav）
                old_path = os.path.join(root, item)
                new_path = os.path.join(root, item_name + '-' + emotion_name + '-'+ people_name + '.wav') # 新音频路径
                try:
                    os.rename(old_path, new_path)
                    print('converting ', old_path, ' to ', new_path)
                except:
                    continue

def move(file_path: str) -> None:
    """把音频按情感分类，放在不同文件夹下"""
    for root, dirs, files in os.walk(file_path):
        for item in files:
            if item.endswith('.wav'):
                emotion_name = root.split('/')[-1]
                old_path = os.path.join(root, item)
                new_path = os.path.join(file_path, emotion_name, item)
                try:
                    shutil.move(old_path, new_path)
                    print("Move ", old_path, " to ", new_path)
                except:
                    continue

def mkdirs(folder_path: str) -> None:
    """检查文件夹是否存在，如果不存在就创建一个"""
    if not os.path.exists(folder_path):
        os.makedirs(folder_path)