Support named servers

Author: rcthomas

batchspawner/api.py

@@ -13,9 +13,14 @@ def post(self):
             # Previous jupyterhub, 0.9.4 and before.
             user = self.get_current_user()
         data = self.get_json_body()
+        if self.allow_named_servers:
+            server_name = data.pop("server_name", "")
+            spawner = user.spawners[server_name]
+        else:
+            spawner = user.spawner
         for key, value in data.items():
-            if hasattr(user.spawner, key):
-                setattr(user.spawner, key, value)
+            if hasattr(spawner, key):
+                setattr(spawner, key, value)
         self.finish(json.dumps({"message": "BatchSpawner data configured"}))
         self.set_status(201)
 

batchspawner/singleuser.py

@@ -8,18 +8,19 @@
 from jupyterhub.services.auth import HubAuth
 
 def main(argv=None):
+    server_name = os.environ.get('JUPYTERHUB_SERVER_NAME', '')
     port = random_port()
     hub_auth = HubAuth()
     hub_auth.client_ca = os.environ.get('JUPYTERHUB_SSL_CLIENT_CA', '')
     hub_auth.certfile = os.environ.get('JUPYTERHUB_SSL_CERTFILE', '')
     hub_auth.keyfile = os.environ.get('JUPYTERHUB_SSL_KEYFILE', '')
     hub_auth._api_request(method='POST',
                           url=url_path_join(hub_auth.api_url, 'batchspawner'),
-                          json={'port' : port})
+                          json={'server_name': server_name, 'port' : port})
 
     cmd_path = which(sys.argv[1])
     sys.argv = sys.argv[1:] + ['--port={}'.format(port)]
     run_path(cmd_path, run_name="__main__")
 
 if __name__ == "__main__":
-    main()
+    main()