character_dnd.py

from dice import *

# Test

# Class to represent a character.
class Character_DnD:
    def __init__(self, name="nobody", species="unknown", job="nobody", level=1, weapon="none", weapon_value=0,
                 armor="none"):
        self.name = name
        self.species = species
        self.job = job
        self.level = level
        self.stats = {"strength": 8, "dexterity": 8, "constitution": 8, "intelligence": 8, "wisdom": 8, "charisma": 8}
        self.is_dead = False
        self.max_hit_points = 0
        self.hit_point_bonus = 0
        self.hit_points = 0
        self.armor = armor
        self.armor_bonus = 0
        self.armor_class = 0
        self.weapon = weapon
        self.weapon_value = weapon_value
        self.attribute_points = 26
        self.attack_bonus = 0
        self.damage_bonus = 0
        self.spell_save_dc = 8
        self.proficiency_modifier = 2
        self.log = {}
        self.gold = 0
        self.inventory = {}

    # Report back character name, species, job, stats, and equipment.
    def __repr__(self):
        return f"\n{self.name} is a level {self.level} {self.species} {self.job} with these stats:" \
               f"\n{self.stats['strength']} Strength \n{self.stats['dexterity']} Dexterity " \
               f"\n{self.stats['constitution']} Constitution \n{self.stats['intelligence']} Intelligence " \
               f"\n{self.stats['wisdom']} Wisdom \n{self.stats['charisma']} Charisma \nThey attack" \
               f" using {self.weapon}  and have an armor class of {self.armor_class}"

    # Report back character is dead.
    def is_dead(self):
        self.is_dead = True
        if self.hit_points <= 0:
            print(f"{self.name} is at zero hit points and is dead.")

    # Display character's stats.
    def stat_display(self):
        print(f"Your stats are as follows:\nStrength: {self.stats['strength']}\nDexterity: {self.stats['dexterity']}"
              f"\nConstitution: {self.stats['constitution']}\nIntelligence: {self.stats['intelligence']}"
              f"\nWisdom: {self.stats['wisdom']} \nCharisma: {self.stats['charisma']}")

    # Update stats based on species selection
    def species_change(self, species):
        if species.lower() == "human":
            self.stats["strength"] += 1
            self.stats["dexterity"] += 1
            self.stats["constitution"] += 1
            self.stats["intelligence"] += 1
            self.stats["wisdom"] += 1
            self.stats["charisma"] += 1
        elif species.lower() == "elf":
            self.stats["dexterity"] += 2
        else:
            self.stats["dexterity"] += 2

    # Character creation input.
    def starting_character_information(self):
        character_name = input(
            "Please type what you would like your character name to be? ")
        self.name = character_name.title()
        character_species = input(
            "Please type what you would like your character species to be? Please choose from Human, "
            "Elf, or Dwarf. Then hit enter. ")
        while character_species.lower() not in ["human", "elf", "dwarf"]:
            character_species = input(
                "Whoops, looks like you didn't choose one of the previous species. Please choose from "
                "Human, Elf, or Dwarf. Then hit enter. ")
        character_job = input(
            "Please type what you would like your character's job to be? Please choose from Fighter or a Rogue."
            " Then hit enter. ")
        while character_job.lower() not in ["fighter", "rogue"]:
            character_job = input(
                "Whoops, looks like you didn't choose one of the previous jobs. PLease choose from Fighter, "
                "Rogue. Then hit enter. ")
        print(f"\nYou have {self.attribute_points} points remaining. and the following stats:\n\nStrength: "
              f"{self.stats['strength']}\nDexterity: {self.stats['dexterity']}\nConstitution: "
              f"{self.stats['constitution']}\nIntelligence: {self.stats['intelligence']}\nWisdom: "
              f"{self.stats['wisdom']}\nCharisma: {self.stats['charisma']}\n")
        while self.attribute_points > 0:
            self.change_attribute_points()
        self.job = character_job.title()
        self.species_change(character_species)
        self.hit_point_calculation()

    def hit_point_calculation(self):
        self.max_hit_points = self.stats["constitution"] * 2 + self.hit_point_bonus


    # Change attribute points based on what stat was increased in stat_increase method.
    def change_attribute_points(self):
        which_stat = input("Which stat would you like to increase? Please select from strength, dexterity, "
                           "constitution, intelligence, wisdom, and Charisma. ")
        if which_stat.lower() not in ["strength", "dexterity", "constitution", "intelligence", "wisdom",
                                      "charisma"]:
            print(f"\nWhoops, '{which_stat}' wasn't an option. Please select from Strength, Dexterity, "
                  f"Constitution, Intelligence,Wisdom, or Charisma.")
        else:
            self.stat_cost(which_stat.lower())

    # Increase a character's base stats.
    def stat_increase(self, stat, amount):
        stats_list = ['strength', 'dexterity', 'constitution', 'intelligence', 'wisdom', 'charisma']
        if stat in stats_list:
            self.stats[stat] += amount
            print(f"\nYour {stat} is now {self.stats[stat]}")
            print(f"\nYou have {self.attribute_points} points remaining. and the following stats:\n\nStrength: "
                  f"{self.stats['strength']}\nDexterity: {self.stats['dexterity']}\nConstitution: "
                  f"{self.stats['constitution']}\nIntelligence: {self.stats['intelligence']}\nWisdom: "
                  f"{self.stats['wisdom']}\nCharisma: {self.stats['charisma']}\n")
        else:
            print("Your code isn't working.")

    # Determine cost of stat increase and ensure it can be increased as requested.
    def stat_cost(self, stat):
        print(f"Attribute Points: {self.attribute_points}")
        print(f"\nHow many points would you like to increase {stat} by? ")
        increase = input("Selection: ")
        cost = 0
        for number in range(self.stats[stat], self.stats[stat] + int(increase)):
            if number <= 13:
                cost += 1
            elif number <= 15:
                cost += 2
            else:
                print(f"\nThat would increase {stat} beyond 15. Please select another stat.")
        if cost > self.attribute_points:
            print(f"\nThis increase requires {cost} attribute points and you only have {self.attribute_points}. Please"
                  f"select another stat.")
        else:
            print(f"\nIt will cost {cost} attribute points to increase {stat} from {self.stats[stat]}"
                  f" to {self.stats[stat] + int(increase)}. Do you want to apply this change Y/N?")
            choice = input("Selection: ")
            if choice.lower() == "y":
                self.attribute_points -= cost
                self.stat_increase(stat, int(increase))
            else:
                print("No worries, let's try again.")



    # User selects from two weapon / magic options based on job.
    def weapon_choice(self):
        if self.job.lower() == "fighter":
            weapon_choice = input(
                "\nWould you like to use a Greatsword (1) for 1d12 damage or a Flail and Shield (2) for "
                " 1d10 damage and +2 Armor Class? Please input 1 or 2. ")
            print(weapon_choice)
            if weapon_choice == "1":
                self.weapon = "a Greatsword"
                self.weapon_value = 12
                self.attack_bonus = 5
            else:
                self.weapon = "a Flail and Shield"
                self.weapon_value = 8
                self.attack_bonus = 5
                self.armor_bonus += 2
                self.armor_class = self.armor_class = 10 + self.armor_bonus + ((self.stats["dexterity"]-10) / 2)
        elif self.job.lower() == "rogue":
            weapon_choice = input(
                "\nWould you like to use a Shortsword (1) for 1d6 damage or a Dagger (2) for "
                " 1d4 damage? Please input 1 or 2. ")
            if weapon_choice == "1":
                self.weapon = "a Shortsword"
                self.weapon_value = 6
            else:
                self.weapon = "a Dagger"
                self.weapon_value = 4

    # Lets the user select different armors, except the wizard. They just get Mage Armor.
    def armor_choice(self):
        if self.job.lower() == "fighter":
            armor_choice = input("\nWould you like Scale Mail (1) for 14 armor class plus your dexterity modifier "
                                 "(maximum +2) or Studded Leather (2) for 12 armor class plus your dexterity modifier?"
                                 "Please input 1 or 2. ")
            if armor_choice == "1":
                self.armor = "Scale Mail"
                self.armor_bonus += 8
                self.armor_class = 10 + self.armor_bonus
            else:
                self.armor = "Studded Leather"
                self.armor_bonus += 3
                self.armor_class = 10 + self.armor_bonus + ((self.stats["dexterity"]-10) / 2)
        elif self.job.lower() == "rogue":
            armor_choice = input("\nWould you like Leather Armor (1) for 11 armor class plus your dexterity modifier or"
                                 " Studded Leather (2) for 12 armor class and no maximum dexterity modifier? Please "
                                 "input 1 or 2. ")
            if armor_choice == "1":
                self.armor = "Leather Armor"
                self.armor_bonus += 5
                self.armor_class = 10 + self.armor_bonus + ((self.stats["dexterity"]-10) / 2)
            else:
                self.armor = "Studded Leather"
                self.armor_bonus += 4
                self.armor_class = 10 + self.armor_bonus + ((self.stats["dexterity"]-10) / 2)

    def player_loadout(self):
        print(f"{self.name} is wearing {self.armor} with {self.armor_class} armor class and wielding "
              f"{self.weapon}.")


    # Player attacks a monster.
    def player_attack(self, monster):
        attack_roll = roll(1, 20, self.attack_bonus)
        print(f"\n{self.name} attempts to attack the {monster.name} with {self.weapon}.")
        print(f"\n{self.name} rolls an attack roll of {attack_roll.results} compared to an AC of "
              f"{monster.armor_class}. ")
        if monster.armor_class < attack_roll.results:
            attack_damage = roll(1, self.weapon_value, self.damage_bonus)
            print(f"\n{self.name} has landed a hit for {attack_damage.results} points of damage.")
            monster.lose_hit_points(attack_damage.results)
        else:
            print(f"\n{monster.name} has successfully evaded the attack.")

    def lose_hit_points(self, amount):
        self.hit_points -= amount
        print(f"\n{self.name} has {self.hit_points} hit points remaining.")
        if self.hit_points <= 0:
            self.hit_points = 0
            self.is_dead = True
            print(f"\n{self.name} has died. Better luck next time.")
        else:
            print(f"\n{self.name} takes {amount} damage and has {self.hit_points} hit points remaining.")

    def attack_or_run(self, monster):
        choice = input("Do you wish to attack (1) or run (2)?")
        if choice == "1":
            self.player_attack(monster)
        elif choice == "2":
            pass # Need to add logic for running away and returning to town.
        else:
            print("You did not select attack (1) or run (2). ")
            self.attack_or_run(monster)

# Accepts input from a user and returns the choice of a specific attribute, to be passed as an argument for other
# methods.
def stat_choice():
    stats = {"1": "strength", "2": "dexterity", "3": "constitution", "4": "intelligence", "5": "wisdom",
             "6": "charisma"}
    print("\nWhich stat would you like to choose. Please select a number")
    print()
    for key, value in stats.items():
        print(key, value)
    print()
    choice = input("Selection: ")
    while choice not in stats.keys():
        choice = input("\nPlease select a valid number.")
    return choice