python linux磁盘空间_python linux 磁盘操作

#coding:utf-8

'''

__author__ = 'similarface'

connection:841196883@qq.com

磁盘操作

'''

import psutil

import os

import commands

import logging

import re

logger = logging.getLogger('Disk')

logger.setLevel(logging.ERROR)

ch = logging.StreamHandler()

# 定义handler的输出格式formatter

formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')

ch.setFormatter(formatter)

logger.addHandler(ch)

from collections import defaultdict

class Disk:

def getDiskSize(self,part):

'''

获取磁盘分区使用信息

'''

return psutil.disk_usage(part)

def getAllCanUseUUID(self):

'''

获取所有可用磁盘的UUID

'''

allcanuse=[]

canusepart=self.getAllCanUseDiskPart()

for part in canusepart:

allcanuse.append(self.getUUIDbyPartName(part))

return allcanuse

def getMountStatus(self):

'''

sdiskpart(device='/dev/sda3', mountpoint='/', fstype='ext4', opts='rw'),

...

sdiskpart(device='/dev/sde1', mountpoint='/data3', fstype='ext4', opts='rw')]

:return:

'''

psutil.disk_partitions()

def getMountPartList(self):

'''

获取挂载了的磁盘的NAME

:return:

'''

partlist=[]

try:

for item in psutil.disk_partitions():

partlist.append(item.device)

return partlist

except Exception,e:

return partlist

def getMountUUID(self):

'''

获取挂载了的磁盘的UUID

:return:

'''

mountuuidlist=[]

for part in self.getMountPartList():

mountuuidlist.append(self.getUUIDbyPartName(part))

return mountuuidlist

def getNoMountDisk(self):

'''

获取没有挂载的磁盘

实现方法：

所有的UUID-挂载了的UUID

:return:

'''

#所有的UUID

ALLUUID=self.getAllCanUseUUID()

#挂载了的UUID

MOUNTUUID=self.getMountUUID()

logger.info(ALLUUID)

logger.info(MOUNTUUID)

return list(set(ALLUUID)-set(MOUNTUUID))

def getPartListNameByUUIDList(self,uuidlist):

'''

根据uuidlist获取partlist

:param uuidlist:

:return:

'''

partlist=[]

for uuid in uuidlist:

partlist.append(self.getPartNameByUUID(uuid))

return partlist

def getPartNameByUUID(self,uuid):

'''

根据UUID 获取partname

:param uuid:

:return: part name

'''

logger.info(uuid)

return self.getUUIDPartNameMap()[uuid]

def getAllCanUseDiskPart(self):

'''

获取所有可用的磁盘的分区名称

:return:

'''

command_str="cat /proc/partitions"

#可用分区

partcanuserlist=[]

try:

status, result=commands.getstatusoutput(command_str)

if status==0:

rereobj = re.compile('\s*')

for item in result.split('\n'):

if not item.startswith("major") and item.strip():

major, minor, blocks,name=rereobj.split(item.strip())

if self.isCanUserPart(major, minor, blocks):

partcanuserlist.append('/dev/'+name)

return partcanuserlist

else:

return partcanuserlist

except Exception,e:

logger.error(e)

return partcanuserlist

def isCanUserPart(self,major, minor, blocks):

'''

该分区是否可用

算法：设备8 子计数 块大于204800

:param itemlist:['major', 'minor', '#blocks', 'name']

:return:是否可用

'''

try:

if int(major)==8 and int(minor)%2==1 and int(blocks)>1:

return True

else:

return False

except Exception,e:

return False

def get(self):

psutil.disk_partitions()

def getUUIDPartNameMap(self):

'''

获取UUID 和 分区名的对应关系

key:UUID

value:partname

:return:

'''

uuidpartmap={}

command_str="ls /dev/disk/by-uuid/ -l|grep -v \"total\"|awk '{ print $9,\"/dev\"substr($11,6,5) }'"

try:

status, result=commands.getstatusoutput(command_str)

if status==0:

for item in result.split('\n'):

uuid,parted=item.split(' ')

uuidpartmap[uuid]=parted

logger.info(uuidpartmap)

return uuidpartmap

else:

logger.error(result)

return uuidpartmap

except Exception,e:

logger.error(e)

return uuidpartmap

def getUUIDbyPartName(self,part):

'''

根据分区名获取分区的UUID

:param part:

:return:该分区对应的UUID

'''

command_str='blkid '+part

try:

logger.info(command_str)

status, result=commands.getstatusoutput(command_str)

if status==0:

logger.debug("result:")

logger.debug(result)

for item in result.strip().split(' '):

if item.startswith("UUID"):

logger.debug(item)

return item.split('=')[1].replace('"','')

else:

return None

except Exception,e:

print(e)

return None

def getPartUUIDNameMap(self):

'''

key:Part

value:UUID

:return:分区－UUID 的字典

'''

uuidpartmap=defaultdict(str)

command_str="ls /dev/disk/by-uuid/ -l|grep -v \"total\"|awk '{ print $9,\"/dev\"substr($11,6,5) }'"

try:

status, result=commands.getstatusoutput(command_str)

if status==0:

for item in result.split('\n'):

uuid,parted=item.split(' ')

uuidpartmap[parted]=uuid

return uuidpartmap

else:

logger.error(result)

return uuidpartmap

except Exception,e:

logger.error(e)

return uuidpartmap

if __name__=='__main__':

a=Disk()

#打印没有挂载的磁盘的分区名称

print(a.getPartNameByUUID(a.getNoMountDisk()[0]))