wehub-resource-sync
49 lines
1.9 KiB
Python
49 lines
1.9 KiB
Python
import os
|
|
import argparse
|
|
import shutil
|
|
from shutil import copyfile
|
|
from tqdm import tqdm
|
|
|
|
|
|
def process_and_rename(character_name):
|
|
dataset_path = f"./dataset/{character_name}"
|
|
if not os.path.exists(dataset_path):
|
|
os.makedirs(dataset_path)
|
|
else:
|
|
print(f'检测到"{dataset_path}" 已存在，执行删除')
|
|
shutil.rmtree(dataset_path)
|
|
os.makedirs(dataset_path)
|
|
|
|
counter = 1
|
|
|
|
all_folders = [folder for r, d, f in os.walk('./merge') for folder in d]
|
|
for folder in tqdm(all_folders, desc="Processing folders", unit="folder"):
|
|
mapping_path = os.path.join('./merge', folder, "new_mapping.list")
|
|
if os.path.exists(mapping_path):
|
|
with open(mapping_path, 'r', encoding='utf-8') as file:
|
|
lines = file.readlines()
|
|
|
|
for line in tqdm(lines, desc=f"Processing files in {folder}", unit="file", leave=False):
|
|
old_filename, text = line.strip().split("|")
|
|
old_filepath = os.path.join('./merge', folder, old_filename)
|
|
|
|
new_filename = f"{character_name}_{counter}.wav"
|
|
new_filepath = os.path.join(dataset_path, new_filename)
|
|
new_mapping_entry = f"./dataset/{character_name}/{new_filename}|{character_name}|ZH|{text}"
|
|
|
|
# Copy and rename the file
|
|
copyfile(old_filepath, new_filepath)
|
|
|
|
with open(os.path.join(dataset_path, "dataset_mapping.list"), 'a', encoding='utf-8') as dataset_file:
|
|
dataset_file.write(new_mapping_entry + "\n")
|
|
|
|
counter += 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Rename and restructure files based on character_name")
|
|
parser.add_argument("character_name", type=str, help="Name of the character for restructuring")
|
|
args = parser.parse_args()
|
|
|
|
process_and_rename(args.character_name)
|