1010
1111from kubespawner import KubeSpawner # noqa: E402
1212
13-
1413# conda-store default page size
1514DEFAULT_PAGE_SIZE_LIMIT = 100
1615
16+
1717@gen .coroutine
1818def get_username_hook (spawner ):
1919 auth_state = yield spawner .user .get_auth_state ()
@@ -28,21 +28,19 @@ def get_username_hook(spawner):
2828
2929def get_total_records (url : str , token : str ) -> int :
3030 import urllib3
31-
31+
3232 http = urllib3 .PoolManager ()
33- response = http .request (
34- "GET" , url , headers = {"Authorization" : f"Bearer { token } " }
35- )
33+ response = http .request ("GET" , url , headers = {"Authorization" : f"Bearer { token } " })
3634 decoded_response = json .loads (response .data .decode ("UTF-8" ))
3735 return decoded_response .get ("count" , 0 )
3836
3937
4038def generate_paged_urls (base_url : str , total_records : int , page_size : int ) -> list [str ]:
4139 import math
42-
40+
4341 urls = []
4442 # pages starts at 1
45- for page in range (1 , math .ceil (total_records / page_size )+ 1 ):
43+ for page in range (1 , math .ceil (total_records / page_size ) + 1 ):
4644 urls .append (f"{ base_url } ?size={ page_size } &page={ page } " )
4745
4846 return urls
@@ -52,12 +50,15 @@ def generate_paged_urls(base_url: str, total_records: int, page_size: int) -> li
5250# adding tests in a traditional sense is not possible. See https://github.com/soapy1/nebari/tree/try-unit-test-spawner
5351# for a demo on one approach to adding test.
5452def get_conda_store_environments (user_info : dict ):
55- import urllib3
5653 import os
5754
55+ import urllib3
56+
5857 # Check for the environment variable `CONDA_STORE_API_PAGE_SIZE_LIMIT`. Fall
5958 # back to using the default page size limit if not set.
60- page_size = os .environ .get ("CONDA_STORE_API_PAGE_SIZE_LIMIT" , DEFAULT_PAGE_SIZE_LIMIT )
59+ page_size = os .environ .get (
60+ "CONDA_STORE_API_PAGE_SIZE_LIMIT" , DEFAULT_PAGE_SIZE_LIMIT
61+ )
6162
6263 external_url = z2jh .get_config ("custom.conda-store-service-name" )
6364 token = z2jh .get_config ("custom.conda-store-jhub-apps-token" )
0 commit comments