/
BtBatStat.py
150 lines (121 loc) · 5.33 KB
/
BtBatStat.py
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
import subprocess,re,time,sys,webbrowser,urllib2,decimal
from Foundation import NSDate,NSObject,NSTimer,NSRunLoop,NSDefaultRunLoopMode
from AppKit import NSImage,NSStatusBar,NSMenuItem,NSApplication,NSMenu,NSVariableStatusItemLength,NSRunAlertPanel
from PyObjCTools import AppHelper
from optparse import OptionParser
if len(sys.argv) > 1 and sys.argv[1][:4] == '-psn':
del sys.argv[1]
VERSION = '0.9.1'
LONGVERSION = 'BtBatStat ' + VERSION
AboutText = """Writen by: Joris Vandalon
App Icon Design by: FIF7Y
Code License: New BSD License
This software will always be free of charge.
Donations can be done via the website and will be much appreciated.
"""
updateText = "There is a new version of BtBatStat Available."
appUrl = 'http://code.google.com/p/btbatstat/'
updateUrl = 'http://code.google.com/p/btbatstat/downloads/list'
parser = OptionParser()
parser.add_option("-d", action="store_true", dest="debug")
(options, args)= parser.parse_args()
start_time = NSDate.date()
def versionCheck():
try:
LatestRelease = urllib2.urlopen("http://btbatstat.vandalon.org/VERSION", None, 2).read().strip()
except:
return False
return ( LatestRelease and map(int, LatestRelease.split('.')) > map(int, VERSION.split('.')) )
#Check for new version
def checkForUpdates():
if versionCheck():
if NSRunAlertPanel(LONGVERSION, updateText , "Download Update", "Ignore for now", None ) == 1:
webbrowser.open(updateUrl)
class Timer(NSObject):
def about_(self, notification):
if versionCheck():
AboutTitle = LONGVERSION + " (Update Available!)"
about = NSRunAlertPanel(AboutTitle, AboutText , "OK", "Visit Website", "Download Update" )
else:
about = NSRunAlertPanel(LONGVERSION, AboutText , "OK", "Visit Website", None )
if about == 0:
webbrowser.open(appUrl)
elif about == -1:
webbrowser.open(updateUrl)
def applicationDidFinishLaunching_(self, notification):
self.noDevice = None
#Create menu
self.menu = NSMenu.alloc().init()
self.barItem = dict()
# Load images
self.noDeviceImage = NSImage.alloc().initByReferencingFile_('icons/no_device.png')
self.barImage = dict(kb = NSImage.alloc().initByReferencingFile_('icons/kb.png'),
magicMouse = NSImage.alloc().initByReferencingFile_('icons/magic_mouse.png'),
mightyMouse = NSImage.alloc().initByReferencingFile_('icons/mighty_mouse.png'),
magicTrackpad = NSImage.alloc().initByReferencingFile_('icons/TrackpadIcon.png'))
#Define menu items
self.statusbar = NSStatusBar.systemStatusBar()
self.menuAbout = NSMenuItem.alloc().initWithTitle_action_keyEquivalent_('About BtBatStat', 'about:', '')
self.separator_menu_item = NSMenuItem.separatorItem()
self.menuQuit = NSMenuItem.alloc().initWithTitle_action_keyEquivalent_('Quit', 'terminate:', '')
# Get the timer going
self.timer = NSTimer.alloc().initWithFireDate_interval_target_selector_userInfo_repeats_(start_time, 10.0, self, 'tick:', None, True)
NSRunLoop.currentRunLoop().addTimer_forMode_(self.timer, NSDefaultRunLoopMode)
self.timer.fire()
#Add menu items
self.menu.addItem_(self.menuAbout)
self.menu.addItem_(self.separator_menu_item)
self.menu.addItem_(self.menuQuit)
#Check for updates
checkForUpdates()
def ioreg(self, key, flags):
return subprocess.Popen(["/usr/sbin/ioreg", flags, key], stdout=subprocess.PIPE).communicate()[0]
def createBarItem(self, icon):
barItem = self.statusbar.statusItemWithLength_(NSVariableStatusItemLength)
barItem.setImage_(icon)
barItem.setHighlightMode_(1)
barItem.setMenu_(self.menu)
return barItem
def tick_(self, notification):
self.devicesFound = 0
if options.debug:
start = time.time()
deviceCmd = dict( mightyMouse = self.ioreg("AppleBluetoothHIDMouse","-rc"),
magicMouse = self.ioreg("BNBMouseDevice","-rc"),
magicTrackpad = self.ioreg("BNBTrackpadDevice","-rc"))
deviceCmd['kb'] = self.ioreg("AppleBluetoothHIDKeyboard","-rc")
if not deviceCmd['kb']:
deviceCmd['kb'] = self.ioreg("IOAppleBluetoothHIDDriver","-n")
for device,Output in deviceCmd.items():
Percentage = 0
if Output:
Percentage = re.search('BatteryPercent" = (\d{1,2}0?)', Output)
if Percentage:
self.hit = True
if options.debug:
print "Found " + device
self.devicesFound += 1
if self.noDevice is not None:
self.statusbar.removeStatusItem_(self.noDevice)
self.menu.removeItem_(self.menuNotice)
self.noDevice = None
if not device in self.barItem:
self.barItem[device] = self.createBarItem(self.barImage[device])
self.barItem[device].setTitle_(Percentage.group(1) + '%')
if device in self.barItem and not Percentage:
self.statusbar.removeStatusItem_(self.barItem[device])
del self.barItem[device]
if options.debug:
print "Found", self.devicesFound, "Devices."
if self.devicesFound == 0 and self.noDevice == None:
self.menuNotice = NSMenuItem.alloc().initWithTitle_action_keyEquivalent_('BtBatStat: No devices found.', '', '')
self.menu.insertItem_atIndex_(self.menuNotice,0)
self.noDevice = self.createBarItem(self.noDeviceImage)
if options.debug:
end = time.time()
print "Time elapsed = ", end - start, "seconds"
if __name__ == "__main__":
app = NSApplication.sharedApplication()
delegate = Timer.alloc().init()
app.setDelegate_(delegate)
AppHelper.runEventLoop()