Skip to content

Commit b44fc3b

Browse files
authored
Merge pull request #5484 from lucabem/master
Added functionality to allow filter kernels by Jupyter Enterprise Gat…
2 parents 6e9256b + 266e297 commit b44fc3b

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

notebook/gateway/managers.py

+12-2
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

0 commit comments

Comments
 (0)