1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
# -*- coding: utf-8 -*-
# sysdata
# Sysdata is a module uses great Py3status (i3status wrapper) to
# display system information (RAM usage) in i3bar (Linux systems).
# For more information read:
# i3wm homepage: http://i3wm.org
# py3status homepage: https://github.com/ultrabug/py3status
# Copyright (C) <2013> <Shahin Azad [ishahinism at Gmail]>
# 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/>.
import subprocess
from time import time
class GetData:
"""Get system status
"""
def execCMD(self, cmd, arg):
"""Take a system command and its argument, then return the result.
Arguments:
- `cmd`: system command.
- `arg`: argument.
"""
result = subprocess.check_output([cmd, arg])
return result
def cpu(self):
"""Get the cpu usage data from /proc/stat :
cpu 2255 34 2290 22625563 6290 127 456 0 0
- user: normal processes executing in user mode
- nice: niced processes executing in user mode
- system: processes executing in kernel mode
- idle: twiddling thumbs
- iowait: waiting for I/O to complete
- irq: servicing interrupts
- softirq: servicing softirqs
- steal: involuntary wait
- guest: running a normal guest
- guest_nice: running a niced guest
These numbers identify the amount of time the CPU has spent performing
different kinds of work. Time units are in USER_HZ (typically hundredths of a
second)
"""
with open('/proc/stat', 'r') as fd:
line = fd.readline()
cpu_data = line.split()
total_cpu_time = sum(map(int, cpu_data[1:]))
cpu_idle_time = int(cpu_data[4])
#return the cpu total&idle time
return total_cpu_time, cpu_idle_time
def memory(self):
"""Execute 'free -m' command, grab the memory capacity and used size
then return; Memory size 'total_mem', Used_mem, and percentage
of used memory.
"""
# Run 'free -m' command and make a list from output.
mem_data = self.execCMD('free', '-m').split()
total_mem = int(mem_data[7]) / 1024.
used_mem = int(mem_data[15]) / 1024.
# Caculate percentage
used_mem_percent = int(used_mem / (total_mem / 100))
# Results are in kilobyte.
return total_mem, used_mem, used_mem_percent
class Py3status:
"""
System status in i3bar
"""
def __init__(self):
self.data = GetData()
self.cpu_total = 0
self.cpu_idle = 0
def cpuInfo(self, json, i3status_config):
"""calculate the CPU status and return it.
"""
response = {'full_text': '', 'name': 'cpu_usage'}
cpu_total, cpu_idle = self.data.cpu()
used_cpu_percent = 1 - float(cpu_idle-self.cpu_idle)/float(cpu_total-self.cpu_total)
self.cpu_total = cpu_total
self.cpu_idle = cpu_idle
if used_cpu_percent <= 40:
response['color'] = i3status_config['color_good']
elif used_cpu_percent <= 75:
response['color'] = i3status_config['color_degraded']
else:
response['color'] = i3status_config['color_bad']
response['full_text'] = "CPU: %.2f%%" % (used_cpu_percent*100)
#cache the status for 10 seconds
response['cached_until'] = time() + 10
return (0, response)
def ramInfo(self, json, i3status_config):
"""calculate the memory (RAM) status and return it.
"""
response = {'full_text': '', 'name': 'ram_info'}
total_mem, used_mem, used_mem_percent = self.data.memory()
if used_mem_percent <= 40:
response['color'] = i3status_config['color_good']
elif used_mem_percent <= 75:
response['color'] = i3status_config['color_degraded']
else:
response['color'] = i3status_config['color_bad']
response['full_text'] = "RAM: %.2f/%.2f GB (%d%%)" % \
(used_mem, total_mem, used_mem_percent)
response['cached_until'] = time()
return (0, response)
|
