Changed the header keys inline with the change to the H51 application

Author: anthonyjb

h51/client.py

@@ -10,7 +10,7 @@
 
 class Client:
     """
-    A client for the Hangar51 API.
+    A client for the H51 (Hangar51) API.
     """
 
     def __init__(self, api_key, api_base_url='https://api.h51.io'):
@@ -60,7 +60,7 @@ def __call__(self,
         """Call the API"""
 
         # Build headers
-        headers = {'X-Hangar51-APIKey': self._api_key}
+        headers = {'X-H51-APIKey': self._api_key}
 
         if json_type_body:
             headers['Content-Type'] = 'application/json'
@@ -83,12 +83,12 @@ def __call__(self,
         )
 
         # Update the rate limit
-        if 'X-Hangar51-RateLimit-Limit' in r.headers:
-            self._rate_limit = int(r.headers['X-Hangar51-RateLimit-Limit'])
+        if 'X-H51-RateLimit-Limit' in r.headers:
+            self._rate_limit = int(r.headers['X-H51-RateLimit-Limit'])
             self._rate_limit_reset \
-                    = float(r.headers['X-Hangar51-RateLimit-Reset'])
+                    = float(r.headers['X-H51-RateLimit-Reset'])
             self._rate_limit_remaining \
-                    = int(r.headers['X-Hangar51-RateLimit-Remaining'])
+                    = int(r.headers['X-H51-RateLimit-Remaining'])
 
         # Handle a successful response
         if r.status_code in [200, 204]:

h51/resources.py

@@ -99,6 +99,19 @@ def __init__(self, client, document):
     def __str__(self):
         return f'Asset: {self.uid}'
 
+    def analyze(self, analyzers, notification_url=None):
+        """Analyze the asset"""
+
+        r = self._client(
+            'post',
+            f'assets/{self.uid}/analyze',
+            params={'notification_url': notification_url},
+            json_type_body=[a.to_json_type() for a in analyzers]
+        )
+
+        if not notification_url:
+            self.meta = r['meta']
+
     def download(self):
         """Download the asset"""
         return self._client(
@@ -121,8 +134,7 @@ def expire(self, seconds):
 
         self._document.update(r)
 
-
-    def persist(self,):
+    def persist(self):
         """Set the asset to persist (remove the expires time)"""
 
         r = self._client(
@@ -132,19 +144,6 @@ def persist(self,):
 
         self._document.update(r)
 
-    def analyze(self, analyzers, notification_url=None):
-        """Analyze the asset"""
-
-        r = self._client(
-            'post',
-            f'assets/{self.uid}/analyze',
-            params={'notification_url': notification_url},
-            json_type_body=[a.to_json_type() for a in analyzers]
-        )
-
-        if not notification_url:
-            self.meta = r['meta']
-
     @classmethod
     def all(self, client, secure=None, type=None, q=None, rate_buffer=0):
         """

setup.py

@@ -21,7 +21,7 @@
     # Versions should comply with PEP440.  For a discussion on single-sourcing
     # the version across setup.py and the project code, see
     # https://packaging.python.org/en/latest/single_source_version.html
-    version='0.0.1',
+    version='0.0.2',
     description=\
         'The H51 Python library provides a pythonic interface to the H51 API.',
     long_description=long_description,