beets/beetsplug/play.py
2014-04-19 08:52:18 -03:00

114 lines
3.5 KiB
Python

# This file is a plugin on beets.
# Copyright (c) <2013> David Hamp-Gonsalves
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
"""Sends the results of a query to the configured music player as a playlist.
"""
from beets.plugins import BeetsPlugin
from beets.ui import Subcommand
from beets import config
from beets import ui
import platform
import subprocess
import os
from tempfile import NamedTemporaryFile
def play_music(lib, opts, args):
"""Execute query, create temporary playlist and execute player
command passing that playlist.
"""
command = config['play']['command'].get()
is_debug = config['play']['debug'].get()
# If a command isn't set then let the OS decide how to open the playlist.
if not command:
sys_name = platform.system()
if sys_name == 'Darwin':
command = 'open'
elif sys_name == 'Windows':
command = 'start'
else:
# If not Mac or Win then assume Linux(or posix based).
command = 'xdg-open'
# Preform search by album and add folders rather then tracks to playlist.
if opts.album:
albums = lib.albums(ui.decargs(args))
paths = []
for album in albums:
paths.append(album.item_dir())
itemType = 'album'
# Preform item query and add tracks to playlist.
else:
paths = [item.path for item in lib.items(ui.decargs(args))]
itemType = 'track'
itemType += 's' if len(paths) > 1 else ''
if not paths:
ui.print_(ui.colorize('yellow', 'no {0} to play.'.format(itemType)))
return
# Warn user before playing any huge playlists.
if len(paths) > 100:
ui.print_(ui.colorize('yellow',
'do you really want to play {0} {1}?'
.format(len(paths), itemType)))
if ui.input_options(('Continue', 'Abort')) == 'a':
return
# Create temporary m3u file to hold our playlist.
m3u = NamedTemporaryFile('w', suffix='.m3u', delete=False)
for item in paths:
m3u.write(item + '\n')
m3u.close()
# Prevent player output from poluting our console(unless debug is on).
if not is_debug:
FNULL = open(os.devnull, 'w')
subprocess.Popen([command, m3u.name],
stdout=FNULL, stderr=subprocess.STDOUT)
FNULL.close()
else:
subprocess.Popen([command, m3u.name])
ui.print_('playing {0} {1}.'.format(len(paths), itemType))
class PlayPlugin(BeetsPlugin):
def __init__(self):
super(PlayPlugin, self).__init__()
config['play'].add({
'command': None,
'debug': False
})
def commands(self):
play_command = Subcommand('play',
help='send query results to music player as playlist.')
play_command.parser.add_option('-a', '--album',
action='store_true', default=False,
help='query and load albums(folders) rather then tracks.')
play_command.func = play_music
return [play_command]