import os
from mutagen.easyid3 import EasyID3
from mutagen.mp3 import MP3
import shutil
import eyeD3
import MySQLdb
import mysql.connector

path = '/home/lissa/Desktop/m/'
os.chdir(path)
files = os.listdir(path)
mp3_files = []
for i in files:
    if os.path.isfile(path + i):
        mp3_files.append(i)
conn = mysql.connector.connect(host = 'localhost', database = 'catalog', user = 'root', password = '123456')
cursor = conn.cursor()
cursor.execute("DROP TABLE music")
cursor.execute("create table `music`(`id` int, `name` varchar(50), `artist` varchar(50), `album` varchar(50), `genre` varchar(50), `year` int, `size` int)")


for i in range(len(mp3_files)):
    #audio = MP3(i, ID3=EasyID3)
    tag = eyeD3.Tag()
    tag.link(path + mp3_files[i])
    #song = "{0} - {1}".format(tag.getArtist(), tag.getTitle())
    #os.rename(mp3_files[i], song)

    #cursor.execute('update music set id = i')
    n = "{0} - {1}".format(tag.getArtist(), tag.getTitle())
    #cursor.execute('update music set name = "fddffd"')
    ar = "{0}".format(tag.getArtist())
    print ar
    query  = 'update music set artist =' + "'" + ar + "'"
    print query
    cursor.execute(query)
    al = "{0}".format(tag.getAlbum())
    g = "{0}".format(tag.getGenre())
    #cursor.execute('update music set album = al')
    y = "{0}".format(tag.getYear())
    int(y)
    #cursor.execute('update music set year = y')
    s = "{0}".format(os.path.getsize(mp3_files[i]))
    int(s)
    #cursor.execute('update music set size = s')
    cursor.execute('insert into music(id, name, artist, album, genre, year, size) values(%s, %s, %s, %s, %s, %s, %s)', (i, n, ar, al, g, y, s))


def normal(my_tag):
    n_tag = ''
    if ' /' in my_tag:
        n_tag = my_tag.replace('/', ',')
    if '/' in my_tag:
        n_tag = my_tag.replace('/', ',')
    else:
        n_tag = my_tag
    return n_tag

def creating_folder(my_tag):
    if my_tag == 'None':
        if os.path.exists('Do it by yourself or go away') == False:
            os.mkdir(path + 'Do it by yourself or go away')

    elif os.path.exists(path+my_tag) == False:
        os.mkdir(path+my_tag)


sorted_tag = raw_input('Print it! ')

for i in mp3_files:
    tag = eyeD3.Tag()
    tag.link(path+i)
    if sorted_tag == 'Genre':
        my_tag = '{0}'.format(tag.getGenre())
        good_tag = normal(my_tag)
        #tag.clear()
        #tag.setGenre(good_tag)

        creating_folder(good_tag)
    if sorted_tag == 'Album':
        my_tag = '{0}'.format(tag.getAlbum())
        good_tag = normal(my_tag)
        #tag.clear()
        #tag.setAlbum(good_tag)
        creating_folder(good_tag)
    if sorted_tag == 'Year':
        my_tag = '{0}'.format(tag.getYear())
        creating_folder(my_tag)
    if sorted_tag == 'Artist':
        my_tag = '{0}'.format(tag.getArtist())
        good_tag = normal(my_tag)
        #tag.clear()
        #tag.setArtist(good_tag)
        creating_folder(good_tag)


'''
for i in mp3_files:
   audio = MP3(i, ID3=EasyID3)
   if 'genre' in audio:
       for n in audio['genre']:
           if '/' in n:
               n = n.replace('/', ',')
       if os.path.exists(path+n) == False:
           os.mkdir(path+n)
   elif os.path.exists('Do it by yourself or go away') == False:
           os.mkdir(path + 'Do it by yourself or go away')
           os.path.splitext("path_to_file")[0]'''


files = os.listdir(path)
folders = []

for i in files:
    if os.path.isdir(path + i):
        folders.append(i)

for i in files:
    if os.path.isfile(path+i):
        tag = eyeD3.Tag()
        tag.link(path + i)
        if sorted_tag == 'Genre':
            m = '{0}'.format(tag.getGenre())
            moving = normal(m)
        if sorted_tag == 'Album':
            m = '{0}'.format(tag.getAlbum())
            moving = normal(m)
        if sorted_tag == 'Year':
            m = '{0}'.format(tag.getYear())
            moving = normal(m)
        if sorted_tag == 'Artist':
            m = '{0}'.format(tag.getArtist())
            moving = normal(m)


        for v in folders:

            if v == moving:
                song = "{0} - {1}".format(tag.getArtist(), tag.getTitle())
                os.rename(i, song)
                shutil.move(song, path+v)
                break
            if moving == 'None':
                song = "{0} - {1}".format(tag.getArtist(), tag.getTitle())
                os.rename(i, song)
                shutil.move(song, path+'Do it by yourself or go away')
                break

'''
for i in mp3_files:
    audio = MP3(i, ID3=EasyID3)
    if 'genre' in audio:
        for n in audio['genre']:
            if '/' in n:
                n = n.replace('/', ',')

        for v in folders:
            if v == n:
                for q in audio['title']:
                    shutil.move(i, path + v)
    else:

        for q in audio['title']:
            shutil.move(i, path + 'Do it by yourself or go away')'''