Skip to content

Commit 7c6772d

Browse files
committed
pass the body and the is_form_encoding flag in to the Request object (for future use in implementing the hash-of-body extension)
1 parent 615b1ff commit 7c6772d

File tree

1 file changed

+12
-10
lines changed

1 file changed

+12
-10
lines changed

oauth2/__init__.py

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -317,8 +317,9 @@ class Request(dict):
317317
"""
318318

319319
version = OAUTH_VERSION
320-
321-
def __init__(self, method=HTTP_METHOD, url=None, parameters=None):
320+
321+
def __init__(self, method=HTTP_METHOD, url=None, parameters=None,
322+
body=None, is_form_encoded=False):
322323
if url is not None:
323324
self.url = to_unicode(url)
324325
self.method = method
@@ -327,6 +328,8 @@ def __init__(self, method=HTTP_METHOD, url=None, parameters=None):
327328
k = to_unicode(k)
328329
v = to_unicode_optional_iterator(v)
329330
self[k] = v
331+
self.body = body
332+
self.is_form_encoded = is_form_encoded
330333

331334

332335
@setter
@@ -513,7 +516,8 @@ def from_request(cls, http_method, http_url, headers=None, parameters=None,
513516

514517
@classmethod
515518
def from_consumer_and_token(cls, consumer, token=None,
516-
http_method=HTTP_METHOD, http_url=None, parameters=None):
519+
http_method=HTTP_METHOD, http_url=None, parameters=None,
520+
body=None, is_form_encoded=False):
517521
if not parameters:
518522
parameters = {}
519523

@@ -532,7 +536,8 @@ def from_consumer_and_token(cls, consumer, token=None,
532536
if token.verifier:
533537
parameters['oauth_verifier'] = token.verifier
534538

535-
return Request(http_method, http_url, parameters)
539+
return Request(http_method, http_url, parameters, body=body,
540+
is_form_encoded=is_form_encoded)
536541

537542
@classmethod
538543
def from_token_and_callback(cls, token, callback=None,
@@ -619,15 +624,12 @@ def request(self, uri, method="GET", body=None, headers=None,
619624

620625
req = Request.from_consumer_and_token(self.consumer,
621626
token=self.token, http_method=method, http_url=uri,
622-
parameters=parameters)
627+
parameters=parameters, body=body, is_form_encoded=is_form_encoded)
623628

624629
req.sign_request(self.method, self.consumer, self.token)
625630

626-
if method == "POST":
627-
if is_form_encoded:
628-
body = req.to_postdata()
629-
else:
630-
headers.update(req.to_header())
631+
if is_form_encoded:
632+
body = req.to_postdata()
631633
elif method == "GET":
632634
uri = req.to_url()
633635
else:

0 commit comments

Comments
 (0)