Plachta
28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
from google.colab import files
|
|
import shutil
|
|
import os
|
|
import argparse
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--type", type=str, required=True, help="type of file to upload")
|
|
args = parser.parse_args()
|
|
file_type = args.type
|
|
|
|
basepath = os.getcwd()
|
|
uploaded = files.upload() # 上传文件
|
|
assert(file_type in ['zip', 'audio', 'video'])
|
|
if file_type == "zip":
|
|
upload_path = "./custom_character_voice/"
|
|
for filename in uploaded.keys():
|
|
#将上传的文件移动到指定的位置上
|
|
shutil.move(os.path.join(basepath, filename), os.path.join(upload_path, "custom_character_voice.zip"))
|
|
elif file_type == "audio":
|
|
upload_path = "./raw_audio/"
|
|
for filename in uploaded.keys():
|
|
#将上传的文件移动到指定的位置上
|
|
shutil.move(os.path.join(basepath, filename), os.path.join(upload_path, filename))
|
|
elif file_type == "video":
|
|
upload_path = "./video_data/"
|
|
for filename in uploaded.keys():
|
|
# 将上传的文件移动到指定的位置上
|
|
shutil.move(os.path.join(basepath, filename), os.path.join(upload_path, filename)) |