diff --git a/player.py b/player.py index 0da3156..6bdf065 100644 --- a/player.py +++ b/player.py @@ -7,14 +7,17 @@ class CreationError(Exception): class Player(object): """A hockey player with attributes and various functions.""" - def __init__(self, name:str): + def __init__(self, name:str, number:int=0): if name is None: self.attributes = [] elif len(name) > 30: raise CreationError("Player name too long.") + elif number < 0 or number > 99: + raise CreationError("Player number not valid.") else: self.name = name self.attributes = self.loadAttributes() + self.number = number def loadAttributes(self): """Generates attributes based on name, or loads from database if present.""" @@ -53,13 +56,24 @@ class Player(object): else: return False + def __str__(self): + return f"#{str(self.number)} {self.initials()}" + + def initials(self): + names = self.name.split() + outString = "" + for name in names: + outString += f"{name[0]}." + return outString class Skater(Player): """A hockey player that is not a goalie.""" + pass class Goalie(Player): """A hockey player that *is* a goalie.""" + pass class AtkAction(Enum): SkateB = 0 diff --git a/team.py b/team.py index 9d76f41..ff1f31c 100644 --- a/team.py +++ b/team.py @@ -5,6 +5,14 @@ class Team(object): """A team of either 6 or 10 skaters and 1-3 goalies.""" roster = [] #ordered, first line then second line; (#, name) goalies = [] # (#, name) + name = None + shortname = None + + def __init__(self, skaters, goalies, name:str, shortname:str): + self.roster = skaters + self.goalies = goalies + self.name = name + self.shortname = shortname def chooseGoalie(self): - return sample(goalies,1) \ No newline at end of file + return sample(self.goalies,1) \ No newline at end of file