Also update session.get_logout_url

mthadley · mthadley · commit 6bcd718b55e8 · 2025-01-14T09:48:54.000-08:00
diff --git a/workos/session.py b/workos/session.py
@@ -159,7 +159,7 @@ def refresh(
                 authenticated=False, reason=str(e)
             )
 
-    def get_logout_url(self) -> str:
+    def get_logout_url(self, return_to: Optional[str] = None) -> str:
         auth_response = self.authenticate()
 
         if isinstance(auth_response, AuthenticateWithSessionCookieErrorResponse):
@@ -168,7 +168,8 @@ def get_logout_url(self) -> str:
             )
 
         result = self.user_management.get_logout_url(
-            session_id=auth_response.session_id
+            session_id=auth_response.session_id,
+            return_to=return_to,
         )
         return str(result)
 

Original file line number	Diff line number	Diff line change
`@@ -159,7 +159,7 @@ def refresh(`
`159`	`159`	`authenticated=False, reason=str(e)`
`160`	`160`	`)`
`161`	`161`
`162`		`- def get_logout_url(self) -> str:`
	`162`	`+ def get_logout_url(self, return_to: Optional[str] = None) -> str:`
`163`	`163`	`auth_response = self.authenticate()`
`164`	`164`
`165`	`165`	`if isinstance(auth_response, AuthenticateWithSessionCookieErrorResponse):`
`@@ -168,7 +168,8 @@ def get_logout_url(self) -> str:`
`168`	`168`	`)`
`169`	`169`
`170`	`170`	`result = self.user_management.get_logout_url(`
`171`		`- session_id=auth_response.session_id`
	`171`	`+ session_id=auth_response.session_id,`
	`172`	`+ return_to=return_to,`
`172`	`173`	`)`
`173`	`174`	`return str(result)`
`174`	`175`