dotfiles/waybar/mediaplayer.py

70 lines
2.3 KiB
Python

#!/usr/bin/env python3
import argparse
import logging
import sys
import gi
import json
gi.require_version('Playerctl', '2.0')
from gi.repository import Playerctl, GLib
logger = logging.getLogger(__name__)
def write_output(text, player):
logger.info('Writing output')
output = {'text': text,
'class': 'custom-' + player.props.player_name,
'alt': player.props.player_name}
sys.stdout.write(json.dumps(output) + '\n')
sys.stdout.flush()
def on_play(player, status, manager):
logger.info('Received new playback status')
on_metadata(player, player.props.metadata, manager)
def on_metadata(player, metadata, manager):
logger.info('Received new metadata')
track_info = ''
if player.props.player_name == 'spotify':
track_info = '{artist} - {title}'.format(artist=metadata['xesam:artist'][0], title=metadata['xesam:title'])
else:
if 'xesam:artist' in metadata.keys() and 'xesam:title' in metadata.keys():
track_info = '{artist} - {title}'.format(artist=metadata['xesam:artist'][0], title=metadata['xesam:title'])
else:
track_info = metadata['xesam:title']
write_output(track_info, player)
def on_player_appeared(manager, player):
logger.info('Player has appeared')
player.connect('playback-status', on_play, manager)
player.connect('metadata', on_metadata, manager)
on_metadata(player, player.props.metadata, manager)
def on_player_vanished(manager, player):
logger.info('Player has vanished')
sys.stdout.write('\n')
sys.stdout.flush()
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--ignore', help='List of player names to ignore', default=[])
args = parser.parse_args()
logging.basicConfig(level=logging.INFO)
manager = Playerctl.PlayerManager()
loop = GLib.MainLoop()
manager.connect('name-appeared', lambda *args: on_player_appeared(*args))
manager.connect('player-vanished', lambda *args: on_player_vanished(*args))
for player in manager.props.player_names:
if args.ignore is not None and player.name in args.ignore:
logger.info("Ignoring player: {player}".format(player=player.name))
continue
on_player_appeared(manager, player)
loop.run()
if __name__ == '__main__':
main()