1d26dcbc81339457d0d11e7b0135c53f1b4cd25e
[collecty.git] / src / collecty / plugins / cpu.py
1 #!/usr/bin/python
2 ###############################################################################
3 #                                                                             #
4 # collecty - A system statistics collection daemon for IPFire                 #
5 # Copyright (C) 2012 IPFire development team                                  #
6 #                                                                             #
7 # This program is free software: you can redistribute it and/or modify        #
8 # it under the terms of the GNU General Public License as published by        #
9 # the Free Software Foundation, either version 3 of the License, or           #
10 # (at your option) any later version.                                         #
11 #                                                                             #
12 # This program is distributed in the hope that it will be useful,             #
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of              #
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
15 # GNU General Public License for more details.                                #
16 #                                                                             #
17 # You should have received a copy of the GNU General Public License           #
18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
19 #                                                                             #
20 ###############################################################################
21
22 from __future__ import division
23
24 import base
25
26 from ..i18n import _
27
28 class GraphTemplateCPU(base.GraphTemplate):
29         name = "cpu"
30
31         rrd_graph = [
32                 "DEF:user=%(file)s:user:AVERAGE",
33                 "DEF:nice=%(file)s:nice:AVERAGE",
34                 "DEF:sys=%(file)s:sys:AVERAGE",
35                 "DEF:idle=%(file)s:idle:AVERAGE",
36                 "DEF:wait=%(file)s:wait:AVERAGE",
37                 "DEF:irq=%(file)s:irq:AVERAGE",
38                 "DEF:sirq=%(file)s:sirq:AVERAGE",
39
40                 "AREA:user#90EE90:%-15s" % _("User"),
41                 "VDEF:usermin=user,MINIMUM",
42                 "VDEF:usermax=user,MAXIMUM",
43                 "VDEF:useravg=user,AVERAGE",
44                 "GPRINT:usermax:%12s\:" % _("Maximum") + " %6.2lf" ,
45                 "GPRINT:usermin:%12s\:" % _("Minimum") + " %6.2lf",
46                 "GPRINT:useravg:%12s\:" % _("Average") + " %6.2lf\\n",
47
48                 "STACK:nice#4169E1:%-15s" % _("Nice"),
49                 "VDEF:nicemin=nice,MINIMUM",
50                 "VDEF:nicemax=nice,MAXIMUM",
51                 "VDEF:niceavg=nice,AVERAGE",
52                 "GPRINT:nicemax:%12s\:" % _("Maximum") + " %6.2lf" ,
53                 "GPRINT:nicemin:%12s\:" % _("Minimum") + " %6.2lf",
54                 "GPRINT:niceavg:%12s\:" % _("Average") + " %6.2lf\\n",
55
56                 "STACK:sys#DC143C:%-15s" % _("System"),
57                 "VDEF:sysmin=sys,MINIMUM",
58                 "VDEF:sysmax=sys,MAXIMUM",
59                 "VDEF:sysavg=sys,AVERAGE",
60                 "GPRINT:sysmax:%12s\:" % _("Maximum") + " %6.2lf" ,
61                 "GPRINT:sysmin:%12s\:" % _("Minimum") + " %6.2lf",
62                 "GPRINT:sysavg:%12s\:" % _("Average") + " %6.2lf\\n",
63
64                 "STACK:wait#483D8B:%-15s" % _("Wait"),
65                 "VDEF:waitmin=wait,MINIMUM",
66                 "VDEF:waitmax=wait,MAXIMUM",
67                 "VDEF:waitavg=wait,AVERAGE",
68                 "GPRINT:waitmax:%12s\:" % _("Maximum") + " %6.2lf" ,
69                 "GPRINT:waitmin:%12s\:" % _("Minimum") + " %6.2lf",
70                 "GPRINT:waitavg:%12s\:" % _("Average") + " %6.2lf\\n",
71
72                 "STACK:irq#DAA520:%-15s" % _("Interrupt"),
73                 "VDEF:irqmin=irq,MINIMUM",
74                 "VDEF:irqmax=irq,MAXIMUM",
75                 "VDEF:irqavg=irq,AVERAGE",
76                 "GPRINT:irqmax:%12s\:" % _("Maximum") + " %6.2lf" ,
77                 "GPRINT:irqmin:%12s\:" % _("Minimum") + " %6.2lf",
78                 "GPRINT:irqavg:%12s\:" % _("Average") + " %6.2lf\\n",
79
80                 "STACK:sirq#FFD700:%-15s" % _("Soft interrupt"),
81                 "VDEF:sirqmin=sirq,MINIMUM",
82                 "VDEF:sirqmax=sirq,MAXIMUM",
83                 "VDEF:sirqavg=sirq,AVERAGE",
84                 "GPRINT:sirqmax:%12s\:" % _("Maximum") + " %6.2lf" ,
85                 "GPRINT:sirqmin:%12s\:" % _("Minimum") + " %6.2lf",
86                 "GPRINT:sirqavg:%12s\:" % _("Average") + " %6.2lf\\n",
87
88                 "STACK:idle#EFEFEF:%-15s" % _("Idle"),
89                 "VDEF:idlemin=idle,MINIMUM",
90                 "VDEF:idlemax=idle,MAXIMUM",
91                 "VDEF:idleavg=idle,AVERAGE",
92                 "GPRINT:idlemax:%12s\:" % _("Maximum") + " %6.2lf" ,
93                 "GPRINT:idlemin:%12s\:" % _("Minimum") + " %6.2lf",
94                 "GPRINT:idleavg:%12s\:" % _("Average") + " %6.2lf\\n",
95         ]
96
97         rrd_graph_args = [
98                 "--title", _("CPU usage"),
99                 "--vertical-label", _("Jiffies"),
100
101                 "--lower-limit", "0", "--rigid",
102         ]
103
104
105 class ProcessorPlugin(base.Plugin):
106         name = "cpu"
107         description = "CPU Usage Data Source"
108
109         templates = [GraphTemplateCPU,]
110
111         rrd_schema = [
112                 "DS:user:DERIVE:0:U",
113                 "DS:nice:DERIVE:0:U",
114                 "DS:sys:DERIVE:0:U",
115                 "DS:idle:DERIVE:0:U",
116                 "DS:wait:DERIVE:0:U",
117                 "DS:irq:DERIVE:0:U",
118                 "DS:sirq:DERIVE:0:U",
119         ]
120
121         @classmethod
122         def autocreate(cls, collecty, **kwargs):
123                 # Every system has got at least one CPU.
124                 return cls(collecty, **kwargs)
125
126         def read(self):
127                 """
128                         Reads the CPU usage.
129                 """
130                 f = None
131
132                 try:
133                         f = open("/proc/stat")
134
135                         for line in f.readlines():
136                                 if not line.startswith("cpu"):
137                                         continue
138
139                                 columns = line.split()
140                                 if len(columns) < 8:
141                                         continue
142
143                                 entry = [
144                                         columns[1], # user
145                                         columns[2], # nice
146                                         columns[3], # sys
147                                         columns[4], # idle
148                                         columns[5], # wait
149                                         columns[6], # irq
150                                         columns[7], # sirq
151                                 ]
152
153                                 return ":".join(entry)
154                 finally:
155                         if f:
156                                 f.close()