python_scripts/script.py at main · TKuzas-R7/python_scripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
#!/usr/bin/env python3
# This is so stupid, but it helped kill an hour
import json
import random
import textwrap


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:
        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)]
        # Check if Spell has been used
        with open('used_spells.json') as file:
            used_spells = json.load(file)
            dif_spell = False
            while not dif_spell:
                for i in range(len(used_spells)):
                    if used_spells[i]['name'] == randomSpell['name']:
                        randomSpell = filteredData[random.randint(0, len(filteredData) - 1)]
                    else:
                        dif_spell = True
        # Print random spell
        return Color.GREEN + randomSpell['name'] + Color.END + "\n" + textwrap.fill(
            textwrap.shorten(randomSpell['desc'], width=300), width=100)


# Print out list of all SpellClass variables
# Then take user input for class
# Return class name
def choose_class():
    print("+ " + "Bard - An inspiring magician")
    print("+ " + "Cleric - A priestly champion")
    print("+ " + "Druid - A priest of the Old Faith")
    print("+ " + "Paladin - A holy warrior")
    print("+ " + "Ranger - A warrior")
    print("+ " + "Sorcerer - A spellcaster")
    print("+ " + "Warlock - A wielder of magic")
    print("+ " + "Wizard - A scholarly magic-user")
    print()
    class_name = input(Color.BOLD + "Choose a class:" + Color.END + " ").capitalize()
    # Error handling
    while class_name not in SpellClass.__dict__.values():
        class_name = input(Color.BOLD + "Choose a class:" + Color.END + " ")

    # Return class name
    return class_name


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))
    print()
    status = True

    # Main Loop
    while status is True:
        # Print random entry
        chosen_spell = choose_spell(choose_class())
        print('\n' + Color.BOLD + chosen_spell + Color.END + '\n')
        # Ask if spell should be added to used spells list
        add = input('\n' + "Would you like to add this spell to the Used Spells List? " + " [" + Color.GREEN + 'y' + Color.END + '/' + Color.RED + 'n' + Color.END + ']' + "\n" ).lower()
        if add == 'y':
            with open('used_spells.json') as spells:
                used_spells_list = json.load(spells)
                chosen_spell = chosen_spell.partition('\n')[0]
                chosen_spell = chosen_spell.replace("\x1b[0m", "")
                chosen_spell = chosen_spell.replace("\x1b[92m", "")
                used_spells_list.append({'name': chosen_spell})
                with open('used_spells.json', "w+") as used_spell:
                    json.dump(used_spells_list, used_spell)
            print("Spell Added to Used Spells List")
        # 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.RED + Color.BOLD + "May the sprint begin!" + Color.END)
            break
        if v == 'y':
            continue