from Tkinter import Tk
import tkFileDialog
import os
root=Tk()
root.withdraw()
def main():
    urlfile=open("vidurls.txt",'w')
    htmlfile=open("buttonhtml.txt",'w')
    maindir=tkFileDialog.askdirectory()
    for root, dirs, files in os.walk(maindir)
        for htmldir in dirs:
            for item in os.listdir(htmldir):
                if item=="index.html":
                    f=open(item).read().split("\n")
                    h1found=False
                    for line in f:
                        if "<h1>" in line:
                            if h1found!=True:
                                h1found=True
                                header=(line.split("<h1>")[1]).split("</h1>")[0]
                            else:
                                if "://" in line and ".mp4" in line:
                                    url="http"+line.split(".mp4")[0].split("http")[1]+".mp4"
                                    urlfile.write('{0}\n'.format(url))
                                    htmlfile.write('<h1><a href="{0}" class="tvshowsButton">{1}</a>\n'.format(url,header))
                                    break
                                else:
                                    pass
if __name__=="__main__":
    main()