import sys
import os

def getSuffix(stri):
    return stri[-4:]

def suffixIsMp3(stri):
    suf = getSuffix(stri)
    return suf == ".mp3" or suf == ".Mp3" or suf == ".MP3"

def switch(li):
    try:
        return [li[1], li[0]]
    except:
        return li

os.chdir("C:\Documents and Settings\cweidert\My Documents\karaoke")
a = os.listdir(".")

a = filter(suffixIsMp3, a)

li = []
for fil in a:
    song = fil.split('-')
    for i in range(len(song)):
        song[i] = (song[i].strip()).capitalize()
    li.append(song)

toWrite = ""
toWrite = toWrite + "--------BY ARTIST--------\n"
for spl in li:
    try:
        stri = "%s\t%s\n" % (spl[0], spl[1][:-4])
    except:
        stri = "%s\n" % spl[0][:-4]
    toWrite = toWrite + stri
toWrite = toWrite + "\n\n"

newli = []
for ele in li:
        newli.append(switch(ele))

li = newli
li.sort()

toWrite = toWrite + "--------BY SONG--------\n"
for spl in li:
    try:
        stri = "%s\t%s\n" % (spl[0][:-4], spl[1])
    except:
        stri = "%s\n" % spl[0][:-4]
    toWrite = toWrite + stri

f = open("!songs.txt", "w")

print "printing"

try:
    f.write(toWrite)
    f.close()
except:
    print "some sort of error"

