【Python教程】常用的os操作详情

太过爱你忘了你带给我的痛 2022-09-11 01:29 120阅读 0赞

Python自动的os库是和操作系统交互的库，常用的操作包括文件/目录操作，路径操作，环境变量操作和执行系统命令等。

**文件/目录操作**

*  获取当前目录(pwd): os.getcwd()
 *  切换目录(cd): os.chdir(’/usr/local/’)
 *  列出目录所有文件(ls)：os.listdir(’/usr/local/’)
 *  创建目录(mkdir)：os.makedirs(’/usr/local/tmp’)
 *  删除目录(rmdir)：os.removedirs(’/usr/local/tmp’) \# 只能删除空目录，递归删除可以使用import shutil;shutil.rmtree(’/usr/local/tmp’)
 *  删除文件(rm)：os.remove(’/usr/local/a.txt’)
 *  递归遍历目录及子目录：os.walk()

示例：遍历/usr/local目录及子下所有文件和目录，并组装出每个文件完整的路径名

import os
    for root, dirs, files in os.walk("/usr/local", topdown=False):
        for name in files:
            print('文件：', os.path.join(root, name))
        for name in dirs:
            print('目录：', os.path.join(root, name))

**路径操作**

*  当前Python脚本文件：`__file__`
 *  获取文件所在路径：`os.path.basename(__file__) # 不含当前文件名`
 *  获取文件绝对路径：`os.path.abspath(__file__) # 包含当前文件名`
 *  获取所在目录路径：`os.path.dirname(__file__)`
 *  分割路径和文件名：`os.path.split('/usr/local/a.txt') # 得到一个[路径,文件名]的列表`
 *  分割文件名和扩展名：`os.path.splitext('a.txt') # 得到['a', '.txt']`
 *  判断路径是否存在：`os.path.exists('/usr/local/a.txt')`
 *  判断路径是否文件：`os.path.isfile('/usr/local/a.txt')`
 *  判断路径是否目录：`os.path.isdir('/usr/local/a.txt')`
 *  组装路径：`os.path.join('/usr', 'local', 'a.txt')`

示例：获取项目根路径和报告文件路径

假设项目结构如下

project/
      data'
      reports/
        report.html
      testcases/
      config.py
      run.py

在run.py中获取项目的路径和report.html的路径

''' 学习中遇到问题没人解答？小编创建了一个Python学习交流QQ群：531509025 寻找有志同道合的小伙伴，互帮互助,群里还有不错的视频学习教程和PDF电子书！ '''
    import os
    
    base_dir = os.path.dirname(__file__)  # __file__是run.py文件，os.path.dirname获取到其所在的目录project即项目根路径
    report_file = os.path.join(base_dir, 'reports', 'report.html')  # 使用系统路径分隔符('\')连接项目根目录base_dir和'reports'及'report.html'得到报告路径
    print(report_file)

**环境变量操作**

*  获取环境变量：`os.environ.get('PATH')或os.getenv('PATH')`
 *  设置环境变量：`os.environ['MYSQL_PWD']='123456'`

**执行系统命令**

执行系统命令：`os.system("jmeter -n -t /usr/local/demo.jmx") # 无法获取屏幕输出的信息，相要获取运行屏幕信息，可以使用subprocess`