tagmon.py

import random
import os
import shutil

from anki.lang import _
from aqt import *
from aqt.qt import *
from aqt import mw

from .utils import *
from .compute import load_pokemon_gen_all, alertMsgText
from .stats import TagStats
from .display import pokemonDisplayText


def tagmonDisplay():
    # Assign Pokemon Image and Progress Bar folder directory names
    pkmnimgfolder = currentdirname / "pokemon_images"
    progressbarfolder = currentdirname / "progress_bars"
    # Move Pokemon Image folder to collection.media folder if not already there (Anki reads from here when running anki.stats.py)
    if os.path.exists("%s/pokemon_images" % mediafolder) == False and os.path.exists(pkmnimgfolder):
        shutil.copytree(pkmnimgfolder, "%s/pokemon_images" % mediafolder)
    if os.path.exists("%s/progress_bars" % mediafolder) == False and os.path.exists(progressbarfolder):
        shutil.copytree(progressbarfolder, "%s/progress_bars" % mediafolder)
    tagmon = TagPokemon()
    result = ""
    if tagmon:
        result = _show(tagmon, "Pokémon", "Your Pokémon")
    return result


def _show(data, title, subtitle):
    # Set text equal to title text to start
    txt = "<h1>{}</h1>{}".format(title, subtitle)
    # Return empty if no data
    if not data:
        return txt
    # Line text variable, apparently needed for bottom line
    text_lines = []
    # Table text
    table_text = ""
    _num_pokemon = len(data)
    pokemon_names = []
    pokemon_tags = []
    pokemon_levels = []
    pokemon_nicknames = []
    sorteddata = sorted(data, key=lambda k: k[2], reverse=True)
    for thing in sorteddata:
        if len(thing) == 4:
            pokemon_nicknames.append(thing[3])
        else:
            pokemon_nicknames.append(None)
        pokemon_names.append(thing[0])
        pokemon_tags.append(thing[1])
        pokemon_levels.append(str(thing[2]))
    pokemon_collection = tuple(
        zip(pokemon_names, pokemon_tags, pokemon_levels, pokemon_nicknames))
    pokemon_progress = []
    for level in pokemon_levels:
        if float(level) < 5:
            pokemon_progress.append(None)
        else:
            pokemon_progress.append(
                int(float(20*(float(level) - int(float(level))))))
        pokemon_progress_text = []
        for item in pokemon_progress:
            if item is not None:
                pokemon_progress_text.append(
                    """<img src="/progress_bars/%s.png">""" % (item))
            else:
                pokemon_progress_text.append("")
    pokemon_text = []
    table_size = 0
    for name, tag, level, nickname in pokemon_collection:
        (text, held, special) = pokemonDisplayText(name, tag, level, nickname)
        pokemon_text.append(text)

    def get(array, index):
        if index < len(array):
            return array[index]
        else:
            return ""

    while table_size < len(pokemon_text):
        table_text += "\n".join((
            "<tr>",
            table_image_html(get(pokemon_names, table_size)),
            table_image_html(get(pokemon_names, table_size + 1)),
            table_image_html(get(pokemon_names, table_size + 2)),
            "</tr>"
        ))
        table_text += "\n".join((
            "<tr>",
            table_text_html(get(pokemon_text, table_size), "", True),
            table_text_html(get(pokemon_text, table_size + 1), "", True),
            table_text_html(get(pokemon_text, table_size + 2), "", True),
            "</tr>"
        ))
        table_text += "\n".join((
            "<tr>",
            table_text_html(get(pokemon_progress_text, table_size)),
            table_text_html(get(pokemon_progress_text, table_size + 1)),
            table_text_html(get(pokemon_progress_text, table_size + 2)),
            "</tr>"
        ))
        table_size += 3

        # Assign table_text to txt
    txt += "<table width = 750>" + table_text + "</table>"
    # Make bottom line using function from stats.py and assign to text_lines
    line(
        text_lines,
        "<b>Total</b>",
        "</b>%s Pokémon<b>" % _num_pokemon)
    return txt


def TagPokemon():
    tagmonlist = get_json('_tagmon.json', [])
    savedtags = get_json("_tags.json", [])
    modifiedtagmon = []
    for item in reversed(tagmonlist):
        if item[1] in savedtags:
            modifiedtagmon.append(item)

    thresholdsettings = get_json("_tagmonsettings.json", [
                                 50, 125, 250, 375, 500])

    pokemonlist = []
    tiers = []
    evolutionLevel1 = []
    evolution1 = []
    evolutionLevel2 = []
    evolution2 = []
    load_pokemon_gen_all(pokemonlist, tiers, evolutionLevel1,
                         evolution1, evolutionLevel2, evolution2)

    pokemon_tuple = tuple(
        zip(pokemonlist, tiers, evolutionLevel1, evolution1, evolutionLevel2, evolution2))
    tierdict = {"A": [], "B": [], "C": [], "D": [], "E": [], "F": []}

    for pokemon, tier, firstEL, firstEvol, secondEL, secondEvol in pokemon_tuple:
        if tier != "S":
            tierdict[tier].append(pokemon)

    # Have default message text here because of future for loop iterating through multideck results
    msgtxt = "Hello Pokémon Trainer!"
    # Assign empty list to multiData (to be returned at end)
    multiData = []
    results = TagStats()
    if len(results) == 0:
        return
    prestigelist = get_json("_prestigelist.json", [])
    for item in results:
        result = item[1]
        sumivl = 0
        for id, ivl in result:
            adjustedivl = (100 * (ivl/100)**(0.5))
            sumivl += adjustedivl
        if len(result) == 0:
            continue
        Level = round((sumivl/len(result)+0.5), 2)

        pokemontier = ""

        if len(result) < thresholdsettings[0]:
            pokemontier = "F"
        elif len(result) < thresholdsettings[1]:
            pokemontier = "E"
        elif len(result) < thresholdsettings[2]:
            pokemontier = "D"
        elif len(result) < thresholdsettings[3]:
            pokemontier = "C"
        elif len(result) < thresholdsettings[4]:
            pokemontier = "B"
        else:
            pokemontier = "A"
        tagmon = ""
        already_assigned = False
        details = ()
        nickname = ""
        previouslevel = 0
        for thing in modifiedtagmon:
            if thing[1] == item[0]:
                tagmon = thing[0]
                already_assigned = True
                previouslevel = thing[2]
                if len(thing) == 4:
                    nickname = thing[3]
                else:
                    nickname = ""
        if tagmon == "":
            if pokemontier == "A":
                msgbox = QMessageBox()
                msgbox.setWindowTitle("Pokemanki")
                msgbox.setText(
                    "Choose a starter Pokémon for the %s tag" % item[0])
                msgbox.addButton("Bulbasaur", QMessageBox.AcceptRole)
                msgbox.addButton("Charmander", QMessageBox.AcceptRole)
                msgbox.addButton("Squirtle", QMessageBox.AcceptRole)
                msgbox.exec_()
                tagmon = msgbox.clickedButton().text()
            else:
                tiernumber = len(tierdict[pokemontier])
                tierlabel = tierdict[pokemontier]
                randno = random.randint(0, tiernumber - 1)
                tagmon = tierlabel[randno]
        details = ()
        for pokemon, tier, firstEL, firstEvol, secondEL, secondEvol in pokemon_tuple:
            if tagmon == pokemon or tagmon == firstEvol or tagmon == secondEvol:
                details = (pokemon, firstEL, firstEvol, secondEL, secondEvol)

        name = details[0]
        firstEL = details[1]
        firstEvol = details[2]
        secondEL = details[3]
        secondEvol = details[4]

        if Level > 100 and item[0] not in prestigelist:
            Level = 100
        if item[0] in prestigelist:
            Level -= 50
        if Level < 5:
            name = "Egg"
        if firstEL is not None:
            firstEL = int(firstEL)
            if Level > firstEL:
                name = firstEvol
        if secondEL is not None:
            secondEL = int(secondEL)
            if Level > secondEL:
                name = secondEvol
        if name.startswith("Eevee"):
            name = "Eevee"
        if already_assigned == False and name != "Eevee" and name != "Egg":
            tagmonData = (name, item[0], Level)
            modifiedtagmon.append(tagmonData)
        elif already_assigned == False:
            tagmonData = (tagmon, item[0], Level)
            modifiedtagmon.append(tagmonData)
        msgtxt += alertMsgText(tagmon, item[0], name, int(Level),
                               int(previouslevel), nickname, already_assigned)
        if already_assigned == True:
            for thing in tagmonlist:
                if thing[1] == item[0]:
                    if name == "Eevee" or name == "Egg":
                        if nickname:
                            if (thing[0], thing[1], Level, nickname) in modifiedtagmon:
                                pass
                            else:
                                modifiedtagmon.append(
                                    (thing[0], thing[1], Level, nickname))
                        else:
                            if (thing[0], thing[1], Level) in modifiedtagmon:
                                pass
                            else:
                                modifiedtagmon.append(
                                    (thing[0], thing[1], Level))
                    else:
                        if nickname:
                            if (name, thing[1], Level, nickname) in modifiedtagmon:
                                pass
                            else:
                                modifiedtagmon.append(
                                    (name, thing[1], Level, nickname))
                        else:
                            if (name, thing[1], Level) in modifiedtagmon:
                                pass
                            else:
                                modifiedtagmon.append((name, thing[1], Level))
        displayData = (name, item[0], Level)
        multiData.append(displayData)

    # After iterating through each deck, store data into pokemanki.json
    write_json("_tagmon.json", modifiedtagmon)
    # Only display message if changes
    if msgtxt != "Hello Pokémon Trainer!":
        msgbox2 = QMessageBox()
        msgbox2.setWindowTitle("Pokemanki")
        msgbox2.setText(msgtxt)
        msgbox2.exec_()
    # Return multiData
    return multiData