Python寫腳本常用模塊OS基礎(chǔ)用法詳解

import os

# 系統(tǒng)操作
print(os.sep)       # 獲取當(dāng)前系統(tǒng)的路徑分隔符
print(os.name)      # 獲取當(dāng)前使用的工作平臺
print(os.getenv('PATH')) # 獲取名為 PATH 的環(huán)境變量
print(os.getcwd())    # 獲取當(dāng)前的路徑
print(os.environ['PATH']) # 可以返回環(huán)境相關(guān)的信息 不傳參時(shí),以字典的方式返回所有環(huán)境變量

# 調(diào)用系統(tǒng)命令
os.system(command) # 將linux命令傳入這里，就可以執(zhí)行 Execute the command in a subshell. 

# 目錄操作 - 增刪改查
dir = "/opt/"
listdir_opt = os.listdir(dir) # 返回指定目錄下的所有文件何目錄名
print(listdir_opt) 
os.mkdir("/opt/os-make/") # 創(chuàng)建一個(gè)目錄，只創(chuàng)建一個(gè)目錄文件
os.mknod("/root/python-test.txt") # 創(chuàng)建一個(gè)文件
os.rmdir("/opt/ooo/") # 刪除一個(gè)空目錄，若目錄中有文件則無法刪除
os.remove("/tmp/test.txt") # 用于刪除文件，若是目錄，則拋出 IsDirectoryError 異常
os.makedirs("/opt/os-make-again/os-make-again-again") # 可以生成多層遞歸目錄，如果目錄全部存在，則創(chuàng)建目錄失敗
os.removedirs() # 從最下級目錄開始，逐級刪除指定路徑，遇到非空目錄即停止
os.chdir("/tmp/") # 改變當(dāng)前目錄，到指定目錄
os.rename("/opt/ooo/","/opt/AAA/") # 重命名目錄名或者文件名。重命名后的文件已存在，則重命名失敗。
"""
os.rename()函數(shù)的作用是將文件或路徑重命名，一般調(diào)用格式為os.rename(src, dst)，即將src指向的文件或路徑重命名為dst指定的名稱。

注意，如果指定的目標(biāo)路徑在其他目錄下，該函數(shù)還可實(shí)現(xiàn)文件或路徑的“剪切并粘貼”功能。但無論直接原地重命名還是“剪切粘貼”，中間路徑都必須要存在，否則就會拋出FileNotFoundError異常。如果目標(biāo)路徑已存在，Windows 下會拋出FileExistsError異常；Linux 下，如果目標(biāo)路徑為空且用戶權(quán)限允許，則會靜默覆蓋原路徑，否則拋出OSError異常，
和上兩個(gè)函數(shù)一樣，該函數(shù)也有對應(yīng)的遞歸版本os.renames()，能夠創(chuàng)建缺失的中間路徑。

注意，這兩種情況下，如果函數(shù)執(zhí)行成功，都會調(diào)用os.removedir()函數(shù)來遞歸刪除源路徑的最下級目錄。
"""

# 判斷
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中的函數(shù)基本上是純粹的字符串操作。換句話說，傳入該模塊函數(shù)的參數(shù)甚至不需要是一個(gè)有效路徑，該模塊也不會試圖訪問這個(gè)路徑，而僅僅是按照“路徑”的通用格式對字符串進(jìn)行處理。
"""
path = "/etc/passwd"
filename = os.path.basename(path) # 返回文件名，如果是目錄則為空 實(shí)際上是傳入路徑最后一個(gè)分隔符之后的子字符串，也就是說，如果最下級目錄之后還有一個(gè)分隔符，得到的就會是一個(gè)空字符串
filedir = os.path.dirname(path) # 返回的是最后一個(gè)分隔符前的整個(gè)字符串
filesplit = os.path.split(path) # 將傳入路徑以最后一個(gè)分隔符為界，分成兩個(gè)字符串，并打包成元組的形式返回
"""
類似的
os.path.splitext("ooo.txt")
('ooo', '.txt')
"""
filesize = os.path.getsize(path) # 獲取文件的大小 相當(dāng)于 ls -l 單位為bytes
fileAbsPath = os.path.abspath(path) # 獲取文件的絕對路徑
filejoin = os.path.join(path,"test.txt") # 拼接新的路徑
"""
如果傳入路徑中存在一個(gè)“絕對路徑”格式的字符串，且這個(gè)字符串不是函數(shù)的第一個(gè)參數(shù)，那么其他在這個(gè)參數(shù)之前的所有參數(shù)都會被丟棄，余下的參數(shù)再進(jìn)行組合。更準(zhǔn)確地說，只有最后一個(gè)“絕對路徑”及其之后的參數(shù)才會體現(xiàn)在返回結(jié)果中。

例子如下:

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

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

import os

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

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

4 directories, 2 files

import random
import string
import os

def gen_code(len=4):
  # 隨機(jī)生成4位驗(yàn)證碼
  li = random.sample(string.ascii_letters+string.digits,len)
  return ''.join(li)
def create_file():
  # 隨機(jī)生成100個(gè)驗(yàn)證碼
  li = {gen_code() for i in range(100)}
  os.mkdir('img')
  for name in li:
    os.mknod('img/' + name + '.png')

create_file()

def modify_suffix(dirname,old_suffix,new_suffix):
  """
  :param dirname:操作的目錄
  :param old_suffix: 之前的后綴名
  :param new_suffix: 新的后綴名
  :return:
  """
  # 1.判斷查找的目錄是否存在，如果不存在，報(bào)錯
  if os.path.exists(dirname):
    # 2.找出所有以old_suffix(.png)結(jié)尾的文件
    pngfile = [filename for filename in os.listdir(dirname)
          if filename.endswith(old_suffix)]
    # 3.將后綴名和文件名分開，留下文件名
    basefiles = [os.path.splitext(filename)[0]
           for filename in pngfile]
    # 4.重命名文件
    for filename in basefiles:
      oldname = os.path.join(dirname,filename+old_suffix)
      newname = os.path.join(dirname,filename+new_suffix)
      os.rename(oldname,newname)
      print('%s命名為%s成功' %(oldname,newname))
  else:
    print('%s不存在,不能操作...' %(dirname))

modify_suffix('redhat','.jpg','.png')

import os
import time

time1 = os.path.getctime('/etc/shadow')   #時(shí)間戳?xí)r間
print(time1)
tuple_time = time.localtime(time1)
print(tuple_time)
year = tuple_time.tm_year
month = tuple_time.tm_mon
day = tuple_time.tm_mday

with open('date.txt','a') as f:
  f.write('%d %d %d' %(year,month,day))
  f.write('\n')