mirror of
https://github.com/balkian/dotfiles.git
synced 2024-11-05 05:01:43 +00:00
418 lines
13 KiB
Python
418 lines
13 KiB
Python
#!/usr/bin/python
|
|
|
|
# Spotify-notify
|
|
#
|
|
# v0.6d (28th aug 11)
|
|
# by JonW (jon.neverwinter@gmail.com)
|
|
# patched 20110907 by Jansen Price (sumpygump@gmail.com)
|
|
# patched 20120729 by Jansen Price (sumpygump@gmail.com) and brandl.matthaeus
|
|
#
|
|
# Original by SveinT (sveint@gmail.com)
|
|
# up to v0.5.2 (27th jan 11)
|
|
|
|
|
|
import dbus
|
|
|
|
import gobject, gtk, os, tempfile, sys, time, re, indicate, urllib2
|
|
from optparse import OptionParser
|
|
from subprocess import *
|
|
|
|
# The url to use when fetching spotify track information.
|
|
SPOTIFY_OPEN_URL = "http://open.spotify.com/track/"
|
|
|
|
# The path to this application's directory.
|
|
APPLICATION_DIR = sys.path[0] + "/"
|
|
|
|
# The file path to spotify. If empty, it will try to auto detect.
|
|
SPOTIFY_PROCESS_NAME = 'spotify'
|
|
|
|
# How often to check if spotify has been closed (in milliseconds).
|
|
SPOTIFY_CLOSED_CHECK = 20000
|
|
|
|
class SpotifyNotify():
|
|
|
|
spotifyPath = ''
|
|
|
|
tryToReconnect = False
|
|
|
|
tmpfile = False
|
|
|
|
def __init__(self, debugger):
|
|
self.debug = debugger
|
|
self.spotifyservice = False
|
|
|
|
self.prev = 0
|
|
self.new = False
|
|
self.prevMeta = {}
|
|
self.notifyid = 0
|
|
|
|
self.connect()
|
|
|
|
def __del__(self):
|
|
if SpotifyNotify and SpotifyNotify.tmpfile:
|
|
SpotifyNotify.tmpfile.close()
|
|
|
|
def connect(self):
|
|
self.debug.out("Connecting to spotify.")
|
|
self.bus = dbus.Bus(dbus.Bus.TYPE_SESSION)
|
|
|
|
try:
|
|
self.spotifyservice = self.bus.get_object(
|
|
'com.spotify.qt',
|
|
'/org/mpris/MediaPlayer2'
|
|
)
|
|
SpotifyNotify.tryToReconnect = False
|
|
except Exception, e:
|
|
self.spotifyservice = False
|
|
self.debug.out("Failed to connect.")
|
|
self.debug.out(e)
|
|
|
|
def executeCommand(self, key):
|
|
if not key:
|
|
return
|
|
|
|
self.connect()
|
|
self.debug.out("Running command: {0}".format(key))
|
|
self.cmd = self.spotifyservice.get_dbus_method(key, 'org.mpris.MediaPlayer2.Player')
|
|
self.cmd()
|
|
|
|
def pollChange(self):
|
|
try:
|
|
self.spotifyservice = self.bus.get_object('com.spotify.qt', '/')
|
|
self.cmd = self.spotifyservice.get_dbus_method(
|
|
'GetMetadata',
|
|
'org.freedesktop.MediaPlayer2'
|
|
)
|
|
self.new = self.cmd()
|
|
except Exception, e:
|
|
self.debug.out('Spotify service not connected.')
|
|
SpotifyNotify.tryToReconnect = True
|
|
|
|
if (self.prev != self.new):
|
|
self.trackChange(self.new)
|
|
self.prev = self.new
|
|
|
|
return 1
|
|
|
|
def trackChange(self, *trackChange):
|
|
if not trackChange[0]:
|
|
return
|
|
|
|
self.prev = trackChange[0]
|
|
|
|
trackInfo = {}
|
|
trackMap = {
|
|
'artist' : 'xesam:artist',
|
|
'album' : 'xesam:album',
|
|
'title' : 'xesam:title',
|
|
'year' : 'xesam:contentCreated',
|
|
'trackhash' : 'mpris:trackid',
|
|
'arturl' : 'mpris:artUrl'
|
|
}
|
|
|
|
# Fetch the track information for the notification window.
|
|
for key in trackMap:
|
|
if not trackMap[key] in trackChange[0]:
|
|
continue
|
|
piece = trackChange[0][trackMap[key]]
|
|
if key == 'year':
|
|
piece = str(piece[:4])
|
|
elif isinstance(piece, list):
|
|
piece = ", ".join(piece)
|
|
|
|
if not isinstance(piece, str):
|
|
piece = str(piece)
|
|
|
|
trackInfo[key] = piece.encode('utf-8')
|
|
|
|
if not self.prevMeta\
|
|
or not SpotifyNotify.tmpfile\
|
|
or 'iconfilename' not in self.prevMeta\
|
|
or self.prevMeta['artist'] != trackInfo['artist']\
|
|
or self.prevMeta['album'] != trackInfo['album']:
|
|
trackInfo['iconfilename'] = self.retrieveCoverImage(trackInfo)
|
|
|
|
cover_image = ''
|
|
|
|
if 'iconfilename' in trackInfo:
|
|
cover_image = trackInfo['iconfilename']
|
|
elif 'iconfilename' in self.prevMeta:
|
|
cover_image = self.prevMeta['iconfilename']
|
|
trackInfo['iconfilename'] = cover_image
|
|
|
|
if cover_image == '':
|
|
cover_image = APPLICATION_DIR + 'icon_spotify.png'
|
|
|
|
self.prevMeta = trackInfo
|
|
|
|
# Connect to notification interface on DBUS.
|
|
self.notifyservice = self.bus.get_object(
|
|
'org.freedesktop.Notifications',
|
|
'/org/freedesktop/Notifications'
|
|
)
|
|
self.notifyservice = dbus.Interface(
|
|
self.notifyservice,
|
|
"org.freedesktop.Notifications"
|
|
)
|
|
notifyText = "{0}\n{1}".format(
|
|
trackInfo['title'],
|
|
trackInfo['album']
|
|
)
|
|
if len(trackInfo['year']) > 0:
|
|
notifyText += " ({0})".format(trackInfo['year'])
|
|
|
|
# Send track change information to stdout
|
|
print "Changing track : {0} | {1} | {2} ({3})".format(
|
|
trackInfo['artist'],
|
|
trackInfo['title'],
|
|
trackInfo['album'],
|
|
trackInfo['year']
|
|
)
|
|
|
|
# The second param is the replace id, so get the notify id back,
|
|
# store it, and send it as the replacement on the next call.
|
|
self.notifyid = self.notifyservice.Notify(
|
|
"Spotify-notify",
|
|
self.notifyid,
|
|
cover_image,
|
|
trackInfo['artist'],
|
|
notifyText,
|
|
[],
|
|
{},
|
|
2000
|
|
)
|
|
|
|
def retrieveCoverImage(self, trackInfo):
|
|
if 'arturl' in trackInfo:
|
|
self.debug.out("Simply retrieving image from {0}".format(trackInfo['arturl']))
|
|
iconfilename = self.fetchCoverImage(trackInfo['arturl'])
|
|
else:
|
|
#if (trackInfo['trackhash'][0:14] == 'spotify:local:'):
|
|
# self.debug.out("Track is a local file. No art available.")
|
|
# return ''
|
|
|
|
self.debug.out("Attempting to fetch image from spotify")
|
|
iconfilename = self.fetchCoverImageSpotify(
|
|
trackInfo['artist'],
|
|
trackInfo['album'],
|
|
trackInfo['trackhash']
|
|
)
|
|
return iconfilename
|
|
|
|
def fetchCoverImageSpotify(self, artist, album, trackhash):
|
|
try:
|
|
trackid = trackhash.split(":")[2]
|
|
url = SPOTIFY_OPEN_URL + trackid
|
|
tracksite = urllib2.urlopen(url).read()
|
|
|
|
# Attempt to get the image url from the open graph image meta tag.
|
|
imageurl = False
|
|
metaMatch = re.search(
|
|
'<meta\s[^\>]*property\s*=\s*["\']og:image["\'][^\>]*/?>',
|
|
tracksite
|
|
)
|
|
if metaMatch:
|
|
contentMatch = re.search(
|
|
'content\s*=\s*["\']([^\"\']*)["\']',
|
|
metaMatch.group(0)
|
|
)
|
|
if contentMatch:
|
|
imageurl = contentMatch.group(1)
|
|
|
|
if not imageurl:
|
|
self.debug.out("No cover available.")
|
|
raise()
|
|
|
|
return self.fetchCoverImage(imageurl)
|
|
except Exception, e:
|
|
self.debug.out("Couldn't fetch cover image.")
|
|
self.debug.out(e)
|
|
|
|
return ''
|
|
|
|
def fetchCoverImage(self, url):
|
|
# Close the temporary image file, we are going to make a new one.
|
|
if SpotifyNotify.tmpfile:
|
|
SpotifyNotify.tmpfile.close()
|
|
SpotifyNotify.tmpfile = False
|
|
|
|
try:
|
|
SpotifyNotify.tmpfile = tempfile.NamedTemporaryFile()
|
|
tmpfilename = SpotifyNotify.tmpfile.name
|
|
self.debug.out("Album art tmp filepath: {0}".format(tmpfilename))
|
|
|
|
coverfile = urllib2.urlopen(url)
|
|
SpotifyNotify.tmpfile.write(coverfile.read())
|
|
SpotifyNotify.tmpfile.flush()
|
|
return tmpfilename
|
|
except Exception, e:
|
|
self.debug.out("Couldn't fetch cover image.")
|
|
self.debug.out(e)
|
|
|
|
return ''
|
|
|
|
@staticmethod
|
|
def startSpotify(Debug):
|
|
if not SpotifyNotify.spotifyPath:
|
|
Debug.out("No spotify process identifier found.")
|
|
return
|
|
|
|
ident = SpotifyNotify.spotifyPath
|
|
Debug.out("Looking for spotify as: {0}".format(ident))
|
|
|
|
procs = SpotifyNotify.checkForProcess(
|
|
'ps x | grep "{0}" | grep -v grep'.format(ident),
|
|
Debug
|
|
)
|
|
if len(procs):
|
|
Debug.out("Spotify process found as: {0}".format(" ".join(procs[0])))
|
|
return
|
|
|
|
Debug.out("Starting new Spotify now.")
|
|
|
|
FNULL = open('/dev/null', 'w')
|
|
spid = Popen([ident], stdout=FNULL, stderr=FNULL).pid
|
|
if spid:
|
|
Debug.out("Spotify started, pid: {0}.".format(spid))
|
|
else:
|
|
Debug.out("Spotify could not be started.")
|
|
|
|
@staticmethod
|
|
def checkForClosedSpotify(SN, Debug):
|
|
if not SpotifyNotify.spotifyPath:
|
|
Debug.out("No spotify process identifier found.")
|
|
return False
|
|
|
|
ident = SpotifyNotify.spotifyPath
|
|
Debug.out("Looking for spotify as: {0}".format(ident))
|
|
|
|
procs = SpotifyNotify.checkForProcess(
|
|
'ps x | grep "{0}" | grep -v grep'.format(ident),
|
|
Debug
|
|
)
|
|
if len(procs):
|
|
Debug.out("Spotify process found as: {0}".format(" ".join(procs[0])))
|
|
|
|
if (SpotifyNotify.tryToReconnect):
|
|
SN.connect()
|
|
|
|
return True
|
|
|
|
if SpotifyNotify.tmpfile:
|
|
SpotifyNotify.tmpfile.close()
|
|
|
|
Debug.out("Spotify has been closed, therefore I die.")
|
|
exit(0)
|
|
|
|
@staticmethod
|
|
def preventDuplicate(Debug):
|
|
mypid = os.getpid()
|
|
Debug.out("My pid: {0}".format(mypid))
|
|
|
|
proc = SpotifyNotify.checkForProcess('ps -p {0}'.format(mypid), Debug)
|
|
if not proc[0][3]:
|
|
return
|
|
|
|
process = proc[0][3]
|
|
search = 'ps -C {0}'.format(process)
|
|
|
|
Debug.out("Looking for other processes named: {0}".format(process).strip())
|
|
|
|
if process == 'python':
|
|
if not sys.argv[0]:
|
|
Debug.out("Process started using python, cannot determine script name.")
|
|
return
|
|
|
|
search = 'ps ax | grep "python {0}" | grep -v grep'.format(sys.argv[0])
|
|
|
|
for line in SpotifyNotify.checkForProcess(search, Debug):
|
|
if int(line[0]) != mypid:
|
|
print("This program was already running.")
|
|
Debug.out("I am a duplicate. I shall end myself. ({0})".format(" ".join(line)))
|
|
exit(0)
|
|
|
|
@staticmethod
|
|
def checkForProcess(proc, Debug):
|
|
output = []
|
|
|
|
for line in Popen(proc, shell=True, stdout=PIPE).stdout:
|
|
fields = line.split()
|
|
if not fields[0].isdigit():
|
|
continue
|
|
|
|
output.append(fields)
|
|
|
|
return output
|
|
|
|
class DebugMe():
|
|
def __init__(self, toggle):
|
|
if toggle:
|
|
self.output = True
|
|
else:
|
|
self.output = False
|
|
|
|
def out(self, msg):
|
|
if not self.output:
|
|
return
|
|
|
|
print(">> {0}".format(msg))
|
|
|
|
if __name__ == "__main__":
|
|
parser = OptionParser()
|
|
parser.add_option(
|
|
'-a',
|
|
'--action',
|
|
dest = 'action',
|
|
default = None,
|
|
type = 'choice',
|
|
choices = ['playPause', 'play', 'pause', 'next', 'previous'],
|
|
help = 'Music player actions (playPause/play/pause/next/previous).'
|
|
)
|
|
parser.add_option(
|
|
'-n',
|
|
'--skip_notify',
|
|
dest = 'skipNotify',
|
|
action = 'store_true',
|
|
default = False,
|
|
help = 'Song change notifications will be turned off.'
|
|
)
|
|
parser.add_option(
|
|
'-d',
|
|
'--debug',
|
|
dest = 'debug',
|
|
action = 'store_true',
|
|
default = False,
|
|
help = 'Debug messages will be displayed.'
|
|
)
|
|
|
|
(options, args) = parser.parse_args()
|
|
|
|
Debug = DebugMe(options.debug)
|
|
print("Spotify-notify v0.6")
|
|
|
|
if SPOTIFY_PROCESS_NAME:
|
|
SpotifyNotify.spotifyPath = SPOTIFY_PROCESS_NAME
|
|
else:
|
|
print "Spotify is not running"
|
|
sys.exit(0)
|
|
|
|
SN = SpotifyNotify(Debug)
|
|
|
|
if options.action:
|
|
action = options.action
|
|
action = action[0:1].upper() + action[1:]
|
|
SN.executeCommand(action)
|
|
|
|
SpotifyNotify.preventDuplicate(Debug)
|
|
|
|
try:
|
|
indicateserver = indicate.indicate_server_ref_default()
|
|
indicateserver.set_type("music.spotify")
|
|
indicateserver.set_desktop_file("/usr/share/applications/spotify.desktop")
|
|
indicateserver.show()
|
|
except:
|
|
pass
|
|
SN.pollChange()
|
|
print "Done"
|
|
|