如何正確的使用Python 的os庫

本篇文章給大家分享的是有關如何正確的使用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') # 只能刪除空目錄，遞歸刪除可以使用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的路徑

# filename: run.py
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

以上就是如何正確的使用Python 的os庫，小編相信有部分知識點可能是我們日常工作會見到或用到的。希望你能通過這篇文章學到更多知識。更多詳情敬請關注億速云行業資訊頻道。