This repository was archived by the owner on Aug 3, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbatterytray
executable file
·160 lines (136 loc) · 6.51 KB
/
batterytray
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
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
#! /usr/bin/env python
from gi.repository import Gtk, GLib
import os,time,glob
#Hack for pygtk ignoring ^C
import signal
signal.signal(signal.SIGINT, signal.SIG_DFL)
class Battery():
def __init__(self, id):
self.id = id
self.voltage = self.get("voltage_now") / 1000
self.status = self.gets("status")
self.state = self.status == "Charging"
if os.path.exists (os.path.join("/sys/class/power_supply",self.id,"energy_full")):
self.max = self.get("energy_full")/ self.voltage
self.orig_max = self.get("energy_full_design")/ self.voltage
self.now = self.get("energy_now")/ self.voltage
self.rate = self.get("power_now") / self.voltage
else:
self.max = self.get("charge_full") / 1000
self.orig_max = self.get("charge_full_design") / 1000
self.now = self.get("charge_now") / 1000
self.rate = self.get("current_now") / 1000
try:
assert (os.path.exists (os.path.join ("/sys/devices/platform/smapi/",self.id)))
assert (self.tpget ("installed") is 1)
self.tp = True
self.dom = self.tpgets ("manufacture_date")
self.temp = self.tpget ("temperature")/1000
self.range = "%i%% – %i%%" % (self.tpget ("start_charge_thresh"),self.tpget("stop_charge_thresh"))
self.inhibit = self.tpget ("inhibit_charge_minutes")
self.forced = self.tpget ("force_discharge") is 1
except Exception as e:
self.tp = False
def get (self, attr):
try: return int(self.gets (attr))
except: return 0
def tpget (self, attr):
try: return int(self.tpgets (attr))
except: return 0
def gets (self, attr): return self._get (os.path.join ("/sys/class/power_supply", self.id,attr))
def tpgets (self, attr): return self._get (os.path.join ("/sys/devices/platform/smapi/", self.id,attr))
def _get (self, path):
try:
with open (path) as f: return "\n".join(f.readlines()).strip()
except: return ""
def get_percent (self):
try: return self.now / self.max
except: return 0
def get_designpercent (self):
try: return self.now / self.orig_max
except: return 0
def ttl (self):
try:
if self.tp: ttl = self.tpget ("remaining_running_time") * 60
else:
if self.state: ttl = 3600 * (self.max - self.now) / self.rate
else: ttl = 3600 * self.now / self.rate
return time.strftime ("%T", time.gmtime(ttl))
except: return "(unknown)"
def __str__ (self):
retval = "Battery %s:\n\t%s %.2f%%\n\t%.2f%% design capacity\n\t%s remaining" % (self.id, self.status, self.get_percent()*100, self.get_designpercent()*100, self.ttl())
retval +="\n\t%s %s (S/N %s)\n\t%i mAh / %i Wh" % (self.gets("manufacturer"), self.gets("model_name"), self.gets("serial_number"), self.max, int (self.max*self.voltage/1000000.0))
if self.tp:
retval += "\n\t%i°C\n\tCharge range: %s\n\tManufactured: %s"%(self.temp,self.range,self.dom)
if self.inhibit > 0: retval += "\n\tCharge inhibited for %i min" % self.inhibit
if self.forced: retval += "\n\tForced discharge active"
return retval
class TrayIcon():
def __init__(self):
self.icon = Gtk.StatusIcon()
self.update_interval = 5
self.icon.embedded = True
self.last_state = ""
self.is_online = self.online()
self.offline_time = self.suspend_time = 0
self.last_check_time = time.time ()
self.update_icon()
self.icon.set_visible(True)
GLib.timeout_add_seconds(self.update_interval, self.update_icon)
def destroy (self, widget, data=None):
Gtk.main_quit()
def main(self):
Gtk.main()
def online (self):
return any (map (lambda x: int(open(x).read().strip())==1, glob.glob ("/sys/class/power_supply/*/online")))
def update_icon(self):
self.batteries = {}
for entry in os.listdir ("/sys/class/power_supply"):
if "BAT" in entry:
self.batteries[entry] = Battery(entry)
if not self.batteries:
self.icon.set_from_stock(Gtk.STOCK_CONNECT)
self.icon.set_tooltip_text("No battery found.")
return True
online_stats = ''
status, charging, self.is_online, curtime = 0, False, self.online(), time.time()
if not self.is_online:
self.offline_time += self.update_interval #don't rely on timestamps as it cannot factor in suspends
timediff = curtime - self.last_check_time
if timediff > (self.update_interval*1.5): #allow some overhead, assume suspend if longer delay
print ("%i %i %i %i" % (curtime, self.last_check_time, timediff, self.update_interval*1.5))
self.suspend_time += timediff
if self.offline_time <= self.update_interval: online_stats = '\n\nOn battery, unknown duration.'
else:
online_stats = time.strftime('\n\nOn battery for ca. %T.', time.gmtime(self.offline_time))
if self.suspend_time: online_stats += time.strftime('\n\tSuspended for ca. %T.', time.gmtime(self.suspend_time))
else:
self.offline_time = self.suspend_time = 0
self.last_check_time = curtime
now = rate = 0
for battery in self.batteries:
status += self.batteries[battery].get_percent()
charging |= self.batteries[battery].state
now += self.batteries[battery].now
rate += self.batteries[battery].rate
try: ttl = time.strftime ("%T", time.gmtime(3600 * now / rate))
except: ttl = '(unknown)'
status /= len(self.batteries)
if status < 0.25: state = "crit"
elif status < 0.5: state = "low"
elif status < 0.75: state = "high"
else: state = "full"
totals = "\nTotal: %.0f%%, %s remaining" % (status*100, ttl)
if status < 0.1 and not self.is_online:
os.system ("notify-send 'Battery low!' -i battery-caution")
tooltip_text = "\n".join([str(self.batteries[battery]) for battery in self.batteries])+online_stats+totals
self.icon.set_tooltip_text (tooltip_text)
if status != self.last_state:
iconname = "/usr/share/pixmaps/battery/battery-"+state+ ("-charging" if charging else "")
iconname += "-ac.png" if self.is_online else ".png"
self.icon.set_from_file(iconname)
self.last_state = status
return True
if __name__ == "__main__":
i = TrayIcon()
i.main()