From db0cc383655034a36c46a3dbe9037847835470fb Mon Sep 17 00:00:00 2001 From: Jim Duchek Date: Wed, 19 Jan 2022 20:07:04 -0800 Subject: [PATCH] Fixed error checking with updated content/body --- requests_http_signature/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/requests_http_signature/__init__.py b/requests_http_signature/__init__.py index 05a1fe3..21f5949 100644 --- a/requests_http_signature/__init__.py +++ b/requests_http_signature/__init__.py @@ -82,9 +82,10 @@ class HTTPSignatureAuth(requests.auth.AuthBase): return request.body def add_digest(self, request): - if request.body is None and "digest" in self.headers: + content = self.digest_content + if content is None and "digest" in self.headers: raise RequestsHttpSignatureException("Could not compute digest header for request without a body") - if request.body is not None and "Digest" not in request.headers: + if content is not None and "Digest" not in request.headers: if "digest" not in self.headers: self.headers.append("digest") hash = hashlib.new(self.digest_alg)