beets/beetsplug/freedesktop.py
2014-11-01 21:47:55 -05:00

43 lines
1.1 KiB
Python

from beets.plugins import BeetsPlugin
from beets.ui import Subcommand
from beets.ui import decargs
import os.path
freedesktop_command = Subcommand("freedesktop", help="Create .directory files")
def process_query(lib, opts, args):
for album in lib.albums(decargs(args)):
albumpath = album.item_dir()
fullartpath = album.artpath
artfile = os.path.split(fullartpath)[1]
create_file(albumpath, artfile)
def create_file(albumpath, artfile):
file_contents = "[Desktop Entry]\nIcon=./" + artfile
outfilename = os.path.join(albumpath, ".directory")
if not os.path.exists(outfilename):
file = open(outfilename, 'w')
file.write(file_contents)
file.close()
pass
freedesktop_command.func = process_query
class FreedesktopPlugin(BeetsPlugin):
"""Creates freedesktop.org-compliant .directory files on an album level.
"""
def __init__(self):
super(FreedesktopPlugin, self).__init__()
self.config.add({
'auto': False
})
def commands(self):
return [freedesktop_command]