Open all files in a with statement block

[collecty.git] / src / collecty / plugins / processor.py

#!/usr/bin/python3
###############################################################################
#                                                                             #
# collecty - A system statistics collection daemon for IPFire                 #
# Copyright (C) 2012 IPFire development team                                  #
#                                                                             #
# This program is free software: you can redistribute it and/or modify        #
# it under the terms of the GNU General Public License as published by        #
# the Free Software Foundation, either version 3 of the License, or           #
# (at your option) any later version.                                         #
#                                                                             #
# This program is distributed in the hope that it will be useful,             #
# but WITHOUT ANY WARRANTY; without even the implied warranty of              #
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
# GNU General Public License for more details.                                #
#                                                                             #
# You should have received a copy of the GNU General Public License           #
# along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
#                                                                             #
###############################################################################

from . import base

from ..colours import *

class GraphTemplateProcessor(base.GraphTemplate):
	name = "processor"

	@property
	def rrd_graph(self):
		_ = self.locale.translate

		return [
			"CDEF:total=user,nice,+,sys,+,wait,+,irq,+,sirq,+,idle,+",

			"CDEF:user_p=100,user,*,total,/",
			"AREA:user_p%s:%-15s" % (CPU_USER, _("User")),
			"GPRINT:user_p_max:%12s\:" % _("Maximum") + " %6.2lf%%",
			"GPRINT:user_p_min:%12s\:" % _("Minimum") + " %6.2lf%%",
			"GPRINT:user_p_avg:%12s\:" % _("Average") + " %6.2lf%%\\n",

			"CDEF:nice_p=100,nice,*,total,/",
			"STACK:nice_p%s:%-15s" % (CPU_NICE, _("Nice")),
			"GPRINT:nice_p_max:%12s\:" % _("Maximum") + " %6.2lf%%",
			"GPRINT:nice_p_min:%12s\:" % _("Minimum") + " %6.2lf%%",
			"GPRINT:nice_p_avg:%12s\:" % _("Average") + " %6.2lf%%\\n",

			"CDEF:sys_p=100,sys,*,total,/",
			"STACK:sys_p%s:%-15s" % (CPU_SYS, _("System")),
			"GPRINT:sys_p_max:%12s\:" % _("Maximum") + " %6.2lf%%",
			"GPRINT:sys_p_min:%12s\:" % _("Minimum") + " %6.2lf%%",
			"GPRINT:sys_p_avg:%12s\:" % _("Average") + " %6.2lf%%\\n",

			"CDEF:wait_p=100,wait,*,total,/",
			"STACK:wait_p%s:%-15s" % (CPU_WAIT, _("Wait")),
			"GPRINT:wait_p_max:%12s\:" % _("Maximum") + " %6.2lf%%",
			"GPRINT:wait_p_min:%12s\:" % _("Minimum") + " %6.2lf%%",
			"GPRINT:wait_p_avg:%12s\:" % _("Average") + " %6.2lf%%\\n",

			"CDEF:irq_p=100,irq,*,total,/",
			"STACK:irq_p%s:%-15s" % (CPU_IRQ, _("Interrupt")),
			"GPRINT:irq_p_max:%12s\:" % _("Maximum") + " %6.2lf%%",
			"GPRINT:irq_p_min:%12s\:" % _("Minimum") + " %6.2lf%%",
			"GPRINT:irq_p_avg:%12s\:" % _("Average") + " %6.2lf%%\\n",

			"CDEF:sirq_p=100,sirq,*,total,/",
			"STACK:sirq_p%s:%-15s" % (CPU_SIRQ, _("Soft Interrupt")),
			"GPRINT:sirq_p_max:%12s\:" % _("Maximum") + " %6.2lf%%",
			"GPRINT:sirq_p_min:%12s\:" % _("Minimum") + " %6.2lf%%",
			"GPRINT:sirq_p_avg:%12s\:" % _("Average") + " %6.2lf%%\\n",

			"CDEF:idle_p=100,idle,*,total,/",
			"STACK:idle_p%s:%-15s" % (CPU_IDLE, _("Idle")),
			"GPRINT:idle_p_max:%12s\:" % _("Maximum") + " %6.2lf%%",
			"GPRINT:idle_p_min:%12s\:" % _("Minimum") + " %6.2lf%%",
			"GPRINT:idle_p_avg:%12s\:" % _("Average") + " %6.2lf%%\\n",
		]

	upper_limit = 100
	lower_limit = 0

	@property
	def graph_title(self):
		_ = self.locale.translate
		return _("Processor Usage")

	@property
	def graph_vertical_label(self):
		_ = self.locale.translate
		return _("Percent")


class ProcessorObject(base.Object):
	rrd_schema = [
		"DS:user:DERIVE:0:U",
		"DS:nice:DERIVE:0:U",
		"DS:sys:DERIVE:0:U",
		"DS:idle:DERIVE:0:U",
		"DS:wait:DERIVE:0:U",
		"DS:irq:DERIVE:0:U",
		"DS:sirq:DERIVE:0:U",
	]

	@property
	def id(self):
		return "default"

	def collect(self):
		"""
			Reads the CPU usage.
		"""
		with open("/proc/stat") as f:
			for line in f:
				if not line.startswith("cpu"):
					continue

				columns = line.split()
				if len(columns) < 8:
					continue

				return (
					columns[1], # user
					columns[2], # nice
					columns[3], # sys
					columns[4], # idle
					columns[5], # wait
					columns[6], # irq
					columns[7], # sirq
				)


class ProcessorPlugin(base.Plugin):
	name = "processor"
	description = "Processor Usage Plugin"

	templates = [GraphTemplateProcessor]

	@property
	def objects(self):
		yield ProcessorObject(self)