Add wrapping functionality for getting a wireguard config

1 files changed, 18 insertions, 3 deletions

diff --git a/wrappers/python/main.py b/wrappers/python/main.py
index a248b71..6bd7d86 100644
--- a/wrappers/python/main.py
+++ b/wrappers/python/main.py
@@ -1,12 +1,27 @@
 import eduvpncommon.main as eduvpn
+import webbrowser
 
 
 _eduvpn = eduvpn.EduVPN("org.eduvpn.app.linux", "configs")
 
 
-@_eduvpn.event.on("REGISTERED", eduvpn.StateType.Enter)
-def registered(data):
-    print(f"REGISTERED PYTHON WITH DATA {data}")
+@_eduvpn.event.on("OAuthInitialized", eduvpn.StateType.Enter)
+def oauth_initialized(url):
+    print(f"Got OAUTH url {url}")
+    webbrowser.open(url)
+
+
+@_eduvpn.event.on("OAuthFinished", eduvpn.StateType.Enter)
+def oauth_finished(data):
+    print(f"Oauth finished {data}")
 
 
 _eduvpn.register()
+print(_eduvpn.get_disco())
+
+config, error = _eduvpn.connect("https://eduvpn.jwijenbergh.com")
+#
+if error != "":
+    print("Got connect error", error)
+
+print(config)