utils.py•2.36 kB
import tempfile
import zipfile
import os
def convert_to_seconds(time_input):
"""
将不同格式的时间表示转换为秒数。
参数:
time_input: 可以是以下格式之一：
- str: 格式为 'HH:MM:SS', 'MM:SS' 或 'SS'.
- tuple: 格式为 (HH, MM, SS).
- int or float: 直接表示秒数.
返回:
float: 转换后的秒数表示。
异常:
ValueError: 如果输入格式不被识别。
"""
if isinstance(time_input, (int, float)):
return float(time_input)
if isinstance(time_input, str):
parts = time_input.split(':')
parts = [float(p) for p in parts]
if len(parts) == 3:
hours, minutes, seconds = parts
elif len(parts) == 2:
hours = 0
minutes, seconds = parts
elif len(parts) == 1:
hours = 0
minutes = 0
seconds = parts[0]
else:
raise ValueError(f"Unrecognized time string format: {time_input}")
return hours * 3600 + minutes * 60 + seconds
if isinstance(time_input, tuple):
if len(time_input) == 3:
hours, minutes, seconds = time_input
elif len(time_input) == 2:
hours = 0
minutes, seconds = time_input
elif len(time_input) == 1:
hours = 0
minutes = 0
seconds = time_input[0]
else:
raise ValueError(f"Unrecognized time tuple format: {time_input}")
return hours * 3600 + minutes * 60 + seconds
raise ValueError(f"Unrecognized time input type: {type(time_input)}")
def create_temp_file() -> str:
"""
创建临时文件并返回其路径。
返回:
str: 临时文件的路径
"""
# 使用NamedTemporaryFile创建临时文件
temp_file = tempfile.NamedTemporaryFile(delete=False)
# 获取临时文件的路径
temp_file_path = temp_file.name
# 关闭文件，以便其他进程或程序可以使用它
temp_file.close()
return temp_file_path
def unzip_to_current_directory(zip_file_path):
# 获取当前目录路径
current_directory = os.getcwd()
# 打开ZIP文件
with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
# 解压缩到当前目录
zip_ref.extractall(current_directory)