Client + Exports: Separate cleanup from disconnect

1 files changed, 14 insertions, 2 deletions

diff --git a/wrappers/python/eduvpn_common/main.py b/wrappers/python/eduvpn_common/main.py
index 4204239..7e96d6a 100644
--- a/wrappers/python/eduvpn_common/main.py
+++ b/wrappers/python/eduvpn_common/main.py
@@ -332,7 +332,19 @@ class EduVPN(object):
         if connecting_err:
             raise connecting_err
 
-    def set_disconnected(self, cleanup: bool = True, tokens: Optional[Token] = None) -> None:
+    def cleanup(self, tokens: Optional[Token] = None) -> None:
+        """Cleanup the vpn connection
+
+        :param tokens: Optional[Token]  (Default value = None): The OAuth tokens if available
+
+        :raises WrappedError: An error by the Go library
+        """
+        cleanup_err = self.go_function(self.lib.Cleanup, encode_tokens(tokens))
+
+        if cleanup_err:
+            raise cleanup_err
+
+    def set_disconnected(self, ) -> 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
@@ -340,7 +352,7 @@ class EduVPN(object):
 
         :raises WrappedError: An error by the Go library
         """
-        disconnect_err = self.go_function(self.lib.SetDisconnected, cleanup, encode_tokens(tokens))
+        disconnect_err = self.go_function(self.lib.SetDisconnected)
 
         if disconnect_err:
             raise disconnect_err