mirror of
https://github.com/LukeSpad/BeoGateway.git
synced 2024-12-23 21:51:51 +00:00
140 lines
5.1 KiB
Python
140 lines
5.1 KiB
Python
#!/usr/bin/python
|
|
import indigo
|
|
import logging
|
|
import os
|
|
import unicodedata
|
|
import threading
|
|
from Foundation import NSAppleScript
|
|
from ScriptingBridge import SBApplication
|
|
|
|
''' Module defining a MusicController class for Apple Music, enables:
|
|
basic transport control of the player,
|
|
query of player status,
|
|
playing existing playlists,
|
|
running external appleScripts for more complex control, and
|
|
reporting messages in the notification centre'''
|
|
|
|
PLAYSTATE = dict([
|
|
(1800426320, "Play"),
|
|
(1800426352, "Pause"),
|
|
(1800426323, "Stop"),
|
|
(1800426310, "Wind"),
|
|
(1800426322, "Rewind")
|
|
])
|
|
|
|
|
|
class MusicController(object):
|
|
|
|
def __init__(self):
|
|
self.app = SBApplication.applicationWithBundleIdentifier_("com.apple.Music")
|
|
|
|
# ########################################################################################
|
|
# Player information
|
|
def get_current_track_info(self):
|
|
name = self.app.currentTrack().name()
|
|
album = self.app.currentTrack().album()
|
|
artist = self.app.currentTrack().artist()
|
|
number = self.app.currentTrack().trackNumber()
|
|
if name:
|
|
# Deal with tracks with non-ascii characters such as accents
|
|
name = unicodedata.normalize('NFD', name).encode('ascii', 'ignore')
|
|
album = unicodedata.normalize('NFD', album).encode('ascii', 'ignore')
|
|
artist = unicodedata.normalize('NFD', artist).encode('ascii', 'ignore')
|
|
return [name, album, artist, number]
|
|
|
|
def get_current_play_state(self):
|
|
return PLAYSTATE.get(self.app.playerState())
|
|
|
|
def get_current_track_position(self):
|
|
return self.app.playerPosition()
|
|
|
|
# ########################################################################################
|
|
# Transport Controls
|
|
def playpause(self):
|
|
self.app.playpause()
|
|
|
|
def play(self):
|
|
if PLAYSTATE.get(self.app.playerState()) in ['Wind', 'Rewind']:
|
|
self.app.resume()
|
|
elif PLAYSTATE.get(self.app.playerState()) == 'Pause':
|
|
self.app.playpause()
|
|
elif PLAYSTATE.get(self.app.playerState()) == 'Stop':
|
|
self. app.setValue_forKey_('true', 'shuffleEnabled')
|
|
playlist = self.app.sources().objectWithName_("Library")
|
|
playlist.playOnce_(None)
|
|
|
|
def pause(self):
|
|
if PLAYSTATE.get(self.app.playerState()) == 'Play':
|
|
self.app.pause()
|
|
|
|
def stop(self):
|
|
if PLAYSTATE.get(self.app.playerState()) != 'Stop':
|
|
self.app.stop()
|
|
|
|
def next_track(self):
|
|
self.app.nextTrack()
|
|
|
|
def previous_track(self):
|
|
self.app.previousTrack()
|
|
|
|
def wind(self, time):
|
|
# self.app.wind()
|
|
|
|
# Native wind function can be a bit annoying
|
|
# I provide an alternative below that skips a set number of seconds forwards
|
|
self.set_current_track_position(time)
|
|
|
|
def rewind(self, time):
|
|
# self.app.rewind()
|
|
|
|
# Native rewind function can be a bit annoying
|
|
# I provide an alternative below that skips a set number of seconds back
|
|
self.set_current_track_position(time)
|
|
|
|
# ########################################################################################
|
|
# More complex playback control functions
|
|
def shuffle(self):
|
|
if self.app.shuffleEnabled():
|
|
self.app.setValue_forKey_('false', 'shuffleEnabled')
|
|
else:
|
|
self.app.setValue_forKey_('true', 'shuffleEnabled')
|
|
|
|
def set_current_track_position(self, time, mode='Relative'):
|
|
if mode == 'Relative':
|
|
# Set playback position in seconds relative to current position
|
|
self.app.setPlayerPosition_(self.app.playerPosition() + time)
|
|
elif mode == 'Absolute':
|
|
# Set playback position in seconds from the start of the track
|
|
self.app.setPlayerPosition_(time)
|
|
|
|
def play_playlist(self, playlist):
|
|
self.app.stop()
|
|
playlist = self.app.sources().objectWithName_("Library").playlists().objectWithName_(playlist)
|
|
playlist.playOnce_(None)
|
|
|
|
# ########################################################################################
|
|
# Accessory functions - threaded due to execution time
|
|
@staticmethod
|
|
def run_script(script, debug):
|
|
script = 'run script ("' + script + '" as POSIX file)'
|
|
if debug:
|
|
indigo.server.log(script, level=logging.DEBUG)
|
|
|
|
def applet(_script):
|
|
# Run an external applescript file
|
|
s = NSAppleScript.alloc().initWithSource_(_script)
|
|
s.executeAndReturnError_(None)
|
|
|
|
threading.Thread(target=applet, args=(script,)).start()
|
|
|
|
@staticmethod
|
|
def notify(message, subtitle):
|
|
def applet(body, title):
|
|
# Post message in notification center
|
|
path = os.path.dirname(os.path.abspath(__file__))
|
|
script = 'tell application "' + path + '/Notify.app" to notify("BeoGateway", "' + \
|
|
body + '", "' + title + '")'
|
|
s = NSAppleScript.alloc().initWithSource_(script)
|
|
s.executeAndReturnError_(None)
|
|
|
|
threading.Thread(target=applet, args=(message, subtitle,)).start()
|