Make multiple vnc sessions work

vishalg0wda · Mar 7, 2017 · 9f4d151 · 9f4d151
1 parent e5ae0c2
commit 9f4d151
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 7 deletions.
diff --git a/droid_config.json b/droid_config.json
@@ -4,7 +4,15 @@
     "droids": [
         {
             "port": "5554",
-            "avd": "nexus6-android7"
+            "avd": "nexus6-android7",
+            "display": "1",
+            "ws_port": "6080"
+        },
+        {
+            "port": "5556",
+            "avd": "nexus6-android7-2",
+            "display": "2",
+            "ws_port": "6081"
         }
     ]
 }
diff --git a/hedroid/worker/helpers.py b/hedroid/worker/helpers.py
@@ -35,6 +35,7 @@ def start(self):
                     proc.stop()
                 except:
                     pass
+            raise
 
     def stop(self):
         logger.debug('Stopping droid')
@@ -55,7 +56,7 @@ def __init__(self):
         self.dpi = 560
         self.dim = '720x1280x24'
         self.clip = '530x965+15+30'
-        self.ws_port = 6080
+        self.ws_port = '6080'
 
     def set_port(self, port):
         self.port = port
@@ -138,6 +139,8 @@ def setup(self):
                 self.start_droid(droid)
             except IndexError:
                 break
+            except:
+                pass
 
     def count(self):
         return len(self.initialised)

diff --git a/hedroid/worker/thrift_server.py b/hedroid/worker/thrift_server.py
@@ -1,3 +1,4 @@
+import os
 import atexit
 
 import requests
@@ -37,10 +38,9 @@ def setup(self):
         config = get_config()
         builder = DroidBuilder()
         for droid in config['droids']:
-            if droid['port']:
-                builder.set_port(droid['port'])
-            if droid['avd']:
-                builder.set_avd(droid['avd'])
+            for key, value in droid.iteritems():
+                func = getattr(builder, "set_{}".format(key))
+                func(value)
             self.coordinator.add_droid(builder.build())
         # Start all endpoints now
         self.coordinator.setup()

diff --git a/start_worker_thrift.py b/start_worker_thrift.py
@@ -1,5 +1,5 @@
 #! /usr/bin/env python
-from worker.thrift_server import start_server
+from hedroid.worker.thrift_server import start_server
 
 
 if __name__ == '__main__':