diff --git a/main.py b/main.py index 2586f9d..55929a0 100644 --- a/main.py +++ b/main.py @@ -26,12 +26,18 @@ def main(): required=False) args = arg_parser.parse_args() + events = init_games() if args.team: - # find the team - print(args.team) + found_event= filter(lambda e: e.home_team.abbr.lower() == args.team.lower() or e.away_team.abbr.lower() == args.team.lower(), events) + if found_event: + event = list(found_event).pop() + table = output_game(event) + console.print(table) + else: + print(f"Team not found: {args.team}") else: - for event in init_games(): + for event in events: table = output_game(event) console.print(table)