add hash_registry kward to hash the body of requests to the registry

Author: dy-waltergrantward

dynata_rex/opportunity_registry.py

@@ -27,6 +27,9 @@ def __init__(self,
                  access_key: str,
                  secret_key: str,
                  base_url: str = _BASE_URL,
+                 # TODO: Hack Alert Remove Hash Registry
+                 hash_registry: bool = False,
+                 # END TODO: Hack Alert
                  default_ttl: int = 10,
                  shard_count: int = 1,
                  current_shard: int = 1):
@@ -43,7 +46,8 @@ def __init__(self,
         self.default_ttl = default_ttl
         self.make_request = RexRequest(access_key,
                                        secret_key,
-                                       default_ttl=default_ttl)
+                                       default_ttl=default_ttl,
+                                       hash_registry=hash_registry)
         self.base_url = base_url
 
         if current_shard > shard_count:

dynata_rex/signer.py

@@ -144,10 +144,18 @@ def sign_query_parameters_from_ttl(self,
 class RexRequest:
     """Wrapper for http calls to include our signature"""
 
-    def __init__(self, access_key, secret_key, default_ttl: int = 10):
+    def __init__(self,
+                 access_key,
+                 secret_key,
+                 default_ttl: int = 10,
+                 hash_registry: bool = False):
         self.default_ttl = default_ttl
         self.access_key = access_key
         self.secret_key = secret_key
+        # TODO: Hack Alert: Including this for compatability
+        # Once completed, all refs to the registry hack should be removed
+        self.hash_registry = hash_registry
+        # END TODO: Hack Alert
         self.signer = Signer(access_key, secret_key)
         self.session = make_session()
 
@@ -161,12 +169,15 @@ def _signature(self, ttl: int = None, signing_string: str = None) -> str:
             signing_string=signing_string
         )
 
-    def _create_auth_headers(self, url, additional_headers={}, body=''):
+    def _create_auth_headers(self,
+                             url,
+                             additional_headers={},
+                             body=''):
         signing_string = self.signer._create_request_body_signing_string(body)
 
         # TODO: Hack alert: Registry doesn't like the sha256 body signing
         # string yet
-        if 'https://registry' in url:
+        if self.hash_registry is False and 'https://registry' in url:
             signing_string = ''
         # END Hack alert
 
@@ -188,7 +199,9 @@ def dispatch(self,
             additional_headers = {'Content-type': 'application/json'}
             data = json.dumps(data)
 
-        headers = self._create_auth_headers(url, additional_headers, body=data)
+        headers = self._create_auth_headers(url,
+                                            additional_headers,
+                                            body=data)
 
         if not hasattr(self.session, method.lower()):
             raise AttributeError('Invalid http method provided.')