joeni

# Bus module: Collects bus data for a given stop and returns the data in Python data structures

import json

import requests

from datetime import datetime, tzinfo

from django.core.cache import caches

def departures(stop, amount=0):

    pass

    """ Returns the remaining departures of buses from the given stop for this day.

    If amount is given, a maximum amount of departures will be returned.

    Departures in the past are never returned."""

    #today = datetime.now().strftime("%d-%m-%Y")

    #cache = caches["departures"]

    #cached_data = cache.get(stop)

    #if cached_data is None:

        ####received_json_data = requests.get('https://www.delijn.be/rise-api-core/dienstregeling/halteDienstregeling/'+stop+'/'+today+'/nl').json()

        #buses = list()

        #for bus_json in received_json_data['halteDoorkomsten']:

            #bus = dict()

        ## Some destinations should be a little bit more descriptive, they're handled here hardcoded

        #destination = bus_json["bestemming"]

        #bus["time"] = bus_json["lijn"]["tijdstip"]

# TODO: I've opted not to use the caching system for a complete day timetable, because it wouldn't load

# via destinations and give a rather hard time to work with. So I'm gonna stick with the simple

# realtime thingy, which does the job well enough.

# For the new kind, use https://www.delijn.be/rise-api-core/haltes/doorkomstenditmoment/403040/40

# Where the "40" can be any number. That one does show the via.

def arrivals(stops):

    """ Returns the a list of buses that stop at the given stops.

    The buses are in a list, with dicts with the following keys:

    - number :: The line number of the bus

    - color :: The hex color to display for this bus line

    - destination :: The name of the destination (already formatted properly)

    - time :: The datetime of the departure

    - via :: If the bus does a detour, the name is stored here

    The list can have a maximum of 20 entries.

    The subroutine throws an exception if no stops were given.

    """

    stops_string = stops.pop(0)  # This returns an error in an empty list, which is supposed to happen if no stops are sent

    while len(stops) != 0:

        stops_string = stops_string + "+" + stops.pop(0)

    received_json_data = requests.get('https://www.delijn.be/rise-api-core/haltes/Multivertrekken/'+stops_string+'/20').json()

    buses = list()

    for bus_json in received_json_data["lijnen"]:

        bus = dict()

        # Some destinations should be a little bit more descriptive, they're handled here hardcoded

        destination = bus_json["bestemming"]

        if destination in ["Hasselt", "Genk", "Neerpelt", "Tongeren", "Maastricht", "Bilzen", "Diepenbeek"]:

            destination += " Station"

        elif destination == "Kiewit Kinderb.":

            destination = "Kiewit Kinderboerderij"

        elif destination == "Beringen":

            destination = "Beringen Bogaersveld"

        elif destination == "Maaseik":

            destination = "Maaseik Van Eycklaan"

        bus["destination"] = destination

        bus["number"] = bus_json["lijnNummerPubliek"]

        bus["color"] = bus_json["kleurAchterGrond"]

        unix_time = str(bus_json["vertrekCalendar"])[:10]  # De Lijn API returns too many trailing zeros

        bus["time"] = datetime.fromtimestamp(int(unix_time))

        # About via:

        # The viaBestemming has an annoying tendency to uppercase all destinations.

        # Some can be hardcoded to the right destination, others just need to be

        # properly capitalized.

        via = bus_json["viaBestemming"]

        if via is not None:

            if via == "GENK":

                via = "Genk Station"

            elif via == "MM VILL & GENK" or via == "GENK & MM VILL":

                via = "Genk Station en Maasmechelen Village"

            elif via == "UNIVERSITEIT":

                via = "Universiteit Hasselt"

            elif via == "HASSELT":

                via = "Hasselt station"

            elif via == "PATERSPLEIN":

                via = "Diepenbeek Patersplein"

            elif via == "DIEP. STATION":

                via = "Diepenbeek Station"

            elif via == "GENK EN UNIV.":

                via = "Universiteit Hasselt en Genk Station"

            else:

                via = via.title()  # If it's anything else just capitalize and hope for the best

            bus["via"] = via

        buses.append(bus)

    return buses