KevinMidboe/Node-Com-Handler
1
0
Fork 0
mirror of https://github.com/KevinMidboe/Node-Com-Handler.git synced 2025-10-29 17:50:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Better error handling and dir setting

Browse Source
This commit is contained in:
Kevin
2017-02-18 11:40:23 +01:00
parent 57869e2d66
commit 53fe6d3ba3
1 changed files with 16 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
status/diskusage.py
View File

@@ -3,15 +3,17 @@
# @Author: KevinMidboe # @Author: KevinMidboe
# @Date: 2017-01-28 10:54:06 # @Date: 2017-01-28 10:54:06
# @Last Modified by: KevinMidboe # @Last Modified by: KevinMidboe
# @Last Modified time: 2017-01-31 22:29:43 # @Last Modified time: 2017-02-18 11:39:00
# f_avail = free blocks avail to non sudo # f_avail = free blocks avail to non sudo
# frsize = fundamental file sys byte size # frsize = fundamental file sys byte size
# f_blocks = total number of blocks in the filesystem # f_blocks = total number of blocks in the filesystem
from os import statvfs from os import statvfs, path
from re import sub from re import sub
# TODO: Add multiple dirs.
def sizeof(num, suffix='B'): def sizeof(num, suffix='B'):
for unit in ['','K','M','G','T','P','E','Z']: for unit in ['','K','M','G','T','P','E','Z']:
if abs(num) < 1024.0: if abs(num) < 1024.0:
@@ -20,24 +22,18 @@ def sizeof(num, suffix='B'):
return "%.1f%s%s" % (num, 'Y', suffix) return "%.1f%s%s" % (num, 'Y', suffix)
# diskUsage uses statvfs to calculate disk size and availability for given disk # diskUsage uses statvfs to calculate disk size and availability for given disk
def diskUsage(optionalPath=None): def diskUsage(rootPath=None):
path = '/' if rootPath == None: rootPath = '/'
if optionalPath != None:
path += sub('\+', '/', optionalPath) if not path.isdir(rootPath):
return {"Error": "Directory does not exist."}
try: s = statvfs(rootPath)
s = statvfs(path) byteLeft = s.f_bavail * s.f_frsize
byteLeft = s.f_bavail * s.f_frsize byteTotal = s.f_blocks * s.f_frsize
byteTotal = s.f_blocks * s.f_frsize byteUsed = byteTotal-byteLeft
byteUsed = byteTotal-byteLeft print(byteLeft, byteTotal, byteUsed)
return { 'left': sizeof(byteLeft), 'used': sizeof(byteUsed), 'total': sizeof(byteTotal) }
return { 'left':sizeof(byteLeft), 'used':sizeof(byteUsed), 'total':sizeof(byteTotal) }
except FileNotFoundError:
return None
except TypeError:
return None
if __name__=="__main__": if __name__=="__main__":
n = diskUsage('/media/hdd1') print(diskUsage())
print(n)