/config/waybar/mediaplayer.py

https://github.com/dbrgn/dotfiles · Python · 127 lines · 82 code · 37 blank · 8 comment · 16 complexity · e0520fa4ab6935708d371dd2ea0a37ae MD5 · raw file

  1. #!/usr/bin/env python3
  2. import argparse
  3. import logging
  4. import sys
  5. import signal
  6. import gi
  7. import json
  8. gi.require_version('Playerctl', '2.0')
  9. from gi.repository import Playerctl, GLib
  10. logger = logging.getLogger(__name__)
  11. def write_output(text, player):
  12. logger.info('Writing output')
  13. output = {'text': text,
  14. 'class': 'custom-' + player.props.player_name,
  15. 'alt': player.props.player_name}
  16. sys.stdout.write(json.dumps(output) + '\n')
  17. sys.stdout.flush()
  18. def on_play(player, status, manager):
  19. logger.info('Received new playback status')
  20. on_metadata(player, player.props.metadata, manager)
  21. def on_metadata(player, metadata, manager):
  22. logger.info('Received new metadata')
  23. track_info = ''
  24. if player.props.player_name == 'spotify' and \
  25. 'mpris:trackid' in metadata.keys() and \
  26. ':ad:' in player.props.metadata['mpris:trackid']:
  27. track_info = 'AD PLAYING'
  28. elif player.get_artist() != '' and player.get_title() != '':
  29. track_info = '{artist} - {title}'.format(artist=player.get_artist(),
  30. title=player.get_title())
  31. else:
  32. track_info = player.get_title()
  33. if player.props.status != 'Playing' and track_info:
  34. track_info = ' ' + track_info
  35. write_output(track_info, player)
  36. def on_player_appeared(manager, player, selected_player=None):
  37. if player is not None and (selected_player is None or player.name == selected_player):
  38. init_player(manager, player)
  39. else:
  40. logger.debug("New player appeared, but it's not the selected player, skipping")
  41. def on_player_vanished(manager, player):
  42. logger.info('Player has vanished')
  43. sys.stdout.write('\n')
  44. sys.stdout.flush()
  45. def init_player(manager, name):
  46. logger.debug('Initialize player: {player}'.format(player=name.name))
  47. player = Playerctl.Player.new_from_name(name)
  48. player.connect('playback-status', on_play, manager)
  49. player.connect('metadata', on_metadata, manager)
  50. manager.manage_player(player)
  51. on_metadata(player, player.props.metadata, manager)
  52. def signal_handler(sig, frame):
  53. logger.debug('Received signal to stop, exiting')
  54. sys.stdout.write('\n')
  55. sys.stdout.flush()
  56. # loop.quit()
  57. sys.exit(0)
  58. def parse_arguments():
  59. parser = argparse.ArgumentParser()
  60. # Increase verbosity with every occurrence of -v
  61. parser.add_argument('-v', '--verbose', action='count', default=0)
  62. # Define for which player we're listening
  63. parser.add_argument('--player')
  64. return parser.parse_args()
  65. def main():
  66. arguments = parse_arguments()
  67. # Initialize logging
  68. logging.basicConfig(stream=sys.stderr, level=logging.DEBUG,
  69. format='%(name)s %(levelname)s %(message)s')
  70. # Logging is set by default to WARN and higher.
  71. # With every occurrence of -v it's lowered by one
  72. logger.setLevel(max((3 - arguments.verbose) * 10, 0))
  73. # Log the sent command line arguments
  74. logger.debug('Arguments received {}'.format(vars(arguments)))
  75. manager = Playerctl.PlayerManager()
  76. loop = GLib.MainLoop()
  77. manager.connect('name-appeared', lambda *args: on_player_appeared(*args, arguments.player))
  78. manager.connect('player-vanished', on_player_vanished)
  79. signal.signal(signal.SIGINT, signal_handler)
  80. signal.signal(signal.SIGTERM, signal_handler)
  81. for player in manager.props.player_names:
  82. if arguments.player is not None and arguments.player != player.name:
  83. logger.debug('{player} is not the filtered player, skipping it'
  84. .format(player=player.name)
  85. )
  86. continue
  87. init_player(manager, player)
  88. loop.run()
  89. if __name__ == '__main__':
  90. main()