Base resource

Author: Henry Chan

workos/resources/base.py

@@ -0,0 +1,37 @@
+class WorkOSBaseResource(object):
+  """Representation of a WorkOS Resource as returned through the API.
+    
+  Attributes:
+      OBJECT_FIELDS (list): List of fields a Resource is comprised of.
+  """
+
+  OBJECT_FIELDS = []
+
+  @classmethod
+  def construct_from_response(cls, response):
+    """Returns an instance of WorkOSBaseResource.
+
+    Args:
+        response (dict): Resource data from a WorkOS API response
+
+    Returns:
+        WorkOSBaseResource: Instance of a WorkOSBaseResource with OBJECT_FIELDS fields set
+    """
+    print(response)
+    obj = cls()
+    for field in cls.OBJECT_FIELDS:
+      setattr(obj, field, response[field])
+
+    return obj
+
+  def to_dict(self):
+    """Returns a dict representation of the WorkOSBaseResource.
+        
+    Returns:
+        dict: A dict representation of the WorkOSBaseResource
+    """
+    obj_dict = {}
+    for field in self.OBJECT_FIELDS:
+      obj_dict[field] = getattr(self, field, None)
+
+    return obj_dict

workos/resources/sso.py

@@ -1,4 +1,6 @@
-class WorkOSProfile(object):
+from workos.resources.base import WorkOSBaseResource
+
+class WorkOSProfile(WorkOSBaseResource):
     """Representation of a User Profile as returned by WorkOS through the SSO feature.
     
     Attributes:
@@ -12,34 +14,4 @@ class WorkOSProfile(object):
         "last_name",
         "connection_type",
         "idp_id",
-    ]
-
-    @classmethod
-    def construct_from_response(cls, response):
-        """Returns an instance of WorkOSProfile.
-
-        Args:
-            response (dict): Response from a WorkOS API request as returned by RequestHelper
-
-        Returns:
-            WorkOSProfile: The WorkOS profile of a User
-        """
-        profile_data = response["profile"]
-
-        profile = cls()
-        for field in WorkOSProfile.OBJECT_FIELDS:
-            setattr(profile, field, profile_data[field])
-
-        return profile
-
-    def to_dict(self):
-        """Returns a dict representation of the WorkOSProfile.
-        
-        Returns:
-            dict: A dict representation of the WorkOSProfile
-        """
-        profile_dict = {}
-        for field in WorkOSProfile.OBJECT_FIELDS:
-            profile_dict[field] = getattr(self, field, None)
-
-        return profile_dict
+    ]

workos/sso.py

@@ -100,7 +100,7 @@ def get_profile(self, code):
             TOKEN_PATH, method=REQUEST_METHOD_POST, params=params
         )
 
-        return WorkOSProfile.construct_from_response(response)
+        return WorkOSProfile.construct_from_response(response["profile"])
 
     def promote_draft_connection(self, token):
         """Promote a Draft Connection