Plugins: copyfiles.2.py

# Copyright 2008 Felix McCoey <felix.mccoey@gmail.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation

import os
import gtk
from plugins.songsmenu import SongsMenuPlugin
import config
import shutil
from parse import FileFromPattern

try: config.get("plugins", __name__+'_link')
except: config.set("plugins", __name__+'_link', 'False')
try: config.get("plugins", __name__+'_pattern')
except: config.set("plugins", __name__+'_pattern', '/tmp/<artist>/<tracknumber>_<title>')
try: config.get("plugins", __name__+'_spaces')
except: config.set("plugins", __name__+'_spaces', 'True')

class CopyFiles(SongsMenuPlugin):
  PLUGIN_ID = "Copy Files"
  PLUGIN_NAME = _("Copy Files")
  PLUGIN_DESC = _("Copy (or link) the selected song list to a directory")
  PLUGIN_ICON = gtk.STOCK_CONVERT
  PLUGIN_VERSION = "0.2"

  c_link = __name__ + '_link'
  c_pattern = __name__ + '_pattern'
  c_spaces = __name__ + '_spaces'

  def plugin_songs(self, songs):
    if not songs: return

    msgbox = gtk.MessageDialog(
        parent=None,
        flags=0, 
        type=gtk.MESSAGE_INFO, 
        buttons=gtk.BUTTONS_OK, 
        message_format=None)

    try:
      for song in songs:
        self.__copy(song)
    except Exception, e:
      msgbox.set_markup("An error occurred: %s" % e)
    else:
      msgbox.set_markup("Copy successful")

    msgbox.run()
    msgbox.destroy()

  def __copy(self, song):
    pattern = config.get('plugins', self.c_pattern)
    target = FileFromPattern(pattern) % song
    if config.getboolean('plugins', self.c_spaces):
      target = target.replace(" ","_")
    source_file = song["~filename"]
    # make the path
    try: 
      os.makedirs(os.path.dirname(target))
    except: pass

    make_links = config.getboolean('plugins', self.c_link)
    try:
      if make_links:
        os.symlink(source_file,target)
      else:
        shutil.copyfile(source_file,target)
    except: raise

  def PluginPreferences(self, parent):
    def pattern_changed(entry):
      config.set('plugins', self.c_pattern, entry.get_text())

    def link_changed(entry):
      config.set('plugins', self.c_link, str(entry.get_active()))

    def spaces_changed(entry):
      config.set('plugins', self.c_spaces, str(entry.get_active()))

    tooltips = gtk.Tooltips().set_tip
    vb = gtk.VBox(spacing = 3)
    pattern_box = gtk.HBox(spacing = 3)
    pattern_box.set_border_width(3)
    pattern = gtk.Entry()
    pattern.set_text(config.get('plugins',self.c_pattern))
    tooltips(pattern, "Enter the absolute path, using tags if need be, for the target location.")
    pattern_box.pack_start(gtk.Label("Destination:"), expand = False)
    pattern_box.pack_start(pattern)

    link = gtk.CheckButton(label="Link files, don't copy")
    link.set_active(config.getboolean('plugins', self.c_link))
    tooltips(link, "Symbolically link the files instead of copying them.")

    spaces = gtk.CheckButton(label="Convert spaces to underscores")
    spaces.set_active(config.getboolean('plugins', self.c_spaces))
    tooltips(spaces, "Convert spaces in the path to underscores.")

    pattern.connect('changed', pattern_changed)
    link.connect('toggled',link_changed)
    spaces.connect('toggled',spaces_changed)

    vb.pack_start(pattern_box)
    vb.pack_start(link)
    vb.pack_start(spaces)
    return vb

  PluginPreferences = classmethod(PluginPreferences)