svn.andrew.net.au/usbspindownd/diskstats.py

#!/usr/bin/python

#
# Python module for parsing /proc/diskstats on Linux
#
# Author: Andrew Pollock <me@andrew.net.au>
# Copyright: (c) 2008 Andrew Pollock
#

from UserDict import UserDict

class Error(Exception):
        pass

class DiskStat(UserDict):

        def __init__(self, nrread, nrmergedread, nrsectorsread, msread, nrwrite, nrmergedwrite, nrsectorswrite, mswrite, nrio, msio, weightedmsio):
                UserDict.__init__(self)
                self["nrread"] = int(nrread)
                self["nrmergedread"] = int(nrmergedread)
                self["nrsectorsread"] = int(nrsectorsread)
                self["msread"] = int(msread)
                self["nrwrite"] = int(nrwrite)
                self["nrmergedwrite"] = int(nrmergedwrite)
                self["nrsectorswrite"] = int(nrsectorswrite)
                self["mswrite"] = int(mswrite)
                self["nrio"] = int(nrio)
                self["msio"] = int(msio)
                self["weightedmsio"] = int(weightedmsio)


class DiskStats:

        def __init__(self, disk):
                self.disk = disk.lstrip("/dev/")

        def diskstat(self, key=None):
                proc_diskstats = open("/proc/diskstats")
                stat = None
                while True:
                        line = proc_diskstats.readline()
                        if line == "":
                                break
                        else:
                                line = line.split()
                        
                        if line[2] == self.disk:
                                stat = line[3:]
                proc_diskstats.close()
                if stat:
                        ds = DiskStat(stat[0], stat[1], stat[2], stat[3], stat[4], stat[5], stat[6], stat[7], stat[8], stat[9], stat[10])
                        if not key:
                                return ds
                        else:
                                return ds[key]
                else:
                        # Need to raise an exception here
                        raise Error("No such disk: %s" % (self.disk))

if __name__ == "__main__":
        ds = DiskStats("/dev/sda")
        print ds.diskstat("msio")
1	apollock	29	#!/usr/bin/python
2
3			#
4			# Python module for parsing /proc/diskstats on Linux
5			#
6			# Author: Andrew Pollock <me@andrew.net.au>
7			# Copyright: (c) 2008 Andrew Pollock
8			#
9
10	apollock	30	from UserDict import UserDict
11	apollock	29
12	apollock	30	class Error(Exception):
13			pass
14
15			class DiskStat(UserDict):
16
17	apollock	29	def __init__(self, nrread, nrmergedread, nrsectorsread, msread, nrwrite, nrmergedwrite, nrsectorswrite, mswrite, nrio, msio, weightedmsio):
18	apollock	30	UserDict.__init__(self)
19			self["nrread"] = int(nrread)
20			self["nrmergedread"] = int(nrmergedread)
21			self["nrsectorsread"] = int(nrsectorsread)
22			self["msread"] = int(msread)
23			self["nrwrite"] = int(nrwrite)
24			self["nrmergedwrite"] = int(nrmergedwrite)
25			self["nrsectorswrite"] = int(nrsectorswrite)
26			self["mswrite"] = int(mswrite)
27			self["nrio"] = int(nrio)
28			self["msio"] = int(msio)
29			self["weightedmsio"] = int(weightedmsio)
30	apollock	29
31
32			class DiskStats:
33
34			def __init__(self, disk):
35			self.disk = disk.lstrip("/dev/")
36
37	apollock	30	def diskstat(self, key=None):
38	apollock	29	proc_diskstats = open("/proc/diskstats")
39			stat = None
40			while True:
41			line = proc_diskstats.readline()
42			if line == "":
43			break
44			else:
45			line = line.split()
46
47			if line[2] == self.disk:
48			stat = line[3:]
49			proc_diskstats.close()
50			if stat:
51			ds = DiskStat(stat[0], stat[1], stat[2], stat[3], stat[4], stat[5], stat[6], stat[7], stat[8], stat[9], stat[10])
52	apollock	30	if not key:
53			return ds
54			else:
55			return ds[key]
56	apollock	29	else:
57			# Need to raise an exception here
58	apollock	30	raise Error("No such disk: %s" % (self.disk))
59	apollock	29
60			if __name__ == "__main__":
61	apollock	30	ds = DiskStats("/dev/sda")
62			print ds.diskstat("msio")