cleaning up code

This commit is contained in:
Zachary Watts
2026-05-03 22:46:33 -04:00
parent fbe69be2ac
commit d1829184cd
3 changed files with 13 additions and 14 deletions

View File

@@ -137,7 +137,7 @@ class Adventurer:
# return highest, found this at https://stackoverflow.com/a/280156
return max(prime_attributes, key=prime_attributes.get)
def set_equipment(self):
def set_equipment(self) -> list:
equipment = [ 'backpack', 'tinderbox', 'waterskin' ]
for i in range(2):
item = ""
@@ -147,7 +147,7 @@ class Adventurer:
equipment.append(item)
return equipment
def select_spells(self, spell_list) -> list:
def select_spells(self, spell_list: dict) -> list:
spell_book = []
for spell_level, count in self.spells.items():
if count != "-":
@@ -178,7 +178,7 @@ class Adventurer:
def get_class_progression_for_level(self) -> list:
return list(filter(lambda d: d['level'] == self.level, self.__class__.progression))[0]
def set_level(self, new_level) -> None:
def set_level(self, new_level: int) -> None:
self.level = new_level
self.hp = sum(self.hp_rolls[:self.level])
if self.player_class in [ "magic user", "elf" ]: