support ptr properly

screepers · May 25, 2016 · f4b6df1 · f4b6df1
1 parent b9f38fe
commit f4b6df1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/screeps_console/console.py b/screeps_console/console.py
@@ -113,7 +113,10 @@ def connect(self):
             logging.getLogger('websocket').addHandler(logging.NullHandler())
             websocket.enableTrace(False)
 
-        url = 'wss://screeps.com/socket/websocket'
+        if not self.ptr:
+            url = 'wss://screeps.com/socket/websocket'
+        else:
+            url = 'wss://screeps.com/ptr/socket/websocket'
 
         ws = websocket.WebSocketApp(url=url,
                                     on_message=self.on_message,