import re
import sys
import os
input_filenames = os.listdir(sys.argv[1])
input_filenames = [x for x in input_filenames if ".xml" in x]
filenames = [x for x in input_filenames if "intact" in x]
a0 = []
a15 = []
a30 = []
a45 = []
a60 = []
b0 = [0,0,0,0,0,0,0]
b15 = [0,0,0,0,0,0,0]
b30 = [0,0,0,0,0,0,0]
b45 = [0,0,0,0,0,0,0]
b60 = [0,0,0,0,0,0,0]

ans = input("modded? (y/n)")

if ans == "y":
    for i in range(len(filenames)):
        check = re.split("\W+|_", filenames[i])
        flex = int(check[2])
        if flex == 0:
            a0.append(filenames[i])
        elif flex == 15:
            a15.append(filenames[i])
        elif flex == 30:
            a30.append(filenames[i])
        elif flex == 45:
            a45.append(filenames[i])
        elif flex == 60:
            a60.append(filenames[i])

    listlist = [a0,a15,a30,a45,a60]
    listlistb = [b0,b15,b30,b45,b60]

    for i in range(len(listlist)):
        for j in range (len(listlist[i])):
            check = re.split("\W+|_", listlist[i][j])
            k = int(int(check[3]) / 100)
            if check[4] == 'M':
                v = 1
            else:
                v = int(check[4])
            if listlistb[i][k] == 0:
                listlistb[i][k] = listlist[i][j]
            else:
                check2 = re.split("\W+|_", listlistb[i][k])
                if int(check2[4]) >= v:
                    listlistb[i][k] = listlistb[i][k]
                else:
                    listlistb[i][k] = listlist[i][j]
    textfilename = sys.argv[1]+'/filelist.txt'
    text_file = open(textfilename, "w")
    text_file.truncate()
    for i in range(len(listlistb)):
        for j in range(len(listlistb[i])):
            text_file.write(listlistb[i][j]+"\n")
else:
    textfilename = sys.argv[1] + '/filelist.txt'
    text_file = open(textfilename, "w")
    text_file.truncate()
    for i in range(len(filenames)):
        text_file.write(filenames[i]+"\n")
text_file.close()