#!/usr/bin/env python

import itertools

import os

import pickle

import time

from pyechonest import config

from pyechonest import song

from pyechonest import track

from pyechonest import util

from subprocess import check_output

# NOTE: We treat echonest's c-sharp as d-flat for the camelot mixing system.

camelot_scale = {'Ab minor': '1A', 'B major': '1B',

'C# minor': '12A', 'E major': '12B'}

chromatic_scale = ['C', 'C#', 'D', 'Eb', 'E', 'F',

'F#', 'G', 'Ab', 'A', 'Bb', 'B']

mode_enum = ['minor', 'major']

track_metadata = {}

track_missing = []

mixtape_files = []

PICKLE_BACKUP = os.path.expanduser('~/.mixtape_data')

def initialize():

with open(os.path.expanduser('~/.echonest'), 'r') as f:

config.ECHO_NEST_API_KEY = f.readline()

def get_md5sum(path):

return check_output(['md5sum', path]).split()[0]

def short_path(path):

return os.path.basename(path)

def add_mixtape_song(path):

print "The file '%s' was not found." % path

def add_mixtape_songs():

with open(os.path.expanduser('~/.mixtape_files'), 'r') as f:

for path in f.read().splitlines():

add_mixtape_song(path)

def get_track_data(path):

print "Couldn't find track '%s'. Try song.search?" % shortname

def group_by(n, iterable):

args = [iter(iterable)] * n

return itertools.izip_longest(*args)

def get_echonest_data():

time.sleep(60)

def display_metadata(path, tempo, key, mode):

formatter = "{0} is {1} bpm, Camelot block {2}; ({3})."

key_mode = "{0} {1}".format(chromatic_scale[key], mode_enum[mode])

print formatter.format(path, tempo, camelot_scale[key_mode], key_mode)

def show_results():

display_metadata(path, metahash['tempo'], metahash['key'], metahash['mode'])

## Helpers

def check_song(artist, title):

meta['mode'])

def save_metadata():

with open(PICKLE_BACKUP, 'w') as f:

pickle.dump(track_metadata, f)

def load_metadata():

with open(PICKLE_BACKUP, 'r') as f:

return pickle.load(f)

def create_track_from_path(path):

print "Could not create track from file"

def main():

add_mixtape_songs()

get_echonest_data()

show_results()

initialize()

from IPython import embed; embed()