Add mode selection for ogClient

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