feat: can now pass 'true' or 'false' to tab_mute

This commit is contained in:
David Leavenworth 2022-11-29 11:35:42 -08:00
parent 39e577c36e
commit 069e5bb714
1 changed files with 8 additions and 2 deletions

View File

@ -339,7 +339,8 @@ def debug_webaction(tab: apitypes.Tab, action: str, count: int = 1) -> None:
@cmdutils.register()
@cmdutils.argument('tab', value=cmdutils.Value.count_tab)
def tab_mute(tab: Optional[apitypes.Tab]) -> None:
@cmdutils.argument('to_set', choices=['true', 'false'])
def tab_mute(tab: Optional[apitypes.Tab], to_set: str = None) -> None:
"""Mute/Unmute the current/[count]th tab.
Args:
@ -348,7 +349,12 @@ def tab_mute(tab: Optional[apitypes.Tab]) -> None:
if tab is None:
return
try:
tab.audio.set_muted(not tab.audio.is_muted(), override=True)
if to_set == 'true':
tab.audio.set_muted(True, override=True)
elif to_set == 'false':
tab.audio.set_muted(False, override=True)
else:
tab.audio.set_muted(not tab.audio.is_muted(), override=True)
except apitypes.WebTabError as e:
raise cmdutils.CommandError(e)