在代码中添加条件判断,以确保不允许将“主目录下载”用于媒体文件。以下是一个示例:
import os
def download_file(file_path):
# 获取主目录路径
home_dir = os.path.expanduser("~")
# 判断文件路径是否在主目录下
if file_path.startswith(home_dir):
# 获取文件扩展名
file_extension = os.path.splitext(file_path)[1]
# 判断文件扩展名是否为媒体文件(可根据需求自行修改)
if file_extension in ['.mp3', '.mp4', '.avi', '.mov']:
raise ValueError("不允许将“主目录下载”用于媒体文件。")
# 其他下载文件的代码逻辑
# ...
在上述示例中,我们使用os.path.expanduser("~")
方法获取主目录的路径,并将其与文件路径进行比较。如果文件路径以主目录路径开头,则判断文件的扩展名是否为媒体文件,如果是,则抛出ValueError
异常。
你可以根据实际需求修改媒体文件的扩展名列表,以适应不同类型的媒体文件。
下一篇:不允许角色从窗户移动