You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
138 lines
2.8 KiB
138 lines
2.8 KiB
"""
|
|
工具函数模块
|
|
提供文件读写、路径处理等辅助功能
|
|
"""
|
|
import os
|
|
import shutil
|
|
from typing import Optional
|
|
|
|
|
|
def read_file_bytes(filepath: str) -> bytes:
|
|
"""
|
|
读取文件内容（二进制）
|
|
|
|
Args:
|
|
filepath: 文件路径
|
|
|
|
Returns:
|
|
bytes: 文件内容
|
|
"""
|
|
with open(filepath, 'rb') as f:
|
|
return f.read()
|
|
|
|
|
|
def write_file_bytes(filepath: str, data: bytes):
|
|
"""
|
|
写入文件内容（二进制）
|
|
|
|
Args:
|
|
filepath: 文件路径
|
|
data: 要写入的数据
|
|
"""
|
|
os.makedirs(os.path.dirname(filepath) or '.', exist_ok=True)
|
|
with open(filepath, 'wb') as f:
|
|
f.write(data)
|
|
|
|
|
|
def read_file_text(filepath: str, encoding: str = 'utf-8') -> str:
|
|
"""
|
|
读取文件内容（文本）
|
|
|
|
Args:
|
|
filepath: 文件路径
|
|
encoding: 编码格式
|
|
|
|
Returns:
|
|
str: 文件内容
|
|
"""
|
|
with open(filepath, 'r', encoding=encoding) as f:
|
|
return f.read()
|
|
|
|
|
|
def write_file_text(filepath: str, text: str, encoding: str = 'utf-8'):
|
|
"""
|
|
写入文件内容（文本）
|
|
|
|
Args:
|
|
filepath: 文件路径
|
|
text: 要写入的文本
|
|
encoding: 编码格式
|
|
"""
|
|
os.makedirs(os.path.dirname(filepath) or '.', exist_ok=True)
|
|
with open(filepath, 'w', encoding=encoding) as f:
|
|
f.write(text)
|
|
|
|
|
|
def get_file_size(filepath: str) -> int:
|
|
"""
|
|
获取文件大小
|
|
|
|
Args:
|
|
filepath: 文件路径
|
|
|
|
Returns:
|
|
int: 文件大小（字节）
|
|
"""
|
|
return os.path.getsize(filepath)
|
|
|
|
|
|
def file_exists(filepath: str) -> bool:
|
|
"""
|
|
检查文件是否存在
|
|
|
|
Args:
|
|
filepath: 文件路径
|
|
|
|
Returns:
|
|
bool: 文件是否存在
|
|
"""
|
|
return os.path.isfile(filepath)
|
|
|
|
|
|
def ensure_dir(dirpath: str):
|
|
"""
|
|
确保目录存在，不存在则创建
|
|
|
|
Args:
|
|
dirpath: 目录路径
|
|
"""
|
|
os.makedirs(dirpath, exist_ok=True)
|
|
|
|
|
|
def copy_file(src: str, dst: str):
|
|
"""
|
|
复制文件
|
|
|
|
Args:
|
|
src: 源文件路径
|
|
dst: 目标文件路径
|
|
"""
|
|
os.makedirs(os.path.dirname(dst) or '.', exist_ok=True)
|
|
shutil.copy2(src, dst)
|
|
|
|
|
|
def get_file_extension(filepath: str) -> str:
|
|
"""
|
|
获取文件扩展名
|
|
|
|
Args:
|
|
filepath: 文件路径
|
|
|
|
Returns:
|
|
str: 文件扩展名（含点）
|
|
"""
|
|
return os.path.splitext(filepath)[1]
|
|
|
|
|
|
def get_filename_without_ext(filepath: str) -> str:
|
|
"""
|
|
获取不带扩展名的文件名
|
|
|
|
Args:
|
|
filepath: 文件路径
|
|
|
|
Returns:
|
|
str: 不带扩展名的文件名
|
|
"""
|
|
return os.path.splitext(os.path.basename(filepath))[0]
|