仿写linux下的cp命令

在 Linux 下，`cp` 命令用于复制文件或目录。以下是一个简单的 Python 脚本，可以实现类似的功能：

import os
import shutil

def cp(src, dest):
    if os.path.isfile(src):
        shutil.copy2(src, dest)
    elif os.path.isdir(src):
        shutil.copytree(src, dest)
    else:
        print(f"{src} doesn't exist or is not a file or directory.")

在这个脚本中，我们使用了 Python 自带的 `os` 和 `shutil` 模块。`os.path.isfile()` 和 `os.path.isdir()` 用于判断文件路径是否为文件或目录；`shutil.copy2()` 和 `shutil.copytree()` 用于复制文件和目录。

使用示例：

cp("file.txt", "new_file.txt")  # 复制文件
cp("dir", "new_dir")  # 复制目录
cp("non_existent_path", "new_path")  # 错误示例