Add function to check multiple Velodi stations at a time + fix bug + bump version n° to 2.1

Author: filau

divia_api/__init__.py

@@ -25,6 +25,6 @@
 __author__ = 'Firmin Launay'
 __license__ = 'LGPL-3.0'
 __copyright__ = 'divia_api  Copyright (C) 2021  Firmin Launay'
-__version__ = '2.0'
+__version__ = '2.1'
 
 from .api import DiviaAPI

divia_api/velodi.py

@@ -86,5 +86,29 @@ def find_station(self, station_name: str) -> VelodiStation:
 
     def get_station(self, station_code: str) -> VelodiStation:
         corresponding_stations = list(item for item in self.stations if item.code == station_code)
-        if len(corresponding_stations > 0):
+        if len(corresponding_stations) > 0:
             return corresponding_stations[0]
+
+    def find_stations(self, station_names: list) -> list:
+        stations = []
+        for station_name in station_names:
+            stations.append(self.find_station(station_name))
+        return stations
+
+    def get_stations(self, station_codes: list) -> list:
+        stations = []
+        for station_code in station_codes:
+            stations.append(self.get_station(station_code))
+        return stations
+
+    def check_multiple_stations(self, stations: list) -> list:
+        velodi_data = update_source()
+        results = []
+        for station in stations:
+            query_result = list(item for item in velodi_data if (item["infos"]["code_cykleo"] == station.code))
+            try:
+                free = query_result[0]["infos"]["qucit"]["realtime"]
+                results.append(FreeVelodi(free["bikes"], free["docks"]))
+            except IndexError:
+                raise Exception("Data not found.")
+        return results

setup.py

@@ -31,15 +31,15 @@
 setup(
     name='divia_api',
     packages=['divia_api'],
-    version='2.0',
+    version='2.1',
     license='LGPL-3.0',
     description='divia_api is a Python library that allows to retrieve the timetable of Divia’s bus and tramways, along with some infos about DiviaVélodi bikes, straight from a Python script.',
     long_description=long_description,
     long_description_content_type='text/markdown',
     author='Firmin Launay',
     author_email='hey@firminlaunay.me',
     url='https://github.com/filau/python_divia_api',
-    download_url='https://github.com/filau/python_divia_api/archive/refs/tags/2.0.tar.gz',
+    download_url='https://github.com/filau/python_divia_api/archive/refs/tags/2.1.tar.gz',
     keywords=['divia', 'api', 'firmin', 'launay', 'dijon', 'bus', 'tram'],
     install_requires=[
         'requests'