Port to python

This commit is contained in:
Andros Fenollosa 2016-06-04 11:56:04 +02:00
parent c0b9964a5a
commit ffee10eb16
2 changed files with 82 additions and 72 deletions

82
battery_health.2s.py Executable file
View File

@ -0,0 +1,82 @@
#!/usr/bin/env python2
#- * -coding: utf - 8 - * -
#
# <bitbar.title>Battery Health</bitbar.title>
# <bitbar.version>v1.1</bitbar.version>
# <bitbar.author>Andros Fenollosa</bitbar.author>
# <bitbar.author.github>tanrax</bitbar.author.github>
# <bitbar.desc>Shows power percentaje and notice when you load</bitbar.desc>
import os, math, subprocess, pickle, tempfile
# Get info battery
SAVE_LOCATION = os.path.join(tempfile.gettempdir(), 'batteryHealth2.pkl')
LIM_LOWER = 45
LIM_UPPER = 85
p = subprocess.Popen(["ioreg", "-rc", "AppleSmartBattery"], stdout = subprocess.PIPE)
output = p.communicate()[0]
b_max = 0
o_cur = 0
is_charging = ''
dateSave = False
alertMin = False
alertMax = False
try:
dateFile = open(SAVE_LOCATION)
dateSave = pickle.load(dateFile)
alertMin = dateSave['alertMin']
alertMax = dateSave['alertMax']
except:
pass
# Get variables
for l in output.splitlines():
if 'MaxCapacity' in l:
o_max = l
if 'CurrentCapacity' in l:
o_cur = l
if 'IsCharging' in l:
is_charging = l
b_max = float(o_max.rpartition('=')[-1].strip())
b_cur = float(o_cur.rpartition('=')[-1].strip())
is_charging = str(is_charging.rpartition('=')[-1].strip())
if is_charging == 'Yes':
is_charging = True
else:
is_charging = False
# Calculate porcent battery
charge = b_cur / b_max
charge_porcent = int(math.ceil(100 * charge))
# Logic
## Alert Min
if LIM_LOWER >= charge_porcent and not alertMin:
alertMin = True
os.system('osascript -e \'display notification "Battery too low" with title "Battery health" sound name "Blow"\'')
## Alert Max
if LIM_UPPER <= charge_porcent and not alertMax:
alertMax = True
os.system('osascript -e \'display notification "Very charged battery" with title "Battery health" sound name "Blow"\'')
## Reset alerts
if LIM_UPPER > charge_porcent > LIM_LOWER:
alertMin = False
alertMax = False
# Save
dateTemp = {'alertMax': alertMax, 'alertMin': alertMin}
dateSave = open(SAVE_LOCATION, 'w+')
pickle.dump(dateTemp, dateSave)
# Print
final = ''
if is_charging:
final += '⚡️'
elif alertMin or alertMax:
final += '🔴'
final = '{text}{charge}%'.format(text=final, charge=charge_porcent)
print(final)

View File

@ -1,72 +0,0 @@
#!/bin/bash
#
# <bitbar.title>Battery Health</bitbar.title>
# <bitbar.version>v1.0</bitbar.version>
# <bitbar.author>Andros Fenollosa</bitbar.author>
# <bitbar.author.github>tanrax</bitbar.author.github>
# <bitbar.image>https://programadorwebvalencia.com/wp-content/uploads/2016/04/Screen-Shot-2016-04-05-at-12.47.25.jpg</bitbar.image>
# <bitbar.desc>Shows power percentaje and notice when you load</bitbar.desc>
# Variables
BATTERY=$(ioreg -l | awk '$3~/Capacity/{c[$3]=$5}END{OFMT="%.0f%";max=c["\"MaxCapacity\""];print(max>0?100*c["\"CurrentCapacity\""]/max:"?")}')
TYPE=$(pmset -g cap | sed -ne 's/^Capabilities for \(.*\) Power:$/\1/p')
POR_LOW=45
POR_HIGH=85
LOW=False
HIGH=False
SAVE_LOCATION=$TMPDIR/batteryHealth
BAD='🔴'
AC='⚡️'
# Get data
if [ -f "$SAVE_LOCATION" ]; then
DATA=$(cat "$SAVE_LOCATION")
else
DATA="$LOW|$HIGH"
fi
LOW=$(echo "$DATA" | cut -d "|" -f1)
HIGH=$(echo "$DATA" | cut -d "|" -f2)
# Functions
function changeStatus {
osascript -e "display notification \"$2\" with title \"$1\" sound name \"$3\"" &> /dev/null
}
function batteryLow {
LOW=True
changeStatus "Battery Low" "$BATTERY"% "Blow"
}
function batteryNormal {
HIGH=False
LOW=False
}
function batteryHigh {
HIGH=True
changeStatus "Battery high" "$BATTERY"% "Blow"
}
# Logic
if [ "$BATTERY" -le $POR_LOW ] && [ $LOW = False ]; then
batteryLow
elif [ "$BATTERY" -ge $POR_HIGH ] && [ $HIGH = False ]; then
batteryHigh
elif [ "$BATTERY" -le $POR_HIGH ] && [ "$BATTERY" -ge $POR_LOW ]; then
batteryNormal
fi
# Save data
echo "$LOW|$HIGH" > "$SAVE_LOCATION";
# View battery
if [ "$TYPE" = "AC" ]; then
echo $AC "$BATTERY"%
else
if [ $HIGH = True ] || [ $LOW = True ]; then
echo $BAD "$BATTERY"%
else
echo "$BATTERY"%
fi
fi