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	#!/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	from UserDict import UserDict
11
12	class Error(Exception):
13	pass
14
15	class DiskStat(UserDict):
16
17	def __init__(self, nrread, nrmergedread, nrsectorsread, msread, nrwrite, nrmergedwrite, nrsectorswrite, mswrite, nrio, msio, weightedmsio):
18	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
31
32	class DiskStats:
33
34	def __init__(self, disk):
35	self.disk = disk.lstrip("/dev/")
36
37	def diskstat(self, key=None):
38	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	if not key:
53	return ds
54	else:
55	return ds[key]
56	else:
57	# Need to raise an exception here
58	raise Error("No such disk: %s" % (self.disk))
59
60	if __name__ == "__main__":
61	ds = DiskStats("/dev/sda")
62	print ds.diskstat("msio")