request azure authorization token every 5mn, fixes #111

awesometts/service/azure.py

@@ -22,6 +22,7 @@
 """
 
 import time
+import datetime
 import requests
 from xml.etree import ElementTree
 from .base import Service
@@ -165,7 +166,8 @@ class Azure(Service):
     """
 
     __slots__ = [
-        'access_token'
+        'access_token',
+        'access_token_timestamp'
     ]
 
     NAME = "Microsoft Azure"
@@ -235,13 +237,25 @@ def get_token(self, subscription_key, region):
         }
         response = requests.post(fetch_token_url, headers=headers)
         self.access_token = str(response.text)
+        self.access_token_timestamp = datetime.datetime.now()
+        self._logger.debug(f'requested access_token')
+
+    def token_refresh_required(self):
+        if self.access_token == None:
+            self._logger.debug(f'no token, must request')
+            return True
+        time_diff = datetime.datetime.now() - self.access_token_timestamp
+        if time_diff.total_seconds() > 300:
+            self._logger.debug(f'time_diff: {time_diff}, requesting token')
+            return True
+        return False
 
     def run(self, text, options, path):
         """Downloads from Azure API directly to an MP3."""
 
         region = options['region']
         subscription_key = options['key']
-        if self.access_token == None:
+        if self.token_refresh_required():
             self.get_token(subscription_key, region)
 
         voice = options['voice']

tests/test_run.py

@@ -8,6 +8,7 @@
 import magic # to verify file types
 import os
 import sys
+import time
 
 class Success(Exception):
     pass
@@ -184,6 +185,7 @@ def run_service_testcases(self, svc_id, test_cases, extra_option_keys=[], lowerc
         self.logger.info(f'Default options for service {svc_id}: {options}')
 
         for test_case in test_cases:
+            # time.sleep(3) # used this to test azure token refresh code
             options['voice'] = test_case['voice']
             # set extra option keys
             for extra_option_key in extra_option_keys: