Handy battery python script for zsh