Python os库常用操作代码汇总
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')#只能删除空目录，递归删除可以使用importshutil;shutil.rmtree('/usr/local/tmp')
- 删除文件(rm)：os.remove('/usr/local/a.txt')
- 递归遍历目录及子目录：os.walk()
示例：遍历/usr/local目录及子下所有文件和目录，并组装出每个文件完整的路径名
importos
forroot,dirs,filesinos.walk("/usr/local",topdown=False):
fornameinfiles:
print('文件：',os.path.join(root,name))
fornameindirs:
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的路径
#filename:run.py
importos
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
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。