仿写 Linux 下的 cp 命令

下面是一个简单的 Linux 下的 cp 命令的仿写，它可以将一个文件复制到另一个文件或目录中：

import sys
import shutil
import os

def cp(source, destination):
    """
    将源文件或目录复制到目标位置
    """
    if os.path.isfile(source):
        # 复制文件
        shutil.copy(source, destination)
    elif os.path.isdir(source):
        # 复制目录
        if not os.path.exists(destination):
            os.makedirs(destination)
        for item in os.listdir(source):
            s = os.path.join(source, item)
            d = os.path.join(destination, item)
            cp(s, d)
    else:
        print("错误：源文件或目录不存在！")

if __name__ == '__main__':
    if len(sys.argv) < 3:
        print("用法：python cp.py <源文件或目录> <目标位置>")
        sys.exit(1)

    source = sys.argv[1]
    destination = sys.argv[2]

    cp(source, destination)

这个脚本接受两个参数：源文件或目录和目标位置。如果源文件或目录不存在，则会打印错误信息。如果目标位置不存在，则会自动创建该目录。如果源文件是一个目录，则会递归地复制其下的所有文件和子目录。