diff options
Diffstat (limited to 'src/ogClient.py')
| -rw-r--r-- | src/ogClient.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/ogClient.py b/src/ogClient.py index 5c7706c..3e6a0e4 100644 --- a/src/ogClient.py +++ b/src/ogClient.py @@ -23,10 +23,14 @@ class State(Enum): FORCE_DISCONNECTED = 2 class ogClient: - def __init__(self, ip, port): + def __init__(self, ip, port, mode): + if mode not in {'virtual', 'linux'}: + raise ValueError('Mode not supported.') + self.ip = ip self.port = port - self.ogrest = ogRest() + self.mode = mode + self.ogrest = ogRest(self.mode) def get_socket(self): return self.sock |