aboutsummaryrefslogtreecommitdiff
path: root/util.py
blob: fe0ff019585838afb67b30b9809b6fde7d1a63f3 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from datetime import datetime, timezone
from zoneinfo import ZoneInfo


def parse_time(maybe_time):
    if maybe_time:
        return datetime.fromisoformat(maybe_time[:-1] + '+00:00')
    return None


def filter_departures(departures):
    return [
        dep for dep in departures if dep['scheduled_departure_utc']
        and parse_time(dep['scheduled_departure_utc']) >
        datetime.now().astimezone(timezone.utc)
    ]


def format_time(maybe_time):
    if maybe_time:
        return str(maybe_time.astimezone(ZoneInfo('Australia/Melbourne')))[:-6]
    return None