PyTransportNSWv2 2.0.1__tar.gz → 2.0.2__tar.gz

{PyTransportNSWv2-2.0.1 → PyTransportNSWv2-2.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyTransportNSWv2
-Version: 2.0.1
+Version: 2.0.2
 Summary: Get detailed per-trip transport information from TransportNSW
 Home-page: https://github.com/andystewart999/TransportNSW
 Author: andystewart999

{PyTransportNSWv2-2.0.1 → PyTransportNSWv2-2.0.2}/PyTransportNSWv2.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyTransportNSWv2
-Version: 2.0.1
+Version: 2.0.2
 Summary: Get detailed per-trip transport information from TransportNSW
 Home-page: https://github.com/andystewart999/TransportNSW
 Author: andystewart999

{PyTransportNSWv2-2.0.1 → PyTransportNSWv2-2.0.2}/TransportNSWv2/TransportNSWv2.py

@@ -12,6 +12,15 @@ import re
 import json					#For the output
 import time
 
+# Global API variable
+api_calls = 0
+
+def increment_api_counter(source):
+    global api_calls
+    api_calls += 1
+    #print(str(api_calls) + " " + source)
+
+# Constants
 ATTR_DUE_IN = 'due'
 
 ATTR_ORIGIN_STOP_ID = 'origin_stop_id'
@@ -117,7 +126,10 @@ class TransportNSWv2(object):
                 error_code = 0
 
                 if not skip_api_calls:
+                    # Make the call and increment the API counter
                     response = requests.get(url, headers=header, timeout=5)
+                    increment_api_counter('stop_finder')
+
                 else:
                     # An earlier call resulted in an API key error so no point trying again
                     response.status_code = 401
@@ -195,7 +207,7 @@ class TransportNSWv2(object):
             origin_transport_type = [origin_transport_type]
 
         if isinstance(destination_transport_type, int):
-            destiation_transport_type = [destination_transport_type]
+            destination_transport_type = [destination_transport_type]
 
         if isinstance(alert_type, str):
             alert_type = alert_type.split('|')
@@ -242,6 +254,7 @@ class TransportNSWv2(object):
         # Otherwise store the response for the next steps
         try:
             response = requests.get(url, headers=header, timeout=10)
+            increment_api_counter('trip')
 
         except Exception as ex:
             raise TripError (f"Error '{str(ex)}' calling trip API for journey {name_origin} to {name_destination}")
@@ -426,7 +439,7 @@ class TransportNSWv2(object):
 
                 current_journey_index = next_journey_index
 
-        json_output='{"journeys_to_return": ' + str(journeys_to_return) + ', "journeys_with_data": ' + str(found_journeys) + ', "journeys": [' + json_output + ']}'
+        json_output='{"journeys_to_return": ' + str(journeys_to_return) + ', "journeys_with_data": ' + str(found_journeys) + ', "api_calls": '  + str(api_calls) + ', "journeys": [' + json_output + ']}'
         return json_output
 
 
@@ -559,12 +572,14 @@ class TransportNSWv2(object):
         header = {'Accept': 'application/json', 'Authorization': auth}
 
         url_base_path = self._get_base_url(mode)
-
         url_mode_list = self._get_mode_list(mode, agencyid)
+
         if not url_mode_list is None:
             for mode_url in url_mode_list:
                 url = url_base_path + mode_url
                 response = requests.get(url, headers=header, timeout=10)
+                increment_api_counter(url)
+
                 # Only try and process the results if we got a good return code
                 if response.status_code == 200:
                     # Search the feed and see if we can match realtimetripid to trip_id

{PyTransportNSWv2-2.0.1 → PyTransportNSWv2-2.0.2}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="PyTransportNSWv2",
-    version="2.0.1",
+    version="2.0.2",
     author="andystewart999",
     author_email="andy.stewart@live.com",
     description="Get detailed per-trip transport information from TransportNSW",