diff --git a/main.py b/main.py index ca98197..cb3f4c3 100644 --- a/main.py +++ b/main.py @@ -111,5 +111,8 @@ def main(): ) ) + for x in formula_one_2023.getTeamStandings(): + print(x) + if __name__ == "__main__": main() \ No newline at end of file diff --git a/src/championship/championship.py b/src/championship/championship.py index 475858a..047eddd 100644 --- a/src/championship/championship.py +++ b/src/championship/championship.py @@ -88,8 +88,15 @@ def getTeamStandings(self) -> List: returns: List ''' - # TODO - pass + sorted_standings = sorted(self.team_standings.items(), key=lambda x:x[1]) + sorted_standings.reverse() + + printable = list() + + for i in range(len(sorted_standings)): + printable.append(f"{i+1}: {sorted_standings[i][0].name} ({sorted_standings[i][1]} points)") + + return printable def holdRace(self, race: Race, results: List[Driver]): ''' diff --git a/src/team/teamFactory.py b/src/team/teamFactory.py index ba5d319..5355d05 100644 --- a/src/team/teamFactory.py +++ b/src/team/teamFactory.py @@ -5,7 +5,7 @@ class TeamFactory: @staticmethod - def createTeam(self, name: str, engine: EngineInt | str, country: str, drivers: List(Driver)) -> Team: + def createTeam(name: str, engine: EngineInt | str, country: str, drivers: List[Driver]) -> Team: """ create a team with a custom engine @@ -27,7 +27,7 @@ def createTeam(self, name: str, engine: EngineInt | str, country: str, drivers: return Team(name = name, engine = engine, country = country, drivers = drivers) @staticmethod - def createTeamFerrariEngine(self, name: str, country: str, drivers: List(Driver)) -> Team: + def createTeamFerrariEngine(name: str, country: str, drivers: List[Driver]) -> Team: """ create a team with a Ferrari engine @@ -44,7 +44,7 @@ def createTeamFerrariEngine(self, name: str, country: str, drivers: List(Driver) return Team(name = name, engine = FerrariEngine(), country = country, drivers = drivers) @staticmethod - def createTeamHondaRBPTEngine(self, name: str, country: str, drivers: List(Driver)) -> Team: + def createTeamHondaRBPTEngine(name: str, country: str, drivers: List[Driver]) -> Team: """ create a team with a Honda Red Bull Power Trains engine @@ -61,7 +61,7 @@ def createTeamHondaRBPTEngine(self, name: str, country: str, drivers: List(Drive return Team(name = name, engine = HondaRBPTEngine(), country = country, drivers = drivers) @staticmethod - def createTeamMercedesEngine(self, name: str, country: str, drivers: List(Driver)) -> Team: + def createTeamMercedesEngine(name: str, country: str, drivers: List[Driver]) -> Team: """ create a team with a Mercedes engine @@ -78,7 +78,7 @@ def createTeamMercedesEngine(self, name: str, country: str, drivers: List(Driver return Team(name = name, engine = MercedesEngine(), country = country, drivers = drivers) @staticmethod - def createTeamRenaultEngine(self, name: str, country: str, drivers: List(Driver)) -> Team: + def createTeamRenaultEngine(name: str, country: str, drivers: List[Driver]) -> Team: """ create a team with a Renault engine