F1PointsOverTime/utils.py at master · BenTaylor25/F1PointsOverTime · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
from datetime import datetime

# For x-axis formatting.
def non_sprint_dates(dates: list[datetime]) -> list[datetime]:
    """
    Filter sprint race dates from list of dates.
    A sprint race occurs 1 day before the next date.
    """
    sprint_dates = []

    for i, date in enumerate(dates):
        # Ignore last date.
        if date == dates[-1]:
            break

        next_date = dates[i + 1]

        days_between = (next_date - date).days

        if days_between == 1:
            sprint_dates.append(date)

    # Filter sprint_dates from dates.
    non_sprint_dates = [date for date in dates if date not in sprint_dates]

    return non_sprint_dates

def non_sprint_dated_event_names(dated_event_names: list[str], dates: list[datetime]) -> list[str]:
    non_sprint_dated_event_names: list[str] = []

    for (dated_event_name, date) in zip(dated_event_names, dates):
        if date in non_sprint_dates(dates):
            non_sprint_dated_event_names.append(dated_event_name)

    return non_sprint_dated_event_names

# For y-axis formatting.
def highest_team_points(season) -> int:
    """
    Search a season team points.
    """
    highest = 0

    for team_name in season["team_names_to_show"]:
        cum_team_points = season["team_points"][team_name]
        team_highest = cum_team_points[-1]

        highest = max(highest, team_highest)

    return highest