用Python寫運維腳本時,經常需要執行linux shell的命令,Python中的commands模塊專門用于調用Linux shell命令,并返回狀態和結果,下面是commands模塊的3個主要函數:
1. commands.getoutput('shell command')
執行shell命令,返回結果(string類型)
>>> commands.getoutput('pwd')
'/home/oracle'
2. commands.getstatus('file')
該函數已被python丟棄,不建議使用,它返回 ls -ld file 的結果(String)(返回結果太奇怪了,難怪被丟棄)
>>> commands.getstatus('admin.tar')
'-rw-rw-r-- 1 oracle oracle 829440 Jan 29 10:36 admin.tar'
3. commands.getstatusoutput('shell command')
執行shell命令, 返回兩個元素的元組tuple(status, result),status為int類型,result為string類型。
cmd的執行方式是{ cmd ; } 2>&1, 故返回結果包含標準輸出和標準錯誤.
>>> commands.getstatusoutput('pwd')
(0, '/home/oracle')
下面的一個腳本利用commands模塊檢測磁盤使用率,標識出大于10%的磁盤(百分比可根據實際情況調整,一般設為90%,本例為了更好的說明情況,設為10%):
import commandsthreshold = 10flag = Falsetitle=commands.getoutput("df -h|head -1")'''Check sda disk space usage like below format:/dev/sda2 20G 2.3G 17G 13% //dev/sda6 20G 306M 19G 2% /var/dev/sda3 49G 2.8G 44G 7% /home/dev/sda5 49G 4.5G 42G 10% /opt/dev/sda1 194M 12M 172M 7% /boot'''chkDiskList=commands.getoutput("df -h|grep sda").split('/n')usedPercents=commands.getoutput("df -h|grep sda|awk '{print $5}'|grep -Eo '[0-9]+'").split('/n')for i in range(0,len(usedPercents)):if int(usedPercents[i]) >= threshold:chkDiskList[i] += ' ----Caution!!! space usage >= ' + str(threshold)flag = True'''Check disk space usage like below format:/dev/mapper/backup-backup_lv751G 14G 699G 2% /backup/dev/mapper/data-data_lv751G 172G 540G 25% /data''' chkDiskList_2=commands.getoutput("df -h|grep -v sda|grep -v tmp|grep -v system").split('/n')usedPercents_2=commands.getoutput("df -h|grep -v map|grep -v sda|grep -v tmp|grep -v system|awk '{print $4}'|grep -Eo '[0-9]+'").split('/n')for i in range(0,len(usedPercents_2)): if int(usedPercents_2[i]) >= threshold:chkDiskList_2[i*2 + 1] += ' ----Caution!!! space usage >= ' + str(threshold)flag = Trueif flag == True:#combine tile, chkDiskList, chkDisklist_2result = [title,]result.extend(chkDiskList)result.extend(chkDiskList_2)for line in result:print line
假設當前的磁盤使用率如下:
[oracle@lx200 ~/admin/python]$ df -hFilesystem Size Used Avail Use% Mounted on/dev/sda2 20G 2.3G 17G 13% //dev/sda6 20G 306M 19G 2% /var/dev/sda3 49G 2.8G 44G 7% /home/dev/sda5 49G 4.5G 42G 10% /opt/dev/sda1 194M 12M 172M 7% /boottmpfs 18G 0 18G 0% /dev/shm/dev/mapper/backup-backup_lv751G 14G 699G 2% /backup/dev/mapper/data-data_lv751G 174G 539G 25% /data
執行該腳本后的結果如下:
Filesystem Size Used Avail Use% Mounted on/dev/sda2 20G 2.3G 17G 13% / ----Caution!!! space usage >= 10/dev/sda6 20G 306M 19G 2% /var/dev/sda3 49G 2.8G 44G 7% /home/dev/sda5 49G 4.5G 42G 10% /opt ----Caution!!! space usage >= 10/dev/sda1 194M 12M 172M 7% /boot/dev/mapper/backup-backup_lv751G 14G 699G 2% /backup/dev/mapper/data-data_lv751G 174G 539G 25% /data ----Caution!!! space usage >= 10
python Commands模塊 使用方法
要獲得shell命令的輸出只需要`cmd`就可以了,
需要得到命令執行的狀態則需要判斷$?的值, 在Python中有一個模塊commands也很容易做到以上的效果.
看一下三個函數:
1). commands.getstatusoutput(cmd)
用os.popen()執行命令cmd, 然后返回兩個元素的元組(status, result). cmd執行的方式是{ cmd ; } 2>&1, 這樣返回結果里面就會包含標準輸出和標準錯誤.
2). commands.getoutput(cmd)
只返回執行的結果, 忽略返回值.
3). commands.getstatus(file)
返回ls -ld file執行的結果.
看一下這些函數使用的例子:
>>> import commands>>> commands.getstatusoutput('ls /bin/ls')(0, '/bin/ls')>>> commands.getstatusoutput('cat /bin/junk')(256, 'cat: /bin/junk: No such file or directory')>>> commands.getstatusoutput('/bin/junk')(256, 'sh: /bin/junk: not found')>>> commands.getoutput('ls /bin/ls')'/bin/ls'>>> commands.getstatus('/bin/ls')'-rwxr-xr-x 1 root 13352 Oct 14 1994 /bin/ls'