ordnung.py

   1 from path import path
   2 import fnmatch
   3 import os
   4 import shutil
   5 import sys
   6
   7 from optparse import OptionParser, OptionGroup
   8 from tag_wrapper import tag
   9
  10 ACCEPTEXTS = [".ogg", ".mp3"] # Keep it simple for now
  11
  12 def issupportedfile(name):
  13     for ext in ACCEPTEXTS:
  14         if os.path.splitext(name)[1] == ext:
  15             return True
  16     return False
  17
  18
  19 def mkdir(newdir):
  20     if not os.path.isdir(newdir):
  21         os.makedirs(newdir)
  22
  23
  24 def newpath(file, pattern):
  25     # Create the tokens dictionary
  26     tokens = {}
  27     tags = tag(file)
  28     # TODO: add tracknumber, disknumber and possibily compilation
  29     for t in ["title", "artist", "album artist", "album", "composer", "genre", "date"]:
  30         try:
  31             tokens[t] = tags[t]
  32         except KeyError:
  33             tokens[t] = None
  34     # %album artist% is %artist% if nothing can be found in the tags
  35     if tokens["album artist"] is None:
  36         tokens["album artist"] = tokens["artist"]
  37
  38     # Now replace all tokens by their values
  39     for i in tokens:
  40         repl = "%" + i + "%"
  41         if tokens[i] is not None:
  42             pattern = pattern.replace(repl, tokens[i][0])
  43
  44     # Add the extension and return the new path
  45     return pattern + os.path.splitext(file)[1]
  46
  47
  48 def safeprint(string):
  49     """
  50     Print string first trying to normally encode it, sending it to the
  51     console in "raw" UTF-8 if it fails.
  52
  53     This is a workaround for Windows's broken console
  54     """
  55     try:
  56         print string
  57     except UnicodeEncodeError:
  58         print string.encode("utf-8")
  59
  60
  61 def files_to_move(base_dir, pattern, recursive=False):
  62     # Figure out which files to move where
  63     basepath = path(base_dir)
  64     if recursive:
  65         files = basepath.walkfiles()
  66     else:
  67         files = basepath.files()
  68
  69     files_return = []
  70
  71     for file in files:
  72         if issupportedfile(file):
  73             t = [ file, newpath(file, pattern) ]
  74             files_return.append(t)
  75
  76     return files_return
  77
  78 def main():
  79     # Pseudo-enum for actions
  80     COPY = 0
  81     MOVE = 1
  82     PREVIEW = 2
  83
  84     # Handle arguments
  85     usage = "Usage: %prog [options] directory pattern"
  86     parser = OptionParser(usage=usage)
  87     parser.add_option("-r", "--recursive", action="store_true",
  88             dest="recursive", help="also move/copy files from sub-directories", default=False)
  89
  90     actions = OptionGroup(parser, "Possible actions")
  91     actions.add_option("--preview", "-p", action="store_const",
  92             const=PREVIEW, dest="action", help="preview, don't make any changes (default)")
  93     actions.add_option("--copy", "-c", action="store_const",
  94             const=COPY, dest="action", help="copy files")
  95     actions.add_option("--move", "-m", action="store_const",
  96             const=MOVE, dest="action", help="move files")
  97     parser.add_option_group(actions)
  98     (options, args) = parser.parse_args()
  99
 100     try:
 101         base = args[0]
 102     except IndexError:
 103         print "You must specify a directory"
 104         sys.exit()
 105
 106     try:
 107         pattern = args[1]
 108     except IndexError:
 109         print "You must specify a pattern"
 110         sys.exit()
 111
 112     # Do stuff
 113     for i in files_to_move(base_dir=base,
 114             pattern=pattern, recursive=options.recursive):
 115         if options.action == MOVE:
 116             mkdir(os.path.split(i[1])[0])
 117             shutil.move(i[0], i[1])
 118         elif options.action == COPY:
 119             mkdir(os.path.split(i[1])[0])
 120             shutil.copy2(i[0], i[1])
 121         else:
 122             safeprint (i[0] + " --> " + i[1])
 123
 124
 125 if __name__ == "__main__":
 126     main()