tre / releaseprobability.py

from bs4 import BeautifulSoup import wantlist ''' Requires key.html as generated by the usual script. Download https://colors-tcg.eu/releasechecker.php, strip it to just the divs with class "list" and save as release.html. Save a file called wantlist.py with thelist = current wantlist in list format. This is based on release 192 being the only one not listed. Change nextrelease to the number of the next release. ''' nextrelease = 194 values = ["red","orange","yellow","green","blue","purple","brown","gray","special","limited"] deckkey = {} decknames = [] with open("key.html") as decks: decksoup = BeautifulSoup(decks,"html.parser") soupdecks = decksoup.find("table",{"id":"colors"}) souprows = soupdecks.find_all("tr") souprows = souprows[1:] for row in souprows: soupcells = row.find_all("td") deckname = soupcells[2].text.lower().replace(" ","").replace("'","").replace("ç","c").replace(".","").replace("politetness","politeness").replace("á","a") decktype = soupcells[3].text.lower() deckdict = {} if decktype in values: deckdict["type"] = decktype deckkey[deckname] = deckdict decknames.append(deckname) releasedict = {} with open("release.html") as releases: releasesoup = BeautifulSoup(releases,"html.parser") allreleases = releasesoup.find_all("div") for release in allreleases: releaseid = int(release.find("strong").text[-4:-1]) releasedecks = release.find("textarea").text.split(", ") releasedict[releaseid] = releasedecks for deck in decknames: deckkey[deck]["release"] = False releaseno = 1 while releaseno < nextrelease: if releaseno != 192: if deck in releasedict[releaseno]: deckkey[deck]["release"] = releaseno releaseno += 1 if not deckkey[deck]["release"]: deckkey[deck]["release"] = 192 releasecolours = {} therelease = 1 while therelease < nextrelease: red = 0 orange = 0 yellow = 0 green = 0 blue = 0 purple = 0 brown = 0 grey = 0 special = 0 for deck in decknames: if deckkey[deck]["release"] == therelease: if deckkey[deck]["type"] == "red": red += 20 elif deckkey[deck]["type"] == "orange": orange += 20 elif deckkey[deck]["type"] == "yellow": yellow += 20 elif deckkey[deck]["type"] == "green": green += 20 elif deckkey[deck]["type"] == "blue": blue += 20 elif deckkey[deck]["type"] == "purple": purple += 20 elif deckkey[deck]["type"] == "brown": brown += 20 elif deckkey[deck]["type"] == "gray": grey += 20 elif deckkey[deck]["type"] == "special": special += 20 releasecolours[therelease] = {"wantred":0,"hasred":red,"wantorange":0,"hasorange":orange,"wantyellow":0,"hasyellow":yellow,"wantgreen":0,"hasgreen":green,"wantblue":0,"hasblue":blue,"wantpurple":0,"haspurple":purple,"wantbrown":0,"hasbrown":brown,"wantgrey":0,"hasgrey":grey} therelease += 1 wantlist = wantlist.thelist for card in wantlist: deck = card[:-2] deckrelease = deckkey[deck]["release"] decktype = deckkey[deck]["type"] if decktype == "red": releasecolours[deckrelease]["wantred"] += 1 elif decktype == "orange": releasecolours[deckrelease]["wantorange"] += 1 elif decktype == "yellow": releasecolours[deckrelease]["wantyellow"] += 1 elif decktype == "green": releasecolours[deckrelease]["wantgreen"] += 1 elif decktype == "blue": releasecolours[deckrelease]["wantblue"] += 1 elif decktype == "purple": releasecolours[deckrelease]["wantpurple"] += 1 elif decktype == "brown": releasecolours[deckrelease]["wantbrown"] += 1 elif decktype == "gray": releasecolours[deckrelease]["wantgrey"] += 1 foundrelease = False foundcolour = False foundproportion = 0 releaseno = 1 while releaseno < nextrelease: if releasecolours[releaseno]["wantred"] > 0: if releasecolours[releaseno]["wantred"]/releasecolours[releaseno]["hasred"] > foundproportion: foundrelease = releaseno foundcolour = "red" foundproportion = releasecolours[releaseno]["wantred"]/releasecolours[releaseno]["hasred"] if releasecolours[releaseno]["wantorange"] > 0: if releasecolours[releaseno]["wantorange"]/releasecolours[releaseno]["hasorange"] > foundproportion: foundrelease = releaseno foundcolour = "orange" foundproportion = releasecolours[releaseno]["wantorange"]/releasecolours[releaseno]["hasorange"] if releasecolours[releaseno]["wantyellow"] > 0: if releasecolours[releaseno]["wantyellow"]/releasecolours[releaseno]["hasyellow"] > foundproportion: foundrelease = releaseno foundcolour = "yellow" foundproportion = releasecolours[releaseno]["wantyellow"]/releasecolours[releaseno]["hasyellow"] if releasecolours[releaseno]["wantgreen"] > 0: if releasecolours[releaseno]["wantgreen"]/releasecolours[releaseno]["hasgreen"] > foundproportion: foundrelease = releaseno foundcolour = "green" foundproportion = releasecolours[releaseno]["wantgreen"]/releasecolours[releaseno]["hasgreen"] if releasecolours[releaseno]["wantblue"] > 0: if releasecolours[releaseno]["wantblue"]/releasecolours[releaseno]["hasblue"] > foundproportion: foundrelease = releaseno foundcolour = "blue" foundproportion = releasecolours[releaseno]["wantblue"]/releasecolours[releaseno]["hasblue"] if releasecolours[releaseno]["wantpurple"] > 0: if releasecolours[releaseno]["wantpurple"]/releasecolours[releaseno]["haspurple"] > foundproportion: foundrelease = releaseno foundcolour = "purple" foundproportion = releasecolours[releaseno]["wantpurple"]/releasecolours[releaseno]["haspurple"] if releasecolours[releaseno]["wantbrown"] > 0: if releasecolours[releaseno]["wantbrown"]/releasecolours[releaseno]["hasbrown"] > foundproportion: foundrelease = releaseno foundcolour = "brown" foundproportion = releasecolours[releaseno]["wantbrown"]/releasecolours[releaseno]["hasbrown"] if releasecolours[releaseno]["wantgrey"] > 0: if releasecolours[releaseno]["wantgrey"]/releasecolours[releaseno]["hasgrey"] > foundproportion: foundrelease = releaseno foundcolour = "grey" foundproportion = releasecolours[releaseno]["wantgrey"]/releasecolours[releaseno]["hasgrey"] releaseno += 1 print("Release " + str(foundrelease) + ", " + foundcolour + ": " + str(100 * foundproportion) + "%")