refactor all dict methods

data/tests.py

@@ -1,12 +1,13 @@
 import json
-  
+
 # Opening JSON file
-f = open('get_user_details_extended.json')
+f = open("get_user_details_extended.json")
-  
+
-# returns JSON object as 
+# returns JSON object as
 # a dictionary
 response = json.load(f)
 
+
 def is_valid_user_details(response) -> bool:
     """
     validate that the object passed is a dict containing at least the keys
@@ -27,6 +28,7 @@ def is_valid_user_details(response) -> bool:
         return True
     return False
 
+
 def is_wp_oauth_refresh_token_response(response) -> bool:
     """
     validate that the structure of the response contains the keys of
@@ -39,10 +41,9 @@ def is_wp_oauth_refresh_token_response(response) -> bool:
         return True
     return False
 
-print('is_valid_user_details')
+
+print("is_valid_user_details")
 print(is_valid_user_details(response))
 
-print('is_wp_oauth_refresh_token_response')
+print("is_wp_oauth_refresh_token_response")
 print(is_wp_oauth_refresh_token_response(response))
-
-

wp_oauth_backend/wp_oauth.py

@@ -109,19 +109,22 @@ class StepwiseMathWPOAuth2(BaseOAuth2):
         """
         return urlopen(url).read().decode("utf-8")
 
+    def is_valid_dict(self, response, qc_keys) -> bool:
+        if not type(response) == dict:
+            logger.warning(
+                "is_valid_dict() was expecting a dict but received an object of type: {type}".format(
+                    type=type(response)
+                )
+            )
+            return False
+        return all(key in response for key in qc_keys)
+
     def is_valid_user_details(self, response) -> bool:
         """
         validate that the object passed is a json dict containing at least
         the keys in qc_keys. These are the dict keys created in get_user_details()
         default return object.
         """
-        if not type(response) == dict:
-            logger.warning(
-                "is_valid_user_details() was expecting a dict but received an object of type: {type}".format(
-                    type=type(response)
-                )
-            )
-            return False
         qc_keys = [
             "id",
             "date_joined",
@@ -133,41 +136,21 @@ class StepwiseMathWPOAuth2(BaseOAuth2):
             "last_name",
             "username",
         ]
-        if all(key in response for key in qc_keys):
+        return self.is_valid_dict(response, qc_keys)
-            return True
-        return False
 
     def is_wp_oauth_error(self, response) -> bool:
         """
         validate the structure of the response object conforms to a
         wp-oauth error json dict.
         """
-        if not type(response) == dict:
-            logger.warning(
-                "is_wp_oauth_error() was expecting a dict but received an object of type: {type}".format(
-                    type=type(response)
-                )
-            )
-            return False
-        if len(response.keys()) != 2:
-            return False
         qc_keys = ["error" "error_description"]
-        if all(key in response for key in qc_keys):
+        return self.is_valid_dict(response, qc_keys) and len(response.keys()) == 2
-            return True
-        return False
 
     def is_wp_oauth_response(self, response) -> bool:
         """
         validate the structure of the response object from wp-oauth. it's
         supposed to be a dict with at least the keys included in qc_keys.
         """
-        if not type(response) == dict:
-            logger.warning(
-                "is_wp_oauth_response() was expecting a dict but received an object of type: {type}".format(
-                    type=type(response)
-                )
-            )
-            return False
         qc_keys = [
             "ID",
             "capabilities",
@@ -178,21 +161,15 @@ class StepwiseMathWPOAuth2(BaseOAuth2):
             "user_registered",
             "user_status",
         ]
-        if all(key in response for key in qc_keys):
+        return self.is_valid_dict(response, qc_keys)
-            return True
-        return False
 
     def is_wp_oauth_refresh_token_response(self, response) -> bool:
         """
         validate that the structure of the response contains the keys of
         a refresh token dict.
         """
-        if not self.is_valid_user_details(response):
-            return False
         qc_keys = ["access_token", "expires_in", "refresh_token", "scope", "token_type"]
-        if all(key in response for key in qc_keys):
+        return self.is_valid_dict(response, qc_keys)
-            return True
-        return False
 
     def is_get_user_details_extended_dict(self, response) -> bool:
         """
@@ -200,21 +177,17 @@ class StepwiseMathWPOAuth2(BaseOAuth2):
         contains a.) all keys of a get_user_details() return, plus,
         b.) all keys of a wp-oauth refresh token response.
         """
-        if not self.is_valid_user_details(response):
+        return self.is_valid_user_details(
-            return False
+            response
-        if self.is_wp_oauth_refresh_token_response(response):
+        ) and self.is_wp_oauth_refresh_token_response(response)
-            return True
-        return False
 
     def is_valid_get_user_details_response(self, response) -> bool:
         """
         True if the response object can be processed by get_user_details()
         """
-        if self.is_valid_user_details(response):
+        return self.is_valid_user_details(response) or self.is_wp_oauth_response(
-            return True
+            response
-        if self.is_wp_oauth_response(response):
+        )
-            return True
-        return False
 
     def get_response_type(self, response) -> str:
         if type(response) != dict:
@@ -295,7 +268,7 @@ class StepwiseMathWPOAuth2(BaseOAuth2):
             logger.info(
                 "get_user_details() received {t}: {response}".format(
                     t=self.get_response_type(response),
-                    response=json.dumps(response, sort_keys=True, indent=4)
+                    response=json.dumps(response, sort_keys=True, indent=4),
                 )
             )
         # a def in the third_party_auth pipeline list calls get_user_details() after its already