scripts/irename.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3 from django.core.management import setup_environ
   4 from wolnelektury import settings
   5 import sys
   6 from os.path import abspath, join, dirname, splitext
   7 import os
   8
   9 # Add apps and lib directories to PYTHONPATH
  10 sys.path.insert(0, abspath(join(dirname(__file__), 'apps')))
  11 sys.path.insert(0, abspath(join(dirname(__file__), 'lib')))
  12
  13 setup_environ(settings)
  14
  15 from catalogue.models import Book
  16 from mutagen import easyid3
  17 from slughifi import slughifi
  18
  19 chosen_book_slugs = set()
  20
  21 for file_name in os.listdir('mp3'):
  22     base_name, ext = splitext(file_name)
  23     if ext != '.mp3':
  24         continue
  25
  26     audio = easyid3.EasyID3(join('mp3', file_name))
  27     title = audio['title'][0]
  28     artist = title.split(',', 1)[0].strip()
  29     artist_slug = slughifi(artist)
  30     title_part = slughifi(title.rsplit(',', 1)[1].strip())
  31
  32     print "--------------------"
  33     print "File: %s" % file_name
  34     print "Title: %s" % title
  35     print
  36     print "Matching books:"
  37
  38     matching_books = [book for book in Book.tagged.with_all(artist_slug) if book.slug not in chosen_book_slugs]
  39     matching_books = [book for book in matching_books if title_part in book.slug]
  40
  41     if len(matching_books) > 1:
  42         for i, book in enumerate(matching_books):
  43             print "%d: %s (%s)" % (i, book.title, ', '.join(tag.slug for tag in book.tags))
  44         print
  45         i = int(input("Choose which book is read in this file:"))
  46     elif len(matching_books) == 1:
  47         i = 0
  48     else:
  49         print "Skipping %s: No matching book found" % file_name
  50         continue
  51
  52     print "You chose %d (%s)" % (i, matching_books[i].slug)
  53
  54     chosen_book_slugs.add(matching_books[i].slug)
  55     os.rename(join('mp3', file_name), join('new_mp3', matching_books[i].slug + '.mp3'))
  56     os.rename(join('oggvorbis', base_name + '.ogg'), join('new_ogg', matching_books[i].slug + '.ogg'))
  57
  58