Skip to content

Commit 266e297

Browse files
committed
Changed _get_endpoint_for_user_filter to static and used in constructor class instead of each class method
1 parent f5322fc commit 266e297

File tree

1 file changed

+12
-15
lines changed

1 file changed

+12
-15
lines changed

notebook/gateway/managers.py

Lines changed: 12 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -498,15 +498,25 @@ def shutdown_all(self, now=False):
498498
self.remove_kernel(kernel_id)
499499

500500

501+
501502
class 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

Comments
 (0)