script.py

#!/usr/bin/env python3
# This is so stupid but it helped kill an hour
import json
import random
import textwrap
from pick import pick
import os


class SpellClass:
    bard = "Bard"
    cleric = "Cleric"
    druid = "Druid"
    paladin = "Paladin"
    ranger = "Ranger"
    sorcerer = "Sorcerer"
    warlock = "Warlock"
    wizard = "Wizard"


class Color:
    PURPLE = "\033[95m"
    CYAN = "\033[96m"
    DARKCYAN = "\033[36m"
    BLUE = "\033[94m"
    GREEN = "\033[92m"
    YELLOW = "\033[93m"
    RED = "\033[91m"
    BOLD = "\033[1m"
    UNDERLINE = "\033[4m"
    END = "\033[0m"


# Input: class name
# Process: Filter json by class (user input)
# Select random spell from new list
# Output: spell name
def choose_spell(class_name):
    # Load the data from the local spells.json file
    with open("spells.json") as f:
        in_list = True
        while in_list is True:
            data = json.load(f)
            # Filter the data by class
            filteredData = [spell for spell in data if class_name in spell["class"]]
            # Select random spell
            randomSpell = filteredData[random.randint(0, len(filteredData) - 1)]
            # Checks if in list of used spells
            in_list = check_spell_in_list(randomSpell["name"])
            if in_list:
                print("This has been used before")
                break
        # Splits spell description into a list
        chosenSpellList = randomSpell["desc"].split(".")
        # Takes first two elements (first two sentences) and adds them to string
        chosenSpellString = (chosenSpellList[0]) + ". " + chosenSpellList[1] + "."
        # Removing every instance of <p> and </p> from the spell description
        chosenSpellString = (
            chosenSpellString.replace("<p>", "").replace("</p>", "").lstrip()
        )
        return (
            Color.BOLD
            + "How about "
            + Color.GREEN
            + randomSpell["name"]
            + Color.END
            + Color.BOLD
            + "?\n"
            + Color.END
            + textwrap.fill(chosenSpellString, width=100),
            randomSpell["name"],
        )


# Print out list of all SpellClass variables
# Then take user input for class
# Return class name
def choose_class():
    options = [
        "Bard - An inspiring magician",
        "Cleric - A priestly champion",
        "Druid - A priest of the Old Faith",
        "Paladin - A holy warrior",
        "Ranger - A warrior",
        "Sorcerer - A spellcaster",
        "Warlock - A wielder of magic",
        "Wizard - A scholarly magic-user",
    ]
    title = "Choose a class: "
    option, index = pick(options, title)
    optionString = option.split(" ")[0]
    class_name = optionString
    print(*options, sep="\n")
    print("\n" + len(class_name) * "=" + "================")
    print(Color.BOLD + "You have chosen " + class_name + Color.END)
    print(len(class_name) * "=" + "================")

    # Return class name
    return class_name


# This adds a spell to a list of already used spells
def add_spell_to_list(chosen_spell):
    file = open("used_spells.txt", "a+")
    file_info = file.read().split(",")
    if chosen_spell not in file_info:
        file.write(chosen_spell + ",")
        file.close()


# Checks if spell is in list of used spells
def check_spell_in_list(spell_name):
    file = open("used_spells.txt", "r")
    file_info = file.read().split(",")
    file.close()
    if spell_name in file_info:
        return True
    else:
        return False


if __name__ == "__main__":
    # Title
    print()
    title = Color.BOLD + "SOARCERER'S SPELLBOOK" + Color.END
    sparkle = Color.GREEN + "+" + Color.END + Color.RED + "=" + Color.END
    print(sparkle * int(len(title) / 2))
    print((" " * int(len(title) / 8)) + title)
    print(sparkle * int(len(title) / 2))
    status = True
    print()
    os.system('afplay "sound_effects/soarcerers_spellbook.wav"')

    begin = input(
        Color.GREEN
        + "+=+= "
        + Color.BOLD
        + Color.RED
        + "HIT ENTER TO BEGIN"
        + Color.GREEN
        + " =+=+\n"
        + Color.END
    )
    # Main Loop
    while status is True:
        # Print random entry
        chosen_spell = choose_spell(choose_class())
        if "wind" in chosen_spell[0].lower():
            os.system('afplay "sound_effects/fart.wav"')
        print("\n" + Color.BOLD + chosen_spell[0] + Color.END + "\n")
        # Check if already used this spell before
        used_spell = input(
            Color.BOLD
            + "Has This Spell Been Cast Before?"
            + Color.END
            + " ["
            + Color.GREEN
            + "y"
            + Color.END
            + "/"
            + Color.RED
            + "n"
            + Color.END
            + "]"
            + "\n"
        ).lower()
        if used_spell == "y":
            add_spell_to_list(chosen_spell[1])
        # Continue loop or break
        v = input(
            Color.BOLD
            + "Choose Another?"
            + Color.END
            + " ["
            + Color.GREEN
            + "y"
            + Color.END
            + "/"
            + Color.RED
            + "n"
            + Color.END
            + "]"
            + "\n"
        ).lower()
        print()
        if v == "n":
            print(
                Color.GREEN
                + "+=+= "
                + Color.RED
                + Color.BOLD
                + "May the sprint begin!"
                + Color.GREEN
                + "+=+= \n"
                + Color.END
            )
            os.system('afplay "sound_effects/let_the_sprint_begin.wav"')
            break
        if v == "y":
            continue