系统管理模块

1.shutil模块

(1) 复制和移动

• shutil.copyfilebj(fsrc, fdst[, length])
  • 将类似文件的对象fsrc的内容复制到类似文件的对象fdst
• shutil.copyfile(src, dst, *, follow_symlinks=True)
  • 将名为src的文件的内容(无元数据)复制到名为dst的文件，然后返回dst
• shutil.copy(src, dst, *, follow_symlinks=True)
  • 将文件src复制到文件或目录dst，src和dst应为字符串，如果dst指定目录。则文件将使用src的基本文件名复制到dst中，返回新创建的文件的路径。
• shutil.copy2(src, dst, *, follow_symlinks=True)
  • 与copy()相同，但copy2()也尝试保留所有文件元数据，等同"cp -p"
• shutil.move(src, dst, copy_function=copy2)
  • 递归地将文件或目录(src)移动到另一个位置(dst)，并返回目标

(2) 目录操作

• shutil.copytree(src, dst, symlinks=False, ignore=None, copy_function=copy2, ignore_dangling_symlinks=False)
  • 递归地复制以src为根的整个目录树，返回目标目录。由dst命名的目标目录不能已经存在
• shutil.rmtree(path, ignore_errors=False, onerror=None)
  • 删除整个目录树; 路径必须指向目录(而不是指向目录的符号链接)

(3) 权限管理

• shutil.copymode(src, dst, *, follow_symlinks=True)
  • 将权限位从src复制到dst。文件内容、所有者和组不受影响。src和dst是以字符串形式给出的路径和名称。
• shutil.copystat(src, dst, *, follow_symlinks=True)
  • 将权限位、最后访问时间、上次修改时间和标志从src复制到dst
• shutil.chown(path, user=None, group=None)
  • 更改给定路径的所有者用户和/或组

(4) 案例

import shutil
import os  # 删除单个文件需要导入 os 模块


def cpfobj(src, dst):
    # copyfile值了解底层原理，实际代码不需要使用
    f1 = open(src, 'rb')
    f2 = open(dst, 'wb')
    shutil.copyfileobj(f1, f2)
    f1.close()
    f2.close()


def cpf(src, dst):
    # 只拷贝内容
    shutil.copyfile(src, dst)


def cp(src, dst):
    # 既拷贝内容，又拷贝权限
    shutil.copy(src, dst)


def cp2(src, dst):
    # 相当于cp -p
    shutil.copy2(src, dst)


def mv(src, dst):
    # 相当于mv
    shutil.move(src, dst)


def cptr(src, dst):
    # 需要将目标名也写出来
    shutil.copytree(src, dst)


def rmtr(path):
    # 相当于rm -rf，但只能删除目录
    shutil.rmtree(path)


def rm(path):
    # 删除单个文件
    os.remove(path)


if __name__ == '__main__':
    cpfobj('/bin/ls', '/tmp/list5')
    cpf('/bin/ls', '/tmp/list6')
    cp('/bin/ls', '/tmp/list7')
    cp2('/bin/ls', '/tmp/list8')
    mv('/tmp/list8', '/tmp/list')
    cptr('/etc/security', '/tmp/security')
    mv('/tmp/security', '/var/tmp/auquan')
    rmtr('/var/tmp/auquan')
    rm('/tmp/list5')

(6) python官方手册

https://docs.python.org

2.subprocess模块

(1) 概述

• subprocess模块主要用于调用和执行系统命令
• subprocess模块允许你产生新的进程，连接到它们的输入/输出/错误管道，并获得它们的返回代码
• 本模块旨在替换几个较早的模块和功能，如os.system、os.spawn*

import subprocess
import sys


def subr(com, args):
    for i in args:
        print('\033[031;1m%s %s\033[0m' % (com, i))
        subprocess.run(com+' '+i, shell=True)


if __name__ == '__main__':
    print(sys.argv)
    com = sys.argv[1]
    for i in range(2):
        sys.argv.remove(sys.argv[0])
    args = sys.argv
    # print(args)
    subr(com, args)
    
# subprocess.run的返回值(returncode)为系统的$?

# 捕获输出
import subprocess

rc = subprocess.run('id root; id john', shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
print(rc.returncode)
print(rc.stdout.decode(), end='')
print(rc.stderr.decode(), end='')