Python：写脚本常用模块OS基础用法

Python：写脚本常用模块OS基础用法

收集了一些关于OS库的用法,整理归纳一下,方便使用

import os

# 系统操作
print(os.sep)             # 获取当前系统的路径分隔符
print(os.name)            # 获取当前使用的工作平台
print(os.getenv('PATH'))  # 获取名为 PATH 的环境变量
print(os.getcwd())        # 获取当前的路径
print(os.environ['PATH']) # 可以返回环境相关的信息  不传参时,以字典的方式返回所有环境变量

# 调用系统命令
os.system(command)  # 将linux命令传入这里，就可以执行 Execute the command in a subshell. 

# 目录操作 - 增删改查
dir = "/opt/"
listdir_opt = os.listdir(dir) # 返回指定目录下的所有文件何目录名
print(listdir_opt)  
os.mkdir("/opt/os-make/") # 创建一个目录，只创建一个目录文件
os.mknod("/root/python-test.txt") # 创建一个文件
os.rmdir("/opt/ooo/") # 删除一个空目录，若目录中有文件则无法删除
os.remove("/tmp/test.txt") # 用于删除文件，若是目录，则抛出 IsDirectoryError 异常
os.makedirs("/opt/os-make-again/os-make-again-again") # 可以生成多层递归目录，如果目录全部存在，则创建目录失败
os.removedirs() # 从最下级目录开始，逐级删除指定路径，遇到非空目录即停止
os.chdir("/tmp/") # 改变当前目录，到指定目录
os.rename("/opt/ooo/","/opt/AAA/") # 重命名目录名或者文件名。重命名后的文件已存在，则重命名失败。
"""
os.rename()函数的作用是将文件或路径重命名，一般调用格式为os.rename(src, dst)，即将src指向的文件或路径重命名为dst指定的名称。

注意，如果指定的目标路径在其他目录下，该函数还可实现文件或路径的“剪切并粘贴”功能。但无论直接原地重命名还是“剪切粘贴”，中间路径都必须要存在，否则就会抛出FileNotFoundError异常。如果目标路径已存在，Windows 下会抛出FileExistsError异常；Linux 下，如果目标路径为空且用户权限允许，则会静默覆盖原路径，否则抛出OSError异常，
和上两个函数一样，该函数也有对应的递归版本os.renames()，能够创建缺失的中间路径。

注意，这两种情况下，如果函数执行成功，都会调用os.removedir()函数来递归删除源路径的最下级目录。
"""



# 判断
if os.path.exists("/root"):
  print("/root 目录存在！")

if os.path.isfile("/root"):
  print("/root 文件存在！")

if os.path.isdir("/etc"):
  print("/etc 目录存在！")

if os.path.isabs("/etc"):
  print("/etc 是绝对路径！")




# path模块
"""
os.path中的函数基本上是纯粹的字符串操作。换句话说，传入该模块函数的参数甚至不需要是一个有效路径，该模块也不会试图访问这个路径，而仅仅是按照“路径”的通用格式对字符串进行处理。
"""
path = "/etc/passwd"
filename = os.path.basename(path)  # 返回文件名，如果是目录则为空 实际上是传入路径最后一个分隔符之后的子字符串，也就是说，如果最下级目录之后还有一个分隔符，得到的就会是一个空字符串
filedir = os.path.dirname(path)  # 返回的是最后一个分隔符前的整个字符串
filesplit = os.path.split(path)  # 将传入路径以最后一个分隔符为界，分成两个字符串，并打包成元组的形式返回
"""
类似的
os.path.splitext("ooo.txt")
('ooo', '.txt')
"""
filesize = os.path.getsize(path) # 获取文件的大小 相当于 ls -l 单位为bytes
fileAbsPath = os.path.abspath(path) # 获取文件的绝对路径
filejoin = os.path.join(path,"test.txt") # 拼接新的路径
"""
如果传入路径中存在一个“绝对路径”格式的字符串，且这个字符串不是函数的第一个参数，那么其他在这个参数之前的所有参数都会被丢弃，余下的参数再进行组合。更准确地说，只有最后一个“绝对路径”及其之后的参数才会体现在返回结果中。

例子如下:

os.path.join("just", "do", "/opt/", "it")
结果： /opt/it
os.path.join("just", "do", "/opt/", "python", "dot", "/root", "com")
结果：/root/com
"""

print(filename+"\n"+filedir+"\n"+str(filesize)+"\n"+fileAbsPath+"\n"+filejoin+"\n")

以上模块和函数的用法已经做了一些说明,理解起来也不难,下面再介绍一个函数os.walk()

import os

for item in os.walk("/opt/test-walk/"):
    print(item)

程序输出结果

('/opt/test-walk/', ['a', 'b', 'c'], [])
('/opt/test-walk/a', [], ['a.txt'])
('/opt/test-walk/b', ['b2'], [])
('/opt/test-walk/b/b2', [], ['b.txt'])
('/opt/test-walk/c', [], [])

目录结构

[root@open-1 python_scripts]# tree /opt/test-walk/
/opt/test-walk/
├── a
│   └── a.txt
├── b
│   └── b2
│       └── b.txt
└── c

4 directories, 2 files

由上面的结果可以大致明白os.walk()函数的作用:这个函数需要传入一个路径作为参数，函数的作用是在该路径为根节点的目录树中游走，对树中的每个目录生成一个由(dirpath, dirnames, filenames)三项组成的三元组。其中，dirpath是一个指示这个目录路径的字符串，dirnames是一个dirpath下子目录名（除去 . 和 ..）组成的列表，filenames则是由dirpath下所有非目录的文件名组成的列表。简单来说,就是把目标路径下的所有目录和文件都列出来,结合tree命令的结果,就更好理解这个函数的作用了.