Skip to content

Commit a70c724

Browse files
Merge pull request #173 from vmware/AV-182410-22.1.4
Fixing pep8 issues(spacing) in 22.1.4
2 parents 130a1d0 + c5582ed commit a70c724

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

plugins/module_utils/saml_avi_api.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
logger = logging.getLogger(__name__)
2323
warnings.filterwarnings('ignore')
2424

25+
2526
class WS1loginSAMLApiSession(ApiSession):
2627
SAML_URL_SUFFIX = "/sso/login"
2728
# Request RegX
@@ -85,8 +86,8 @@ def saml_assertion(self, username, password):
8586
logger.error("SAML request not generated by controller.")
8687
raise APIError("SAML request not generated by controller.")
8788
saml_request = saml_request_match.group(1)
88-
relay_state = re.search(WS1loginSAMLApiSession.request_relay_state_regex, resp.text,re.M | re.S).group(1)
89-
assertion_url = re.search(WS1loginSAMLApiSession.request_assertion_url_regex, resp.text,re.M | re.S).group(1)
89+
relay_state = re.search(WS1loginSAMLApiSession.request_relay_state_regex, resp.text, re.M | re.S).group(1)
90+
assertion_url = re.search(WS1loginSAMLApiSession.request_assertion_url_regex, resp.text, re.M | re.S).group(1)
9091
headers = {'Content-Type': 'application/x-www-form-urlencoded'}
9192
saml_data = urllib.parse.urlencode({
9293
'SAMLRequest': saml_request,
@@ -125,7 +126,7 @@ def saml_assertion(self, username, password):
125126
json_response = json.loads(auth_resp.content)
126127
idp_token = json_response['sessionToken']
127128
basic_auth_header = "Bearer " + idp_token
128-
idp_session.headers.update({"authorization" : basic_auth_header})
129+
idp_session.headers.update({"authorization": basic_auth_header})
129130
idp_resp = idp_session.post(assertion_url, headers=headers,
130131
data=saml_data, allow_redirects=True)
131132
if idp_resp.status_code not in (200, 301, 302):
@@ -217,6 +218,7 @@ def authenticate_session(self):
217218
self.authenticate_session()
218219
return
219220

221+
220222
class OneloginSAMLApiSession(ApiSession):
221223
"""
222224
Extends the ApiSession class to override authentication

0 commit comments

Comments
 (0)