1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
from . import lib, VPNStateChange, GetDataError, GetPtrString
from ctypes import *
from enum import Enum
class StateType(Enum):
Enter = 1
Leave = 2
# Registers the python app with the Go code
# name: The name of the app to be registered
# url: The url of the server to connect to, FIXME: To be removed
# state_callback: The callback to trigger whenever a state is changed, FIXME: Remove whenever this wrapper has implemented callbacks using function decorations
def Register(name, config_directory, state_callback, debug):
name_bytes = name.encode("utf-8")
dir_bytes = config_directory.encode("utf-8")
ptr_err = lib.Register(name_bytes, dir_bytes, state_callback, debug)
err_string = GetPtrString(ptr_err)
return err_string
def Deregister():
lib.Deregister()
def GetDiscoServers():
servers, serversErr = GetDataError(lib.GetServersList())
organizations, organizationsErr = GetDataError(lib.GetOrganizationsList())
return servers, serversErr, organizations, organizationsErr
def Connect(url):
url_bytes = url.encode("utf-8")
data_error = lib.Connect(url_bytes)
return GetDataError(data_error)
# This has to be global as otherwise the callback is not alive
callback_function = None
def register_callback(eduvpn):
global callback_function
callback_function = VPNStateChange(
lambda old_state, new_state, data: eduvpn.callback(
old_state.decode(), new_state.decode(), data.decode()
)
)
def SendData(data):
lib.SendData(data.encode("utf-8"))
class EduVPN(object):
def __init__(self, name, config_directory):
self.event_handler = EventHandler()
self.name = name
self.config_directory = config_directory
register_callback(self)
def __del__(self):
# It could be that lib is garbage collected already
if lib:
Deregister()
def register(self, debug=False) -> bool:
return Register(self.name, self.config_directory, callback_function, debug) == ""
def get_disco(self):
return GetDiscoServers()
def connect(self, url):
return Connect(url)
@property
def event(self):
return self.event_handler
def callback(self, old_state, new_state, data):
self.event.run(old_state, new_state, data)
def send_data(self, data):
return SendData(data)
class EventHandler(object):
def __init__(self):
self.handlers = {}
def on(self, state, state_type):
def wrapped_f(func):
if (state, state_type) not in self.handlers:
self.handlers[(state, state_type)] = []
self.handlers[(state, state_type)].append(func)
return func
return wrapped_f
def run_state(self, state, state_type, data):
if (state, state_type) not in self.handlers:
return
for func in self.handlers[(state, state_type)]:
func(data)
def run(self, old_state, new_state, data):
if old_state == new_state:
return
self.run_state(old_state, StateType.Leave, data)
self.run_state(new_state, StateType.Enter, data)
|
