Client + Exports: Forward tokens for /disconnect

1 files changed, 3 insertions, 2 deletions

diff --git a/wrappers/python/eduvpn_common/main.py b/wrappers/python/eduvpn_common/main.py
index 07e3b6d..4204239 100644
--- a/wrappers/python/eduvpn_common/main.py
+++ b/wrappers/python/eduvpn_common/main.py
@@ -332,14 +332,15 @@ class EduVPN(object):
         if connecting_err:
             raise connecting_err
 
-    def set_disconnected(self, cleanup: bool = True) -> None:
+    def set_disconnected(self, cleanup: bool = True, tokens: Optional[Token] = None) -> None:
         """Set the FSM to disconnected
 
         :param cleanup: bool:  (Default value = True): Whether or not to call /disconnect to the server. This invalidates the OpenVPN/WireGuard configuration
+        :param tokens: Optional[Token]  (Default value = None): The OAuth tokens if available
 
         :raises WrappedError: An error by the Go library
         """
-        disconnect_err = self.go_function(self.lib.SetDisconnected, cleanup)
+        disconnect_err = self.go_function(self.lib.SetDisconnected, cleanup, encode_tokens(tokens))
 
         if disconnect_err:
             raise disconnect_err