133 lines
3.7 KiB
Python
133 lines
3.7 KiB
Python
"""
|
|
Provides all info about V³ stations
|
|
"""
|
|
|
|
from tbm_api_lib.libs import get_data_from_json
|
|
from time import time
|
|
|
|
vcub_url = "https://ws.infotbm.com/ws/1.0/vcubs"
|
|
|
|
|
|
class Vcub:
|
|
"""
|
|
Retrieves information from V³ stations
|
|
|
|
Data format, as returned by the infotbm website:
|
|
{
|
|
lists: [
|
|
{
|
|
'id': station number,
|
|
'name': str,
|
|
'connexionState': 'CONNECTED' if in service 'DISCONNECTED' otherwise,
|
|
'typeVlsPlus': 'VLS_PLUS' if V³+ 'PAS_VLS_PLUS' otherwise,
|
|
'nbPlaceAvailable': 33,
|
|
'nbBikeAvailable': 0
|
|
'nbElectricBikeAvailable': 6
|
|
'latitude': float,
|
|
'longitude': float,
|
|
},
|
|
]
|
|
}
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
autoupdate_at_creation=None,
|
|
autoupdate=False,
|
|
autoupdate_delay=-1,
|
|
data=None,
|
|
):
|
|
self.last_update = 0
|
|
if type(data) == dict:
|
|
self.data = self.update(data=data)
|
|
else:
|
|
self.data = None
|
|
if autoupdate_at_creation or (
|
|
autoupdate_at_creation is None and self.data is None
|
|
):
|
|
self.update()
|
|
|
|
def update(self, auto=False, data=None):
|
|
"""
|
|
Updates data
|
|
auto optionnal param is to set if a update is a automatic one,
|
|
and must be performed as defined in autoupdate_delay variable
|
|
"""
|
|
|
|
if data is None or type(data) != dict:
|
|
d = get_data_from_json(vcub_url)
|
|
else:
|
|
d = data
|
|
# the original format is awfull, so I change it a little
|
|
if type(d) == dict:
|
|
self.data = {}
|
|
d = d["lists"]
|
|
for i in d:
|
|
e = {
|
|
"name": i["name"],
|
|
"online": i["connexionState"] == "CONNECTEE",
|
|
"plus": i["typeVlsPlus"] == "VLS_PLUS",
|
|
"empty": int(i["nbPlaceAvailable"]),
|
|
"bikes": int(i["nbBikeAvailable"]),
|
|
"ebikes": int(i["nbElectricBikeAvailable"]),
|
|
"location": (float(i["latitude"]), float(i["longitude"])),
|
|
}
|
|
self.data[int(i["id"])] = e
|
|
self.last_update = time()
|
|
|
|
def data_age(self):
|
|
"""
|
|
Computes the data's age
|
|
"""
|
|
|
|
return time() - self.last_update
|
|
|
|
def get_names(self):
|
|
"""
|
|
Returns all names in a dict with id as data
|
|
"""
|
|
|
|
r = {}
|
|
for i in self.data:
|
|
r[self.data[i]["name"]] = i
|
|
return r
|
|
|
|
def get_locations(self):
|
|
"""
|
|
Returns all locations in a dict with id as data
|
|
"""
|
|
|
|
r = {}
|
|
for i in self.data:
|
|
r[self.data[i]["location"]] = i
|
|
return r
|
|
|
|
def get_by_id(self, id):
|
|
"""
|
|
Returns a station by its id
|
|
"""
|
|
|
|
class Station:
|
|
"""
|
|
A V³ station
|
|
"""
|
|
|
|
def __init__(self, data, id):
|
|
self.data = data
|
|
self.id = id
|
|
self.name = self.data["name"]
|
|
self.location = self.data["location"]
|
|
self.online = self.data["online"]
|
|
self.isplus = self.data["plus"]
|
|
self.bikes = self.data["bikes"]
|
|
self.ebikes = self.data["ebikes"]
|
|
self.empty = self.data["empty"]
|
|
|
|
def __int__(self):
|
|
return self.id
|
|
|
|
return Station(self.data[id], id)
|
|
|
|
def get_all_ids(self):
|
|
return tuple(self.data.keys())
|