#!/usr/bin/python
# -*- coding: utf-8 -*- 
import os
import mimetypes
import string
import sys
import shutil
import getopt
import re
 
class main:
 
    def usage(self):
        print('%s %s dir te explore'%('-i', '--indir'))
        print('%s %s playlist output'%('-p', '--playlist'))
        print('exemple: %s %s %s'%(__file__, '--indir=/var/www/mp3', '-p ./blu.xml'))
 
    def __init__(self, argv):
        self.indir = 0
        self.playlist = 'playlist.xml'
        try:
            opts, args = getopt.getopt(argv, "i:p", ["indir=", "playlist="])
        except getopt.GetoptError:
            self.usage()
            sys.exit(2)
 
        for opt, arg in opts:
            if opt in ('-i', '--indir'):
                if(os.path.exists(arg) == 0):
                    sys.exit(1)
                print('walk %s'%(arg))
                self.indir = arg
            elif opt in ('-p', '--playlist'):
                print('playlist %s'%(arg))
                self.playlist = arg
 
        if self.indir != 0:
            self.run()
        else:
            self.usage()
 
    def strReplace(self, str, array):
        for search, replace in array:
            str = str.replace(search, replace)
        return str
 
    def _human_key(self, key):
        parts = re.split('(\d*\.\d+|\d+)', key)
        return tuple((e.swapcase() if i % 2 == 0 else float(e))
                     for i, e in enumerate(parts))
 
    def run(self):
        print 'run'
        xml = '<?xml version="1.0" encoding="UTF-8"?>\n'
        xml += '<playlist version="1" xmlns="http://xspf.org/ns/0/">\n'
        xml += '<title>'+os.path.split(self.playlist)[1].replace('_', ' ').replace('.xml', '')+'</title>\n'
        xml += '<creator>Bayard</creator>\n'
        xml += '<link></link>\n'
        xml += '<info></info>\n'
        xml += '<image></image>\n'
        xml += '<trackList>\n'
        for dirpath, dirnames, filenames in os.walk(self.indir):
            c = os.path.split(dirpath)[1]
            filenames.sort(key=self._human_key)
            for filename in filenames:
                print dirpath+'/'+filename
                newfilename = self.strReplace(filename, [(' ', '_'), ('–', '-')]);
                fo_path = os.path.join(dirpath, filename)
                fn_path = os.path.join(dirpath, newfilename)
                f_mime = mimetypes.guess_type(fo_path)
                if(os.path.exists(fn_path) == 0):
                    print('%s move to %s'%(fo_path, fn_path))
                    shutil.move(fo_path, fn_path)      
                if f_mime[0] == 'audio/mpeg':
                    xml += '  <track>\n'
                    xml += '    <title>'+newfilename.replace('_', ' ').replace('.mp3', '')+'</title>\n';
                    xml += '    <location>http://bayardmusique.bayardcdn.com/'+c+'/'+filename+'</location>\n';
                    xml += '  </track>\n'
        xml += '</trackList>\n'
        xml += '</playlist>'
        self.write(xml)
 
    def write(self, xml):
        print('write %s'%(self.playlist))
        f = open(self.playlist, 'w')
        f.write(xml)
        f.close()
 
 
if __name__ == "__main__":
    main(sys.argv[1:])