TierUpdate9NatDex.py

import string
import sys
import json
import pickle
from common import keyify,readTable,getBattleFormatsData
reload(sys)
sys.setdefaultencoding('utf8')

def getUsage(filename,col,weight,usage):
	tempUsage, nBattles = readTable(filename)
	for i in tempUsage:
		if keyify(i) not in usage:
			usage[keyify(i)]=[0,0,0,0,0]
		if i != 'empty':
			usage[keyify(i)][col] = usage[keyify(i)][col]+weight*6.0*tempUsage[i]/sum(tempUsage.values())/24

def makeTable(table,name,keyLookup):

	print("[HIDE="+name+"][CODE]")
	print("Combined usage for "+name)
	print(" + ---- + ------------------ + ------- + ")
	print(" | Rank | Pokemon            | Percent | ")
	print(" + ---- + ------------------ + ------- + ")
	print(' | %-4d | %-18s | %6.3f%% |' % (1,keyLookup[table[0][0]],table[0][1]*100))
	for i in range(1,len(table)):
		if table[i][1] < 0.001:
			break
		print(' | %-4d | %-18s | %6.3f%% |' % (i+1,keyLookup[table[i][0]],100.0*table[i][1]))
	print(" + ---- + ------------------ + ------- + ")
	print("[/CODE][/HIDE]")

tiers = ['Uber','OU','UUBL','UU','RUBL','RU','NUBL','NU','PUBL','PU','ZUBL','ZU', 'New']
usageTiers = ['nationaldex', 'nationaldexuu']

def main(months):
	file = open('keylookup.json', 'rb')
	keyLookup = json.load(file)
	file.close()

	rise =  [0.08827751140,0.06715839608,0.04515839608][len(months)-1]
	drop =  [0.01528524706,0.02284003156,0.04515839608][len(months)-1]

	rise =  [0.99515839608,0.99515839608,0.04515839608][len(months)-1]
	rise =  [0.04515839608,0.04515839608,0.04515839608][len(months)-1]
	#rise =  [0.99515839608,0.99515839608,0.99515839608][len(months)-1]
	drop =  [0.04515839608,0.04515839608,0.04515839608][len(months)-1]
	#drop =  [0.02,0.02284003156,0.04515839608][len(months)-1]

	formatsData = getBattleFormatsData()

	curTiers = {}
	NFE=[]
	for poke in formatsData:
		if poke in ['pichuspikyeared', 'unownb', 'unownc', 'unownd', 'unowne', 'unownf', 'unowng', 'unownh', 'unowni', 'unownj', 'unownk', 'unownl', 'unownm', 'unownn', 'unowno', 'unownp', 'unownq', 'unownr', 'unowns', 'unownt', 'unownu', 'unownv', 'unownw', 'unownx', 'unowny', 'unownz', 'unownem', 'unownqm', 'burmysandy', 'burmytrash', 'cherrimsunshine', 'shelloseast', 'gastrodoneast', 'deerlingsummer', 'deerlingautumn', 'deerlingwinter', 'sawsbucksummer', 'sawsbuckautumn', 'sawsbuckwinter', 'keldeoresolution', 'genesectdouse', 'genesectburn', 'genesectshock', 'genesectchill', 'basculinbluestriped', 'darmanitanzen','keldeoresolute','pikachucosplay']:
			continue
		#if 'isNonstandard' in formatsData[poke]:
		#	if formatsData[poke]['isNonstandard']:
		#		continue
		#if 'requiredItem' in formatsData[poke]:
		#		continue
		#if poke == 'rayquazamega':
		#	continue
		if 'tier' not in formatsData[poke].keys():
			continue
		old = formatsData[poke]['tier']
		if 'natDexTier' in formatsData[poke].keys():
			old = formatsData[poke]['natDexTier']
		if old[0] == '(' and old[1] == 'P':
			old = 'ZU'
		if old in ['NFE','LC','LC Uber']:
			NFE.append(poke)
		if old == 'Illegal' or old == 'Unreleased':
			continue
		#if old == 'RU': # !!! edit me !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
		#	old = 'New'
		elif old not in tiers and poke not in NFE and old[0] != '(':
			old = tiers[-1]
		curTiers[poke]=old

	usage = {} #track usage across all relevant tiers [OU,UU,RU,NU]

	for i in range(len(months)):
		for j in range(len(usageTiers)):		
			n = {}
			u = {}

			baseline = "1630"
			if usageTiers[j] in ['ou']:
				baseline = "1630"
			for k in ('', 'suspecttest', 'alpha', 'beta'):
				try:
					u[k], n[k] = readTable(months[i]+"/Stats/gen9"+usageTiers[j]+k+"-"+baseline+".txt")

				except IOError:
					pass
			ntot = sum(n.values())
			
			for k in u:
				for poke in u[k]:
					if keyify(poke) not in usage:
						usage[keyify(poke)]=[0]*len(usageTiers)
					if poke != 'empty':
						usage[keyify(poke)][j] += n[k]/ntot*u[k][poke]/len(months)

	#generate three-month tables and start working on that new tier list

	OU = []
	UU = []
	#RU = []
	#NU = []
	#PU = []
	
	for i in usage:
		if usage[i][0] > 0.0:
			OU.append([i,usage[i][0]])
		if usage[i][1] > 0.0:
			UU.append([i,usage[i][1]])
		#if usage[i][2] > 0.0:
		#	RU.append([i,usage[i][2]])
		#if usage[i][3] > 0.0:
		#	NU.append([i,usage[i][3]])
		#if usage[i][4] > 0.0:
		#	PU.append([i,usage[i][4]])

	OU = sorted(OU, key=lambda OU:-OU[1])
	UU = sorted(UU, key=lambda UU:-UU[1])
	#RU = sorted(RU, key=lambda RU:-RU[1])
	#NU = sorted(NU, key=lambda NU:-NU[1])
	#PU = sorted(PU, key=lambda PU:-PU[1])

	makeTable(OU,"National Dex OU (1630 stats)",keyLookup)
	makeTable(UU,"National Dex UU (1630 stats)",keyLookup)
	#makeTable(RU,"RU (1630 stats)",keyLookup)
	#makeTable(NU,"NU (1630 stats)",keyLookup)
	#makeTable(PU,"PU (1630 stats)",keyLookup)

	newTiers={}
	#start with Ubers
	for poke in curTiers.keys():
		if curTiers[poke] == 'Uber':
			newTiers[poke] = 'Uber'

	for poke in curTiers.keys():
		if poke not in usage:
			newTiers[poke] = curTiers[poke]

	#next do the OU rises
	for poke in curTiers.keys():
		if poke not in usage:
			continue
		if usage[poke][0] > rise and poke not in newTiers.keys():
			newTiers[poke] = 'OU'

	#next do the UU drops
	for poke in curTiers.keys():
		if poke not in usage:
			continue
		if curTiers[poke] == 'OU' and poke not in newTiers.keys():
			if usage[poke][0] < drop:
				newTiers[poke] = 'UU'
			else:
				newTiers[poke] = 'OU'

	#next do BL
	for poke in curTiers.keys():
		if poke not in usage:
			continue
		if curTiers[poke] == 'UUBL' and poke not in newTiers.keys():
			newTiers[poke] = 'UUBL'


	#next do the UU rises
	for poke in curTiers.keys():
		if poke not in usage:
			continue
		if usage[poke][1] > rise and poke not in newTiers.keys():
			newTiers[poke] = 'UU'

	#the rest go in the lowest tier
	for poke in curTiers.keys():
		if poke not in newTiers.keys():
			newTiers[poke] = 'RU'

	for poke in newTiers.keys():
		if newTiers[poke] == 'UU' and poke in \
			['blaziken', 'charizardmegax', 'cinderace', 'dondozo', 'gyarados',
			 'gyaradosmega', 'hoopaunbound', 'ironhands', 'kommoo', 'latios',
			 'latiosmega', 'manaphy', 'medichammega', 'meowscarada',
			 'ogerponcornerstone', 'pinsirmega', 'tornadustherian', 'weavile',
			 'xurkitree', 'zapdosgalar', 'annihilape', 'baxcalibur', 'ceruledge',
			 'espathra', 'garganacl', 'kartana', 'kyurem', 'mawilemega', 'zamazenta']:
			newTiers[poke] = 'UUBL'
		elif newTiers[poke] == 'RU' and poke in \
			['aerodactylmega', 'alakazam', 'altariamega', 'bisharp',
			 'blacephalon', 'buzzwole', 'dondozo', 'enamorus', 'gallademega',
			 'gardevoirmega', 'gengar', 'gyarados', 'hawlucha', 'heracrossmega',
			 'ironhands', 'ironleaves', 'jirachi', 'keldeo', 'keldeoresolute',
			 'kyurem', 'latiasmega', 'latios', 'latiosmega', 'lilliganthisui',
			 'mamoswine', 'manaphy', 'mew', 'moltresgalar', 'pecharunt', 'porygonz',
			 'quaquaval', 'salamence', 'sableyemega', 'terrakion', 'victini',
			 'xurkitree', 'zapdosgalar', 'zoroarkhisui']:
			newTiers[poke] = 'RUBL'

	print("")
	for poke in curTiers:
		if newTiers[poke] == 'RU' and poke in NFE:
			continue
		if curTiers[poke] != newTiers[poke]:
			species = keyLookup[poke]
			if species.endswith('-Mega') or species.endswith('-Mega-X') or species.endswith('-Mega-Y') or species.endswith('-Primal'):
				base = keyify(species[:species.index('-')]) #none of the megas have hyphenated names
				if tiers.index(newTiers[base]) < tiers.index(newTiers[poke]): #if the base is in a higher tier
					newTiers[poke] = newTiers[base]
					continue
			print(species+" moved from "+curTiers[poke]+" to "+newTiers[poke])

if __name__ == "__main__":
    main(sys.argv[1:])