collect-fingerprints-of-songs.py

#!/usr/bin/python
import os
import sys
import libs
import libs.fingerprint as fingerprint

from termcolor import colored
from libs.reader_file import FileReader
from libs.db_sqlite import SqliteDatabase
from libs.config import get_config

if __name__ == '__main__':
  config = get_config()

  db = SqliteDatabase()
  path = "mp3/"

  # fingerprint all files in a directory

  for filename in os.listdir(path):
    if filename.endswith(".mp3"):
      reader = FileReader(path + filename)
      audio = reader.parse_audio()

      song = db.get_song_by_filehash(audio['file_hash'])
      song_id = db.add_song(filename, audio['file_hash'])

      msg = ' * %s %s: %s' % (
        colored('id=%s', 'white', attrs=['dark']),       # id
        colored('channels=%d', 'white', attrs=['dark']), # channels
        colored('%s', 'white', attrs=['bold'])           # filename
      )
      print msg % (song_id, len(audio['channels']), filename)

      if song:
        hash_count = db.get_song_hashes_count(song_id)

        if hash_count > 0:
          msg = '   already exists (%d hashes), skip' % hash_count
          print colored(msg, 'red')

          continue

      print colored('   new song, going to analyze..', 'green')

      hashes = set()
      channel_amount = len(audio['channels'])

      for channeln, channel in enumerate(audio['channels']):
        msg = '   fingerprinting channel %d/%d'
        print colored(msg, attrs=['dark']) % (channeln+1, channel_amount)

        channel_hashes = fingerprint.fingerprint(channel, Fs=audio['Fs'], plots=config['fingerprint.show_plots'])
        channel_hashes = set(channel_hashes)

        msg = '   finished channel %d/%d, got %d hashes'
        print colored(msg, attrs=['dark']) % (
          channeln+1, channel_amount, len(channel_hashes)
        )

        hashes |= channel_hashes

      msg = '   finished fingerprinting, got %d unique hashes'

      values = []
      for hash, offset in hashes:
        values.append((song_id, hash, offset))

      msg = '   storing %d hashes in db' % len(values)
      print colored(msg, 'green')

      db.store_fingerprints(values)

  print('end')