forge/res/PerSetTracking.py

pathToMtgData = "mtg-data.txt"

############IMPLEMENTATION FOLLOWS############
import os,sys

if not os.path.exists(pathToMtgData) :
        print("This script requires the text version of Arch's mtg-data to be present.You can download it from slightlymagic.net's forum and either place the text version next to this script or edit this script and provide the path to the file at the top.")
        print("Press Enter to exit")
        raw_input("")
        sys.exit()

if not os.path.isdir(sys.path[0] + os.sep + 'PerSetTracking Results') :
        os.mkdir(sys.path[0] + os.sep + 'PerSetTracking Results')

forgeFolderContents = os.listdir(sys.path[0] + os.sep + "cardsfolder")
forgeFolderFiles = []
forgeCards = []
mtgDataCards = {}
setCodes = []
forgeCardCount = 0
mtgDataCardCount = 0
setCodeCount = 0

hasFetchedSets = False
hasFetchedCardName = False
tmpName = ""
line = ""
prevline = ""

#Parse mtg-data
print("Parsing mtg-data")
with open(pathToMtgData) as mtgdata :
        for line in mtgdata :
                if not hasFetchedSets :
                        if line != "\n" :
                                setCodes.append(line[0:3])
                        else :
                                hasFetchedSets = True
                if hasFetchedSets :
                        if not hasFetchedCardName :
                                tmpName = line
                                tmpName = tmpName.rstrip()
                                hasFetchedCardName = True
                        if line == "\n" :
                                mtgDataCards[tmpName] = prevline.rstrip()
                                hasFetchedCardName = False

                prevline = line

#Parse Forge
print("Parsing Forge")
for i in forgeFolderContents :
        if os.path.isfile(sys.path[0] + os.sep + "cardsfolder" + os.sep + i) == True :
                forgeFolderFiles.append(i)
for file in forgeFolderFiles :
        with open(sys.path[0] + os.sep + "cardsfolder" + os.sep + file) as currentForgeCard :
                tmpname = currentForgeCard.readline()
                tmpname = tmpname[5:].replace("AE","Ae")
                tmpname = tmpname.rstrip()
                forgeCards.append(tmpname)

                
#Compare datasets and output results
print("Comparing datasets and outputting results.")
totalData = {}
currentMissing = []
currentImplemented = []
total = 0
percentage = 0
for currentSet in setCodes :
        if currentSet == 'UNH' or currentSet == 'UGL' : continue #skip Unhinged and Unglued since they are only counting basic lands anyway
        #if currentSet == 'VG1' or currentSet == 'VG2' or currentSet == 'VG3' : continue
        #if currentSet == 'VG4' or currentSet == 'VGO' or currentSet == 'VG ' : continue
        #if currentSet == 'SDC' or currentSet == 'AST' or currentSet == 'DKM' : continue
        #if currentSet == 'ATH' or currentSet == 'MBS' : continue
        for card in mtgDataCards.keys() :
                if mtgDataCards[card].count(currentSet) > 0 :
                        if card in forgeCards :
                                currentImplemented.append(card)
                        else :
                                currentMissing.append(card)
        total = len(currentMissing)+len(currentImplemented)
        percentage = 0       
        if total > 0 :
                percentage = (float(len(currentImplemented))/float(total))*100
        
		currentMissing.sort()
		currentImplemented.sort()
		
        with open(sys.path[0] + os.sep + "PerSetTracking Results" + os.sep + "set_" + currentSet + ".txt", "w") as output :
                output.write("Implemented (" + str(len(currentImplemented)) + "):\n")
                for everyImplemented in currentImplemented :
                        output.write(everyImplemented + '\n')
                output.write("\n")
                output.write("Missing (" + str(len(currentMissing)) + "):\n")
                for everyMissing in currentMissing :
                        output.write(everyMissing + '\n')
                output.write("\n")
                output.write("Total: " + str(total) + "\n")
                output.write("Percentage implemented: " + str(round(percentage,2)) + "%\n")
        totalData[currentSet] = (len(currentImplemented),len(currentMissing),total,percentage)
        del currentMissing[:]
        del currentImplemented[:]

#UNTESTED sort sets by percentage completed
totalDataList = sorted(totalData.items(), key=lambda (key,entry): entry[3], reverse=True)
#UNTESTED

totalPercentage = 0
totalMissing = 0
totalImplemented = 0
fullTotal = 0
with open(sys.path[0] + os.sep + "PerSetTracking Results" + os.sep + "CompleteStats.txt", "w") as statsfile:
        statsfile.write("Set: Implemented (Missing) / Total = Percentage Implemented\n")
        for k,dataKey in totalDataList :
                totalImplemented += dataKey[0]
                totalMissing += dataKey[1]
                fullTotal += dataKey[2]
                statsfile.write(k + ": " + str(dataKey[0]) + " (" + str(dataKey[1]) + ") / " + str(dataKey[2]) + " = " + str(round(dataKey[3], 2)) + "%\n")
        totalPercentage = totalImplemented / fullTotal
        statsfile.write("\n")
        statsfile.write("Total over all sets: " + str(totalImplemented) + " (" + str(totalMissing) + ") / " + str(fullTotal))
        
print "Done!"
print "Press Enter to exit."
raw_input("")