Updating headers according to the upcoming change from Quby

toonapilib/toonapilib.py

@@ -152,6 +152,7 @@ def _authenticate(self):
         self._token = self._get_token(code)
         self._set_headers(self._token)
         self._get_agreements()
+        self._update_headers()
         self._api_url = '{}/toon/v3/{}'.format(self._base_url,
                                                self.agreement.id)
 
@@ -160,6 +161,10 @@ def _set_headers(self, token):
                          'content-type': 'application/json',
                          'cache-control': 'no-cache'}
 
+    def _update_headers(self):
+        self._headers.update({'X-Common-Name': self.agreement.display_common_name,
+                              'X-Agreement-ID': self.agreement.id})
+
     def _get_agreements(self):
         url = '{base_url}/toon/v3/agreements'.format(base_url=self._base_url)
         response = requests.get(url, headers=self._headers)
@@ -273,6 +278,7 @@ def _patched_request(self, *args, **kwargs):
             self._set_headers(self._token)
             kwargs['headers'].update(
                 {'Authorization': 'Bearer {}'.format(self._token.access_token)})
+            self._update_headers()
             self._logger.debug('Updated headers, trying again initial request')
             response = self.original_request(*args, **kwargs)
         return response