@@ -498,15 +498,25 @@ def shutdown_all(self, now=False):
498498 self .remove_kernel (kernel_id )
499499
500500
501+
501502class GatewayKernelSpecManager (KernelSpecManager ):
502503
503504 def __init__ (self , ** kwargs ):
504505 super (GatewayKernelSpecManager , self ).__init__ (** kwargs )
505- self .base_endpoint = url_path_join (GatewayClient .instance ().url ,
506- GatewayClient .instance ().kernelspecs_endpoint )
506+ base_endpoint = url_path_join (GatewayClient .instance ().url ,
507+ GatewayClient .instance ().kernelspecs_endpoint )
508+
509+ self .base_endpoint = GatewayKernelSpecManager ._get_endpoint_for_user_filter (base_endpoint )
507510 self .base_resource_endpoint = url_path_join (GatewayClient .instance ().url ,
508511 GatewayClient .instance ().kernelspecs_resource_endpoint )
509512
513+ @staticmethod
514+ def _get_endpoint_for_user_filter (default_endpoint ):
515+ kernel_user = os .environ .get ('KERNEL_USERNAME' )
516+ if kernel_user :
517+ return '?user=' .join ([default_endpoint , kernel_user ])
518+ return default_endpoint
519+
510520 def _get_kernelspecs_endpoint_url (self , kernel_name = None ):
511521 """Builds a url for the kernels endpoint
512522
@@ -519,12 +529,6 @@ def _get_kernelspecs_endpoint_url(self, kernel_name=None):
519529
520530 return self .base_endpoint
521531
522- def _get_endpoint_for_user_filter (self , default_endpoint ):
523- kernel_user = os .environ .get ('KERNEL_USERNAME' )
524- if kernel_user :
525- return '?user=' .join ([default_endpoint , kernel_user ])
526- return default_endpoint
527-
528532 @gen .coroutine
529533 def get_all_specs (self ):
530534 fetched_kspecs = yield self .list_kernel_specs ()
@@ -549,9 +553,6 @@ def get_all_specs(self):
549553 def list_kernel_specs (self ):
550554 """Get a list of kernel specs."""
551555 kernel_spec_url = self ._get_kernelspecs_endpoint_url ()
552-
553- kernel_spec_url = self ._get_endpoint_for_user_filter (kernel_spec_url )
554-
555556 self .log .debug ("Request list kernel specs at: %s" , kernel_spec_url )
556557 response = yield gateway_request (kernel_spec_url , method = 'GET' )
557558 kernel_specs = json_decode (response .body )
@@ -567,9 +568,6 @@ def get_kernel_spec(self, kernel_name, **kwargs):
567568 The name of the kernel.
568569 """
569570 kernel_spec_url = self ._get_kernelspecs_endpoint_url (kernel_name = str (kernel_name ))
570-
571- kernel_spec_url = self ._get_endpoint_for_user_filter (kernel_spec_url )
572-
573571 self .log .debug ("Request kernel spec at: %s" % kernel_spec_url )
574572 try :
575573 response = yield gateway_request (kernel_spec_url , method = 'GET' )
@@ -598,7 +596,6 @@ def get_kernel_spec_resource(self, kernel_name, path):
598596 The name of the desired resource
599597 """
600598 kernel_spec_resource_url = url_path_join (self .base_resource_endpoint , str (kernel_name ), str (path ))
601- kernel_spec_resource_url = self ._get_endpoint_for_user_filter (kernel_spec_resource_url )
602599 self .log .debug ("Request kernel spec resource '{}' at: {}" .format (path , kernel_spec_resource_url ))
603600 try :
604601 response = yield gateway_request (kernel_spec_resource_url , method = 'GET' )
0 commit comments