import os

path = "E:\Tv Shows"

types = ("avi", "mpg", "mpeg", "divx", "mp4", "ogm", "mkv", "vob", "AVI",
         "MPG", "MPEG", "DIVX", "MP4", "OGM", "MKV", "VOB")

if __name__ == '__main__':
    fp = open('media.html', 'w')
    mediaData = os.walk(path)
    for info in mediaData:
        closeUL = 0
        if not info[1]:
            fp.write('<li class="treenode">\n\t<a>' + info[0] + "</a>\n\t<br />\n\t<ul>")
            closeUL = 1
        if info[2]:
            for vid in info[2]:
                for t in types:
                    trimmer = 0 - len(t)
                    if vid[trimmer:] in t:
                        epWrite = '''
                <li>
                    <form method="get" action="?" enctype="text/plain">
                    <input type="hidden" name="control"  value="add">
                    <input type="hidden" name="mrl"        value="%s">
                    <input type="image" value="submit" src="images/play2.gif" align="middle">
                    <a>%s</a>
                    </form>
                </li>''' % (os.path.join(info[0], vid), vid[:trimmer - 1])
                        fp.write(epWrite + "\n")
        if closeUL:
            fp.write("\t</ul>\n</li>\n")
    fp.close()