1 files changed, 176 insertions, 0 deletions
diff --git a/py3status/modules/player_control.py b/py3status/modules/player_control.py
new file mode 100644
index 0000000..1651327
--- /dev/null
+++ b/py3status/modules/player_control.py
@@ -0,0 +1,176 @@
+# -*- coding: utf-8 -*-
+"""
+Control music/video players.
+
+Provides an icon to control simple functions of audio/video players:
+ - start (left click)
+ - stop  (left click)
+ - pause (middle click)
+
+@author Federico Ceratto <federico.ceratto@gmail.com>, rixx
+@license BSD
+"""
+# Any contributor to this module should add his/her name to the @author
+# line, comma separated.
+
+from syslog import syslog, LOG_INFO
+from time import time, sleep
+import dbus
+import os
+import subprocess
+
+
+def log(msg):
+    syslog(LOG_INFO, "player_control: %s" % msg[:100])
+
+
+class Py3status:
+    """
+    Configuration parameters:
+        - debug: enable verbose logging (bool) (default: False)
+        - supported_players: supported players (str) (comma separated list)
+        - volume_tick: percentage volume change on mouse wheel (int) (positive number or None to disable it)
+
+    """
+
+    debug = False
+    pause_icon = u'❚❚'
+    play_icon = u'▶'
+    stop_icon = u'◼'
+    supported_players = 'audacious,vlc'
+    volume_tick = 1
+
+    def __init__(self):
+        self.status = 'stop'
+        self.icon = self.play_icon
+
+    def on_click(self, i3s_output_list, i3s_config, event):
+        """
+        """
+        buttons = (None, 'left', 'middle', 'right', 'up', 'down')
+        try:
+            button = buttons[event['button']]
+        except IndexError:
+            return
+
+        if button in ('up', 'down'):
+            if self.volume_tick is None:
+                return
+
+            self._change_volume(button == 'up')
+            return
+
+        if self.status == 'play':
+            if button == 'left':
+                self._stop()
+
+            elif button == 'middle':
+                self._pause()
+
+        elif self.status == 'stop':
+            if button == 'left':
+                self._play()
+
+        elif self.status == 'pause':
+            if button in ('left', 'right'):
+                self._play()
+
+    def _run(self, *args):
+        if self.debug:
+            log('running %s' % repr(*args))
+
+        subprocess.check_output(*args, stderr=subprocess.STDOUT)
+
+    def _play(self):
+        self.status = 'play'
+        self.icon = self.stop_icon
+        player_name = self._detect_running_player()
+        if player_name == 'audacious':
+            self._run(['/usr/bin/audacious', '-p'])
+        elif player_name == 'vlc':
+            player = self._get_vlc()
+            if player:
+                player.Play()
+
+    def _stop(self):
+        self.status = 'stop'
+        self.icon = self.play_icon
+        player_name = self._detect_running_player()
+        if player_name == 'audacious':
+            self._run(['/usr/bin/audacious', '-s'])
+        elif player_name == 'vlc':
+            player = self._get_vlc()
+            if player:
+                player.Stop()
+
+    def _pause(self):
+        self.status = 'pause'
+        self.icon = self.pause_icon
+        player_name = self._detect_running_player()
+        if player_name == 'audacious':
+            self._run(['/usr/bin/audacious', '-u'])
+        elif player_name == 'vlc':
+            player = self._get_vlc()
+            if player:
+                player.Pause()
+
+    def _change_volume(self, increase):
+        """Change volume using amixer
+        """
+        sign = '+' if increase else '-'
+        delta = "%d%%%s" % (self.volume_tick, sign)
+        self._run(('/usr/bin/amixer', '-q', 'sset', 'Master', delta))
+
+    def _detect_running_player(self):
+        """Detect running player process, if any
+        """
+        supported_players = self.supported_players.split(',')
+        running_players = []
+        for pid in os.listdir('/proc'):
+            if not pid.isdigit():
+                continue
+
+            fn = os.path.join('/proc', pid, 'comm')
+            try:
+                with open(fn, 'rb') as f:
+                    player_name = f.read().decode().rstrip()
+
+            except:
+                continue
+
+            if player_name in supported_players:
+                running_players.append(player_name)
+
+        # Pick which player to use based on the order in self.supported_players
+        for player_name in supported_players:
+            if player_name in running_players:
+                if self.debug:
+                    log('found player: %s' % player_name)
+
+                return player_name
+
+        return None
+
+    def _get_vlc(self):
+        mpris = 'org.mpris.MediaPlayer2'
+        mpris_slash = '/' + mpris.replace('.', '/')
+        bus = dbus.SessionBus()
+        proxy = bus.get_object(mpris+'.vlc', mpris_slash)
+        return dbus.Interface(proxy, dbus_interface=mpris+'.Player')
+
+    def player_control(self, i3s_output_list, i3s_config):
+        return dict(
+            full_text=self.icon,
+            cached_until=time(),
+        )
+
+
+if __name__ == "__main__":
+    x = Py3status()
+    config = {
+        'color_good': '#00FF00',
+        'color_bad': '#FF0000',
+    }
+    while True:
+        print(x.player_control([], config))
+        sleep(1)