e99882fea5
* Added new source of cpu information Starting with kernel 4.13 /proc/cpuinfo only show the basic clock of the processor not the current frequency. Change adds information from /sys/devices/system/cpu , considering online and offline cpu's. * PEP8 fixes * Fix cpus_online_range
71 lines
2.8 KiB
Python
71 lines
2.8 KiB
Python
from i3pystatus import IntervalModule
|
|
|
|
|
|
class CpuFreq(IntervalModule):
|
|
"""
|
|
class uses by default `/proc/cpuinfo` to determine the current cpu frequency
|
|
|
|
.. rubric:: Available formatters
|
|
|
|
* `{avg}` - mean from all cores in MHz `4.3f`
|
|
* `{avgg}` - mean from all cores in GHz `1.2f`
|
|
* `{coreX}` - frequency of core number `X` in MHz (format `4.3f`), where 0 <= `X` <= number of cores - 1
|
|
* `{coreXg}` - frequency of core number `X` in GHz (fromat `1.2f`), where 0 <= `X` <= number of cores - 1
|
|
|
|
"""
|
|
format = "{avgg}"
|
|
settings = (
|
|
"format",
|
|
("color", "The text color"),
|
|
("file", "override default path"),
|
|
)
|
|
|
|
file = '/proc/cpuinfo'
|
|
color = '#FFFFFF'
|
|
|
|
def createvaluesdict(self):
|
|
"""
|
|
function processes the /proc/cpuinfo file, use file=/sys to use kernel >=4.13 location
|
|
:return: dictionary used as the full-text output for the module
|
|
"""
|
|
cpus_offline = 0
|
|
if self.file == '/sys':
|
|
with open('/sys/devices/system/cpu/online') as f:
|
|
line = f.readline()
|
|
cpus_online = [int(cpu) for cpu in line.split(',') if cpu.find('-') < 0]
|
|
cpus_online_range = [cpu_range for cpu_range in line.split(',') if cpu_range.find('-') > 0]
|
|
|
|
for cpu_range in cpus_online_range:
|
|
cpus_online += [cpu for cpu in range(int(cpu_range.split('-')[0]), int(cpu_range.split('-')[1]) + 1)]
|
|
|
|
mhz_values = [0.0 for cpu in range(max(cpus_online) + 1)]
|
|
ghz_values = [0.0 for cpu in range(max(cpus_online) + 1)]
|
|
for cpu in cpus_online:
|
|
with open('/sys/devices/system/cpu/cpu{}/cpufreq/scaling_cur_freq'.format(cpu)) as f:
|
|
line = f.readline()
|
|
mhz_values[cpu] = float(line.rstrip()) / 1000.0
|
|
ghz_values[cpu] = float(line.rstrip()) / 1000000.0
|
|
cpus_offline = mhz_values.count(0.0)
|
|
else:
|
|
with open(self.file) as f:
|
|
mhz_values = [float(line.split(':')[1]) for line in f if line.startswith('cpu MHz')]
|
|
ghz_values = [value / 1000.0 for value in mhz_values]
|
|
|
|
mhz = {"core{}".format(key): "{0:4.3f}".format(value) for key, value in enumerate(mhz_values)}
|
|
ghz = {"core{}g".format(key): "{0:1.2f}".format(value) for key, value in enumerate(ghz_values)}
|
|
cdict = mhz.copy()
|
|
cdict.update(ghz)
|
|
cdict['avg'] = "{0:4.3f}".format(sum(mhz_values) / (len(mhz_values) - cpus_offline))
|
|
cdict['avgg'] = "{0:1.2f}".format(sum(ghz_values) / (len(ghz_values) - cpus_offline), 2)
|
|
return cdict
|
|
|
|
def run(self):
|
|
cdict = self.createvaluesdict()
|
|
|
|
self.data = cdict
|
|
self.output = {
|
|
"full_text": self.format.format(**cdict),
|
|
"color": self.color,
|
|
"format": self.format,
|
|
}
|