seed.py

from model import db, Artist, Song, Playlist, SongPlaylist, YouTubeVideo
from pprint import pprint
from sqlalchemy import exists

import requests
import os


def gets_artist_and_song_json_from_en_api(artist_and_song):
    """If songs aren't in db, get the api jsons"""

    en_key = os.environ['ECHO_NEST_API_KEY']

    print "seed gets json from en api ", artist_and_song

    en_payload = {'title': artist_and_song[1], 'artist': artist_and_song[0]}

    r = requests.get("http://developer.echonest.com/api/v4/song/search?api_key=%(en_key)s&format=json&results=1&" % locals(), params=en_payload)
    
    # Debugging print statement
    print (r.url)

    # binds dictionary from get request to variable
    dict_from_en_api = r.json()

    # Debugging print statement
    pprint(dict_from_en_api)

    return dict_from_en_api
    

def parses_en_json_results(dict_from_en_api):
    ''' Parses EN JSON results'''

    artist_id = dict_from_en_api['response']['songs'][0]['artist_id']
    artist_name = dict_from_en_api['response']['songs'][0]['artist_name']
    song_id = dict_from_en_api['response']['songs'][0]['id']
    song_title = dict_from_en_api['response']['songs'][0]['title']

    # prints for debugging
    print artist_id
    print artist_name
    print song_id
    print song_title

    parsed_search_results = [artist_id, artist_name, song_id, song_title]

    return parsed_search_results


def adds_en_artist_results_to_db(parsed_search_results):
    ''' Seeds the database with artist results from parses_en_json_results'''

    en_artist_id = parsed_search_results[0]
    print en_artist_id

    is_artist_in_db = db.session.query(exists().where(Artist.en_artist_id==en_artist_id)).scalar()

    if is_artist_in_db == True:
        print "Seed, Artist already exists"
    else:
        artist_info = Artist(en_artist_id=parsed_search_results[0],
                             artist_name=parsed_search_results[1])
        db.session.add(artist_info)
        db.session.flush()   
        print "Artist info successfully flushed"    


def adds_en_song_results_to_db(parsed_search_results):
    ''' Seeds the database with artist results from parses_en_json_results'''
    # This absolutely has to run AFTER adding artist to db. Required to
    # have artist to create foreign key
    # RETURNS artist_id to be added to YT table

    print "Running adds_en_song_resuts_to_db"
    print "parsed_search_results ", parsed_search_results

    en_song_id = parsed_search_results[2]
    en_song_id_str = str(en_song_id)    # needs to be in string format to query db
    song_title = parsed_search_results[3]

    artist = parsed_search_results[1]
    en_artist_id = parsed_search_results[0]
    en_artist_id_str = str(en_artist_id)    # needs to db in string format to query db

    is_artist_en_id_in_db = db.session.query(exists().where(Artist.en_artist_id == en_artist_id_str)).scalar()
    # checks to see if artist's en_id is already in db. Will use db info if artist
    # is already in db. Will use EN API info if artist doesn't exist. Previous 
    # queries checked artist as a STRING
    is_song_en_id_in_db = db.session.query(exists().where(Song.en_song_id == en_song_id_str)).scalar()
    # checks to see if song's en_id is already in db. If en_id exists, song is 
    # in db. If it doesn't, add artist. Previous queries checked song as a 
    # STRING

    if is_song_en_id_in_db == True:
        print "seed, adds_en_song_resuts_to_db, Song's EN ID in DB"
        artist_id_result = db.session.query(Artist.artist_id).filter(Artist.en_artist_id == en_artist_id_str).one()
        # takes the EN artist id and queries for the primary key for that artist

        artist_id = artist_id_result[0]

        return artist_id
    else:
        if is_artist_en_id_in_db == True:
            artist_id_result = db.session.query(Artist.artist_id).filter(Artist.en_artist_id == en_artist_id_str).one()
            # takes the EN artist id and queries for the primary key for that artist

            artist_id = artist_id_result[0]

            song_info = Song(en_song_id=en_song_id,
                         song_title=song_title,
                         artist_id=artist_id)

            print "seed, adds en song resuts to db, ", artist_id
            
            db.session.add(song_info)
            db.session.flush()
            print "Song/Artist info successfully flushed"
            return artist_id

        else:
            print """seed, adds_en_song_resuts_to_db, artist with same
                  name exists but en_id is new. Add artist and song to db."""

            artist_info = Artist(en_artist_id=parsed_search_results[0],
                                 artist_name=parsed_search_results[1])

            db.session.add(artist_info)
            db.session.flush()   
            print "seed, adds_en_song_resuts_to_db, Artist info successfully flushed" 

            song_info = Song(en_song_id=en_song_id,
                         song_title=song_title,
                         artist_id=artist_id)

            db.session.add(song_info)
            db.session.flush()
            print "seed, adds_en_song_resuts_to_db, Artist and Song added"
            return artist_id


def generates_en_playlist_session_id(parsed_search_results):
    """Uses the song's primary_id to seed an EchoNest playlist session and adds to db"""
    # parsed_search_results from seed.py, parses_en_json_results(dictfrom_en_api)

    en_song_id = parsed_search_results[2]

    print "api_helper, generates_en_playlist, en_song_id ", en_song_id

    en_key = os.environ['ECHO_NEST_API_KEY']

    en_payload = {'song_id': en_song_id, 'type': 'song-radio'}

    r = requests.get("http://developer.echonest.com/api/v4/playlist/dynamic/create?api_key=%(en_key)s&" % locals(), params=en_payload)
    
    # Debugging print statement
    # print (r.url)

    # binds dictionary from get request to variable
    en_playlist_info = r.json()

    # Debugging print statement
    pprint(en_playlist_info)

    en_session_id = en_playlist_info['response']['session_id']
    print "api_helper, generates_en_playlist_session_id, en_session_id ", en_session_id

    en_session_id_and_en_song_id = [en_session_id, en_song_id]

    return en_session_id_and_en_song_id


def adds_en_session_id_to_db(en_session_id_and_en_song_id):

    en_session_id = en_session_id_and_en_song_id[0]

    is_en_session_id_in_db = db.session.query(exists().where(Playlist.en_session_id==en_session_id)).scalar()

    if is_en_session_id_in_db == True:
        print "Session_id in db"
    else:
        add_en_session_id = Playlist(en_session_id=en_session_id)
        db.session.add(add_en_session_id)
        db.session.flush()

        print "en_session_id added to Playlist db"


def adds_new_artists_to_db_by_en_id(yt_playlist_query):
    """Using the EchoNest Artist ID, adds new artists to db"""
    # yt_playlist_query returned by gets_playlist_history(en_playlist), api_helper.py

    for item in yt_playlist_query:
        en_artist_id = item['en_artist_id']
        is_en_artist_id_in_db = db.session.query(exists().where(Artist.en_artist_id==en_artist_id)).scalar()
        if is_en_artist_id_in_db == False:
            artist_info = Artist(en_artist_id=en_artist_id,
                                 artist_name=item['artist_name'])
            db.session.add(artist_info)
            db.session.flush


def adds_new_songs_to_db_by_en_id(yt_playlist_query):
    """Using the Echonest Song ID, adds new songs to db"""
    # yt_playlist_query returned by gets_playlist_history(en_playlist), api_helper.py

    for item in yt_playlist_query:
        en_song_id = item['en_song_id']
        is_en_song_id_in_db = db.session.query(exists().where(Song.en_song_id == en_song_id)).scalar()
        if is_en_song_id_in_db == False:
            en_artist_id = item['en_artist_id']
            artist_id = db.session.query(Artist.artist_id).filter(Artist.en_artist_id == en_artist_id).one()
            song_info = Song(en_song_id=en_song_id,
                         song_title=item['song_title'],
                         artist_id=artist_id)
            db.session.add(song_info)
            db.session.flush


def adds_yt_video_info_to_db(contains_yt_playlist_info):
    """Adds unique videos to youtube_videos table"""
    
    # parameter from creates_yt_video_playlist(yt_playlist_query):
    # in youtube.py

    print 'SEED.PY, adds_yt_video_info_to_db, contains_yt_playlist_info ', contains_yt_playlist_info

    for video_item in contains_yt_playlist_info:
        yt_video_id = video_item['yt_video_id']
        video_title = video_item['video_title']
        video_thumbnail = video_item['video_thumbnail']
        searched_artist = video_item['searched_artist']
        searched_song = video_item['searched_song']
        artist_id = video_item['artist_id']

        does_video_exist = db.session.query(exists().where(YouTubeVideo.yt_video_id == yt_video_id)).scalar()

        if does_video_exist:
            print "Video in db"
        else:
            print "Video doesn't exist. Ading to db"
            video_info = YouTubeVideo(yt_video_id=yt_video_id,
                                      video_title=video_title,
                                      video_thumbnail=video_thumbnail,
                                      searched_artist=searched_artist,
                                      searched_song=searched_song,
                                      artist_id=artist_id)

            db.session.add(video_info)
            db.session.flush()
        print "youtube, adds_yt_song_results_to_db, Video and artist_id successfully flushed to database."


##############################################

# if __name__ == "__main__":

#     connect_to_db(app)