如何在python中实现函数"check_space_taken“？

import os

def check_space_taken(path):
    if not os.path.exists(path):
        return "路径不存在"
    
    if os.path.isfile(path):
        size = os.path.getsize(path)
        return f"文件大小为：{size}字节"
    
    if os.path.isdir(path):
        total_size = 0
        for root, dirs, files in os.walk(path):
            for file in files:
                file_path = os.path.join(root, file)
                total_size += os.path.getsize(file_path)
        return f"目录大小为：{total_size}字节"

# 示例用法
file_path = "path/to/file.txt"
directory_path = "path/to/directory"

print(check_space_taken(file_path))
print(check_space_taken(directory_path))