Better error handling

This commit is contained in:
Théophile Bastian 2017-03-06 23:45:54 +01:00
parent 1b8a1183c1
commit e9d8a8768c

View file

@ -19,14 +19,20 @@ Colors:
import subprocess
class Py3status:
format_muted = '\uf131' # FontAwesome
format_unmuted = '\uf130' # FontAwesome
class NoMicrophone(Exception):
pass
class Error(Exception):
pass
class Py3status:
format_muted = '\uf131' # FontAwesome
format_unmuted = '\uf130' # FontAwesome
def microphone(self):
try:
isMute = self.__getMuteStatus__()
color = self.py3.COLOR_MUTE if isMute else self.py3.COLOR_UNMUTE
text = self.format_muted if isMute else self.format_unmuted
@ -34,30 +40,44 @@ class Py3status:
'full_text': text,
'color': color,
}
except NoMicrophone:
return {
'full_text': '',
}
except Error:
return {
'full_text': '...',
}
def __getPaList__(self):
try:
paOutput = subprocess.check_output(["pacmd", "list-sources"])
except CalledProcessError:
except subprocess.CalledProcessError:
raise Error()
return paOutput.decode()
def __getMuteStatus__(self):
try:
paList = self.__getPaList__()
fromCurrent = paList[paList.find('* index:'):]
fromCurrent = paList[paList.index('* index:'):]
MUTED_STR = 'muted: '
fromMutestat = fromCurrent[fromCurrent.find(MUTED_STR)
fromMutestat = fromCurrent[fromCurrent.index(MUTED_STR)
+ len(MUTED_STR):]
return fromMutestat[:3] == 'yes'
except ValueError:
raise NoMicrophone()
def __toggleMuteStatus__(self):
try:
isMuted = self.__getMuteStatus__()
paList = self.__getPaList__()
fromCurrent = paList[paList.find('* index: ')+len('* index: '):]
sourceId = int(fromCurrent[:fromCurrent.find('\n')])
fromCurrent = paList[paList.index('* index: ')+len('* index: '):]
sourceId = int(fromCurrent[:fromCurrent.index('\n')])
subprocess.call(['pacmd', 'set-source-mute', str(sourceId),
str(int(not isMuted))])
except ValueError:
raise NoMicrophone()
def on_click(self, event):
self.__toggleMuteStatus__()