tcg/wantedgen.py

import datetime,os
import log,variables,skel,tcgcore

def wantedgen():
    if not os.path.isdir("build/wanted"):
        os.mkdir("build/wanted")
    if os.path.exists("build/wanted/index.html"):
        os.remove("build/wanted/index.html")
    thefile = "build/wanted/index.html"
    skel.headerwrite(thefile,"wanted")
    content = open(thefile,"a")
    content.write("<h1>wanted cards</h1>\n")
    ownedcards = []
    decksofinterest = []
    for event in log.log:
        try:
            for card in event["received"]:
                ownedcards.append(card)
                decksofinterest.append(card[:-2])
        except:
            pass
    ownedcards = sorted(ownedcards)
    decksofinterest = sorted(list(dict.fromkeys(decksofinterest)))
    wantedcards = []
    for deck in decksofinterest:
        wantedlist = ["01","02","03","04","05","06","07","08","09","10","11","12","13","14","15","16","17","18","19","20"]
        for card in wantedlist:
            combined = deck + card
            if combined not in ownedcards:
                wantedcards.append(combined)
    highpriority = []
    medpriority = []
    lowpriority = []
    for card in wantedcards:
        if card[:-2] in variables.highpriority:
            highpriority.append(card)
        elif card[:-2] in variables.medpriority:
            medpriority.append(card)
        else:
            lowpriority.append(card)
    if len(highpriority) > 0:
        content.write("<p><b>High priority:</b> ")
        for card in highpriority:
            content.write(tcgcore.cardtext(card))
            if highpriority.index(card) == len(highpriority) - 1:
                content.write(" ")
            else:
                content.write(", ")
        content.write("</p>\n")
    if len(medpriority) > 0:
        content.write("<p><b>Medium priority:</b> ")
        for card in medpriority:
            content.write(tcgcore.cardtext(card))
            if medpriority.index(card) == len(medpriority) - 1:
                content.write(" ")
            else:
                content.write(", ")
        content.write("</p>\n")
    if len(lowpriority) > 0:
        content.write("<p><b>Low priority:</b> ")
        for card in lowpriority:
            content.write(tcgcore.cardtext(card))
            if lowpriority.index(card) == len(lowpriority) - 1:
                content.write(" ")
            else:
                content.write(", ")
        content.write("</p>\n")
    content.close()
    skel.footerwrite(thefile)

if __name__ == "__main__":
    wantedgen()