mirror of
https://github.com/beetbox/beets.git
synced 2025-12-06 08:39:17 +01:00
136 lines
4.3 KiB
Python
136 lines
4.3 KiB
Python
# This file is part of beets.
|
|
# Copyright 2014, 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.
|
|
|
|
"""Send the results of a query to the configured music player as a playlist.
|
|
"""
|
|
from functools import partial
|
|
|
|
from beets.plugins import BeetsPlugin
|
|
from beets.ui import Subcommand
|
|
from beets import config
|
|
from beets import ui
|
|
from beets import util
|
|
from os.path import relpath
|
|
import platform
|
|
import shlex
|
|
from tempfile import NamedTemporaryFile
|
|
|
|
|
|
def play_music(lib, opts, args, log):
|
|
"""Execute query, create temporary playlist and execute player
|
|
command passing that playlist.
|
|
"""
|
|
command_str = config['play']['command'].get()
|
|
use_folders = config['play']['use_folders'].get(bool)
|
|
relative_to = config['play']['relative_to'].get()
|
|
if relative_to:
|
|
relative_to = util.normpath(relative_to)
|
|
if command_str:
|
|
command = shlex.split(command_str)
|
|
else:
|
|
# If a command isn't set, then let the OS decide how to open the
|
|
# playlist.
|
|
sys_name = platform.system()
|
|
if sys_name == 'Darwin':
|
|
command = ['open']
|
|
elif sys_name == 'Windows':
|
|
command = ['start']
|
|
else:
|
|
# If not Mac or Windows, then assume Unixy.
|
|
command = ['xdg-open']
|
|
|
|
# Preform search by album and add folders rather then tracks to playlist.
|
|
if opts.album:
|
|
selection = lib.albums(ui.decargs(args))
|
|
paths = []
|
|
|
|
for album in selection:
|
|
if use_folders:
|
|
paths.append(album.item_dir())
|
|
else:
|
|
# TODO use core's sorting functionality
|
|
paths.extend([item.path for item in sorted(
|
|
album.items(), key=lambda item: (item.disc, item.track))])
|
|
item_type = 'album'
|
|
|
|
# Preform item query and add tracks to playlist.
|
|
else:
|
|
selection = lib.items(ui.decargs(args))
|
|
paths = [item.path for item in selection]
|
|
item_type = 'track'
|
|
|
|
item_type += 's' if len(selection) > 1 else ''
|
|
|
|
if not selection:
|
|
ui.print_(ui.colorize('yellow', 'No {0} to play.'.format(item_type)))
|
|
return
|
|
|
|
# Warn user before playing any huge playlists.
|
|
if len(selection) > 100:
|
|
ui.print_(ui.colorize(
|
|
'yellow',
|
|
'You are about to queue {0} {1}.'.format(len(selection), item_type)
|
|
))
|
|
|
|
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:
|
|
if relative_to:
|
|
m3u.write(relpath(item, relative_to) + '\n')
|
|
else:
|
|
m3u.write(item + '\n')
|
|
m3u.close()
|
|
|
|
command.append(m3u.name)
|
|
|
|
# Invoke the command and log the output.
|
|
output = util.command_output(command)
|
|
if output:
|
|
log.debug(u'Output of {0}: {1}',
|
|
util.displayable_path(command[0]),
|
|
output.decode('utf8', 'ignore'))
|
|
else:
|
|
log.debug(u'no output')
|
|
|
|
ui.print_(u'Playing {0} {1}.'.format(len(selection), item_type))
|
|
|
|
util.remove(m3u.name)
|
|
|
|
|
|
class PlayPlugin(BeetsPlugin):
|
|
|
|
def __init__(self):
|
|
super(PlayPlugin, self).__init__()
|
|
|
|
config['play'].add({
|
|
'command': None,
|
|
'use_folders': False,
|
|
'relative_to': None,
|
|
})
|
|
|
|
def commands(self):
|
|
play_command = Subcommand(
|
|
'play',
|
|
help='send music to a player as a playlist'
|
|
)
|
|
play_command.parser.add_option(
|
|
'-a', '--album',
|
|
action='store_true', default=False,
|
|
help='query and load albums rather than tracks'
|
|
)
|
|
play_command.func = partial(play_music, log=self._log)
|
|
return [play_command]
|