C[omp]ute

From: "andrew cooke" <andrew@...>

Date: Sun, 28 Dec 2008 18:02:10 -0300 (CLST)

If you download music you end up with a complete mess of directories.  In
contrast, my own ripped albums are all organised neatly by artist and
album name.  The following script takes all the files below whatever
directory it is run in, and moves them to another directory (hardcoded
below, but you can change it) based on the ID3 tags.

#! /usr/bin/env python

from __future__ import with_statement
from os import walk, getcwd, mkdir
from os.path import join, exists
from shutil import copy
from sys import argv, exc_info
import logging
from mutagen.easyid3 import EasyID3

def main():
  # convert all files below cwd (if not already done)
  dest = '/home/andrew/projects/personal/compressed-music/mp3'
  source = getcwd()
  logging.info('fixing ' + source)
  for root, dirs, files in walk(source):
    for file in files:
      if file.lower().endswith('.mp3'):
        fix(root, file, dest)

def fix(root, name, dest):
  path = join(root, name)
  tags = EasyID3(path)
  try:
    artist = tags['artist'][0]
    assert artist, 'No artist tag for ' + path
    album = tags['album'][0]
    assert album, 'No album tag for ' + path
    title = tags['title'][0]
    assert title, 'No title tag for ' + path
    index = tags['tracknumber'][0]
    i = index.find('/')
    if i > 0:
      index = index[0:i]
    index = int(index)
    assert index, 'No track number tag for ' + path
    dest = create_dir(dest, artist)
    dest = create_dir(dest, album)
    copy_file(root, name, dest, artist, index, title)
  except:
    logging.info('failed to process ' + path)
    logging.error(exc_info())
    print path
    print exc_info()

def create_dir(root, extra):
  path = join(root, extra)
  if not exists(path):
    mkdir(path)
  return path

def copy_file(source, name, destn, artist, index, title):
  path_source = join(source, name)
  path_destn = join(destn, '%s - %02d - %s.mp3' % (artist, index, title))
  if exists(path_destn):
    logging.debug(name + ' already exists')
  else:
    logging.debug('copying ' + name + ' to ' + path_destn)
    copy(path_source, path_destn)
  # copy associated data
  for root, dirs, files in walk(source):
    for file in files:
      if not file.lower().endswith('.mp3'):
        path_source = join(root, file)
        path_destn = join(destn, file)
        if not exists(path_destn):
          logging.debug('copying ' + file + ' to ' + path_destn)
          copy(path_source, path_destn)



if __name__ == "__main__":
  main()