Python + Exports: Implement Go back

3 files changed, 16 insertions, 0 deletions

diff --git a/exports/exports.go b/exports/exports.go
index aa2b40e..331270f 100644
--- a/exports/exports.go
+++ b/exports/exports.go
@@ -215,6 +215,17 @@ func GetIdentifier(name *C.char) (*C.char, *C.char) {
 	return C.CString(identifier), C.CString("")
 }
 
+//export GoBack
+func GoBack(name *C.char) (*C.char) {
+	nameStr := C.GoString(name)
+	state, stateErr := GetVPNState(nameStr)
+	if stateErr != nil {
+		return C.CString(ErrorToString(stateErr))
+	}
+	goBackErr := state.GoBack()
+	return C.CString(ErrorToString(goBackErr))
+}
+
 //export SetIdentifier
 func SetIdentifier(name *C.char, identifier *C.char) *C.char {
 	nameStr := C.GoString(name)
diff --git a/wrappers/python/src/__init__.py b/wrappers/python/src/__init__.py
index 0451f61..a1f8b40 100644
--- a/wrappers/python/src/__init__.py
+++ b/wrappers/python/src/__init__.py
@@ -74,6 +74,7 @@ lib.GetDiscoOrganizations.argtypes, lib.GetDiscoOrganizations.restype = [
     c_char_p
 ], DataError
 lib.GetDiscoServers.argtypes, lib.GetDiscoServers.restype = [c_char_p], DataError
+lib.GoBack.argtypes, lib.GoBack.restype = [c_char_p], None
 lib.CancelOAuth.argtypes, lib.CancelOAuth.restype = [c_char_p], c_void_p
 lib.SetProfileID.argtypes, lib.SetProfileID.restype = [c_char_p, c_char_p], c_void_p
 lib.SetSecureLocation.argtypes, lib.SetSecureLocation.restype = [
diff --git a/wrappers/python/src/main.py b/wrappers/python/src/main.py
index 4868fca..84be0ab 100644
--- a/wrappers/python/src/main.py
+++ b/wrappers/python/src/main.py
@@ -134,6 +134,10 @@ class EduVPN(object):
         self.location_event = threading.Event()
         return self.get_config(url, lib.GetConfigSecureInternet, force_tcp)
 
+    def go_back(self) -> None:
+        # Ignore the error
+        self.go_function(lib.GoBack)
+
     def set_connected(self) -> None:
         connect_err = self.go_function(lib.SetConnected)