Add lookup_key support (#278)

* Add `lookup_key` support

* format

Author: cmatheson

tests/test_organizations.py

@@ -185,6 +185,17 @@ def test_get_organization(self, mock_organization, mock_request_method):
 
         assert organization == mock_organization
 
+    def test_get_organization_by_lookup_key(
+        self, mock_organization, mock_request_method
+    ):
+        mock_request_method("get", mock_organization, 200)
+
+        organization = self.organizations.get_organization_by_lookup_key(
+            lookup_key="test"
+        )
+
+        assert organization == mock_organization
+
     def test_create_organization_with_domain_data(
         self, mock_organization, mock_request_method
     ):

tests/utils/fixtures/mock_organization.py

@@ -26,4 +26,5 @@ def __init__(self, id):
         "created_at",
         "updated_at",
         "domains",
+        "lookup_key",
     ]

workos/organizations.py

@@ -169,6 +169,21 @@ def get_organization(self, organization):
 
         return WorkOSOrganization.construct_from_response(response).to_dict()
 
+    def get_organization_by_lookup_key(self, lookup_key):
+        """Gets details for a single Organization by lookup key
+        Args:
+            lookup_key (str): Organization's lookup key
+        Returns:
+            dict: Organization response from WorkOS
+        """
+        response = self.request_helper.request(
+            "organizations/by_lookup_key/{lookup_key}".format(lookup_key=lookup_key),
+            method=REQUEST_METHOD_GET,
+            token=workos.api_key,
+        )
+
+        return WorkOSOrganization.construct_from_response(response).to_dict()
+
     def create_organization(self, organization, idempotency_key=None):
         """Create an organization
 
@@ -222,6 +237,7 @@ def update_organization(
         allow_profiles_outside_organization=None,
         domains=None,
         domain_data=None,
+        lookup_key=None,
     ):
         """Update an organization
 
@@ -262,6 +278,9 @@ def update_organization(
         if domain_data is not None:
             params["domain_data"] = domain_data
 
+        if lookup_key is not None:
+            params["lookup_key"] = lookup_key
+
         response = self.request_helper.request(
             "organizations/{organization}".format(organization=organization),
             method=REQUEST_METHOD_PUT,

workos/resources/organizations.py

@@ -16,6 +16,7 @@ class WorkOSOrganization(WorkOSBaseResource):
         "created_at",
         "updated_at",
         "domains",
+        "lookup_key",
     ]
 
     @classmethod