UE4 wwnames

import os

def collectNames(f,nameType,dirName):
    f.write("# "+nameType+"\n")
    for dir0, dirs, ff in os.walk(dirName):
        for fname in ff:
            name,ext=os.path.splitext(fname)
            if ext!=".uasset":
                continue
            if name.endswith("_FOLDER"):
                continue

            f.write(name+"\n")
    f.write("\n\n")

def collectHierarchyNames(f,nameType,dirName):
    f.write("# "+nameType+"\n")
    for dir0, dirs, ff in os.walk(dirName):
        for dname in dirs:
            f.write(dname+"\n")
    f.write("\n\n")

def collectGroupNames(f,nameType,dirName):
    f.write("# "+nameType+"\n")
    for dir0, dirs, ff in os.walk(dirName):
        wroteHeader=False
        for fname in ff:
            name,ext=os.path.splitext(fname)
            if ext!=".uasset":
                continue
            if name.endswith("_FOLDER"):
                continue

            group,val=name.split("-")
            if not wroteHeader:
                f.write(group+"\n")
                wroteHeader=True
            f.write("\t"+val+"\n")
    f.write("\n\n")

if __name__=="__main__":
    f=open("wwnames.txt","w")
    collectNames(f,"Events","Events")
    locDir="Localized"
    if os.path.isdir(locDir):
        for lang in os.listdir(locDir):
            collectNames(f,"Events (%s)" % (lang), os.path.join(locDir,lang,"Events"))

    collectNames(f,"RTPC","Game_Parameters")
    collectHierarchyNames(f,"Buses","Master-Mixer_Hierarchy")
    collectGroupNames(f,"States","States")
    collectGroupNames(f,"Switches","Switches")
    collectNames(f,"Triggers","Triggers")
    f.close()