#!/usr/bin/python3 import random armor= { 'None' : 9, 'Leather' : 7, 'Leather, Shield' : 6, 'Chain' : 5, 'Chain, Shield' : 4, 'Plate' : 3, 'Plate, Shield' : 2 } weapons = [ { 'name' : 'battle axe', 'type' : 'melee', 'damage-dice' : 8}, { 'name' : 'crossbow', 'type' : 'missle', 'damage-dice' : 6, 'ammo' : '20 bolts'}, { 'name' : 'hand axe', 'type' : 'melee', 'damage-dice' : 6 }, { 'name' : 'mace', 'type' : 'melee', 'damage-dice' : 6 , 'traits' : [ 'blunt' ] }, { 'name' : 'pole arm', 'type' : 'melee', 'damage-dice' : 10 }, { 'name' : 'short bow', 'type' : 'missle', 'damage-dice' : 6, 'ammo' : '20 arrows'}, { 'name' : 'short sword', 'type' : 'melee', 'damage-dice' : 6 }, { 'name' : 'silver dagger', 'type' : 'melee', 'damage-dice' : 4 }, { 'name' : 'sling', 'type' : 'missle', 'damage-dice' : 4, 'ammo' : '20 stones', 'traits' : [ 'blunt' ] }, { 'name' : 'staff', 'type' : 'melee', 'damage-dice' : 4 , 'traits' : [ 'blunt', 'slow' ] }, { 'name' : 'spear', 'type' : 'melee', 'damage-dice' : 6 }, { 'name' : 'sword', 'type' : 'melee', 'damage-dice' : 8 }, { 'name' : 'war hammer', 'type' : 'melee', 'damage-dice' : 6 , 'traits' : [ 'blunt' ] } ] # Player Character Classes class Adventurer: def __init__(self, level=1, attributes={}) -> None: # using a get() method to pull an attribute else use a default value, https://python-academy.org/en/handbook/get self.player_class = None self.level = level self.strength = attributes.get('strength', roll_dice(3,6)) self.intelligence = attributes.get('intelligence', roll_dice(3,6)) self.wisdom = attributes.get('wisdom', roll_dice(3,6)) self.dexterity = attributes.get('dexterity', roll_dice(3,6)) self.constitution = attributes.get('constitution', roll_dice(3,6)) self.charisma = attributes.get('charisma', roll_dice(3,6)) self.hp = 1 self.torches = roll_dice(1,6) self.rations = roll_dice(1,6) self.equipment = [ 'backpack', 'tinderbox', 'waterskin' ] self.weapons = [] self.armor = [] def __str__(self): return f"{self.player_class}" def character_sheet(self): sheet = [] sheet.append('{0: <26}'.format(f"| {self.player_class.title()} - Level {self.level}")) for key, val in self.get_attributes().items(): key_string = "| " + '{0:12}'.format(f"{key}").capitalize() + f" {val}" sheet += ['{0: <26}'.format(key_string)] sheet.append('| ----------------------- ') sheet.append('{0: <26}'.format(f"| HP: {self.hp} AC: {self.ac}")) sheet.append('{0: <26}'.format(f"| Torches: {self.torches}")) sheet.append('{0: <26}'.format(f"| Rations: {self.rations}")) sheet.append('{0: <26}'.format(f"| Armor: {self.armor}")) sheet.append('{0: <26}'.format(f"| Weapons:")) sheet.append('{0: <26}'.format(f"| {self.weapons[0]}")) sheet.append('{0: <26}'.format(f"| {self.weapons[1]}")) return sheet def get_attributes(self): attribute_list = ['strength', 'intelligence', 'wisdom', 'dexterity', 'constitution', 'charisma'] return {k: self.__dict__[k] for k in attribute_list} def get_best_prime_attribute(self): attribute_list = [ 'strength', 'intelligence', 'wisdom', 'dexterity' ] prime_attributes = {k: self.__dict__[k] for k in attribute_list} # return highest, found this at https://stackoverflow.com/a/280156 return max(prime_attributes, key=prime_attributes.get) def roll_equipment(self): print("testing!") class Fighter(Adventurer): prime_requisite = "strength" requirements = None def __init__(self, level, attributes={}) -> None: Adventurer.__init__(self, level, attributes) self.player_class = "fighter" self.hp = roll_dice(self.level, 8) self.armor = random.choice(list(armor.keys())) self.ac = armor[self.armor] self.weapons = [ "sword", "dagger" ] class MagicUser(Adventurer): prime_requisite = "intelligence" requirements = None def __init__(self, level, attributes={}) -> None: Adventurer.__init__(self, level, attributes) self.player_class = "magic user" self.hp = roll_dice(self.level, 4) self.armor = random.choice(list(armor.keys())) self.ac = armor[self.armor] self.weapons = [ "sword", "dagger" ] class Cleric(Adventurer): prime_requisite = "wisdom" requirements = None def __init__(self, level, attributes={}) -> None: Adventurer.__init__(self, level, attributes) self.player_class = "cleric" self.hp = roll_dice(self.level, 6) self.armor = random.choice(list(armor.keys())) self.ac = armor[self.armor] self.weapons = [ "sword", "dagger" ] class Thief(Adventurer): prime_requisite = "dexterity" requirements = None def __init__(self, level, attributes={}) -> None: Adventurer.__init__(self, level, attributes) self.player_class = "thief" self.hp = roll_dice(self.level, 4) self.armor = random.choice(list(armor.keys())) self.ac = armor[self.armor] self.weapons = [ "sword", "dagger" ] class Dwarf(Adventurer): prime_requisite = "strength" requirements = {'constitution' : 9 } def __init__(self, level, attributes={}) -> None: Adventurer.__init__(self, level, attributes) self.player_class = "dwarf" self.hp = roll_dice(self.level, 8) self.armor = random.choice(list(armor.keys())) self.ac = armor[self.armor] self.weapons = [ "sword", "dagger" ] class Elf(Adventurer): prime_requisite = "intellgence" requirements = {'intelligence' : 9 } def __init__(self, level, attributes={}) -> None: Adventurer.__init__(self, level, attributes) self.player_class = "elf" self.hp = roll_dice(self.level, 6) self.armor = random.choice(list(armor.keys())) self.ac = armor[self.armor] self.weapons = [ "sword", "dagger" ] class Halfling(Adventurer): prime_requisite = "dexterity" requirements = {'constitution' : 9, 'dexterity' : 9 } def __init__(self, level, attributes={}) -> None: Adventurer.__init__(self, level, attributes) self.player_class = "halfling" self.hp = roll_dice(self.level, 6) self.armor = random.choice(list(armor.keys())) self.ac = armor[self.armor] self.weapons = [ "sword", "dagger" ] # Player Class Selector class ClassSelector(): def __init__(self, player: Adventurer) -> None: self.player = player # https://stackoverflow.com/questions/3862310/how-to-find-all-the-subclasses-of-a-class-given-its-name # pull classes that do not have requirements self.available_classes = [cls for cls in Adventurer.__subclasses__() if not cls.requirements] # pull classes that do have requirements self.classes_with_reqs = [cls for cls in Adventurer.__subclasses__() if cls.requirements] # run function to randomly select an adventurer class self.selected_class = self.selection() def return_class_by_best_attribute(self): # for adventurer classes in available classes, return the one where that classes' prime requisite is equal to the players best attribute return [adv_class for adv_class in self.available_classes if adv_class.prime_requisite == self.player.get_best_prime_attribute()][0] def return_classes_with_requirements(self): p_attrs = self.player.get_attributes() possible_classes = [] for c in self.classes_with_reqs: match_count = 0 for r in c.requirements: if p_attrs[r] >= c.requirements[r]: match_count += 1 if match_count >= len(c.requirements): possible_classes.append(c) return possible_classes def selection(self): best_prime_attribute = self.player.get_best_prime_attribute() # create an array of possible player classes, add the best choice per player's best core attributes possible_classes = [ self.return_class_by_best_attribute() ] possible_classes += self.return_classes_with_requirements() # randomly select class selected_class = random.choice(possible_classes) return selected_class class PartyGenerator(): def __init__(self, party_size=4) -> None: self.size = party_size self.adventurers = [] self.adventurer_types = [] def gen_party(self): while len(self.adventurers) < self.size: new_player = Adventurer() attempts = 0 while new_player.player_class not in self.adventurer_types: attempts += 1 selected_class = ClassSelector(new_player).selection() new_player = selected_class(new_player.level, new_player.get_attributes()) # i couldnt randomly generate a scenario where a character couldn't be added, but it seems possible, so this is the hard cut off if (new_player.player_class not in self.adventurer_types) or (attempts > 10): self.adventurers.append(new_player) self.adventurer_types.append(new_player.player_class) def get_character_sheets(self): for i in range(15): for j in range(len(self.adventurers)): adv = self.adventurers[j] print(adv.character_sheet()[i],end='') print('|\n',end='') def __str__(self): return f"{self.adventurers}" # functions def roll_dice(count, sides): return sum(random.randint(1,sides) for _ in range(count)) def main(): new_party = PartyGenerator() new_party.gen_party() new_party.get_character_sheets() main()