add scheduledRefreshTimeZones func support to python

cover with tests

Author: KSDaemon

packages/cubejs-backend-native/js/index.ts

@@ -354,6 +354,8 @@ export interface PyConfiguration {
   checkAuth?: (req: unknown, authorization: string) => Promise<{ 'security_context'?: unknown }>
   queryRewrite?: (query: unknown, ctx: unknown) => Promise<unknown>
   contextToApiScopes?: () => Promise<string[]>
+  scheduledRefreshContexts?: () => Promise<string[]>
+  scheduledRefreshTimeZones?: () => Promise<string[]>
   contextToRoles?: (ctx: unknown) => Promise<string[]>
 }
 

packages/cubejs-backend-native/python/cube/src/__init__.py

@@ -44,7 +44,7 @@ class Configuration:
     allow_js_duplicate_props_in_schema: bool
     jwt: Dict
     scheduled_refresh_timer: Any
-    scheduled_refresh_timezones: list[str]
+    scheduled_refresh_timezones: Union[Callable[[RequestContext], list[str]], list[str]]
     scheduled_refresh_concurrency: int
     scheduled_refresh_batch_size: int
     compiler_cache_size: int
@@ -93,7 +93,6 @@ def __init__(self):
         self.process_subscriptions_interval = None
         self.jwt = None
         self.scheduled_refresh_timer = None
-        self.scheduled_refresh_timezones = None
         self.scheduled_refresh_concurrency = None
         self.scheduled_refresh_batch_size = None
         self.compiler_cache_size = None
@@ -118,6 +117,7 @@ def __init__(self):
         self.query_rewrite = None
         self.extend_context = None
         self.scheduled_refresh_contexts = None
+        self.scheduled_refresh_timezones = None
         self.context_to_api_scopes = None
         self.repository_factory = None
         self.schema_version = None

packages/cubejs-backend-native/test/config.py

@@ -33,6 +33,34 @@ async def context_to_api_scopes():
     return ["meta", "data", "jobs"]
 
 
+@config
+async def scheduled_refresh_time_zones(ctx):
+    print("[python] scheduled_refresh_time_zones ctx=", ctx)
+    return ["Europe/Kyiv", "Antarctica/Troll", "Australia/Sydney"]
+
+
+@config
+async def scheduled_refresh_contexts(ctx):
+    print("[python] scheduled_refresh_contexts ctx=", ctx)
+    return [
+      {
+        "securityContext": {
+          "appid": 'test1', "u": { "prop1": "value1" }
+        }
+      },
+      {
+        "securityContext": {
+          "appid": 'test2', "u": { "prop1": "value2" }
+        }
+      },
+      {
+        "securityContext": {
+          "appid": 'test3', "u": { "prop1": "value3" }
+        }
+      },
+    ]
+
+
 @config
 def schema_version(ctx):
     print("[python] schema_version", ctx)

packages/cubejs-backend-native/test/python.test.ts

@@ -83,6 +83,38 @@ suite('Python Config', () => {
     expect(await config.contextToApiScopes()).toEqual(['meta', 'data', 'jobs']);
   });
 
+  test('scheduled_refresh_time_zones', async () => {
+    if (!config.scheduledRefreshTimeZones) {
+      throw new Error('scheduledRefreshTimeZones was not defined in config.py');
+    }
+
+    expect(await config.scheduledRefreshTimeZones()).toEqual(['Europe/Kyiv', 'Antarctica/Troll', 'Australia/Sydney']);
+  });
+
+  test('scheduled_refresh_contexts', async () => {
+    if (!config.scheduledRefreshContexts) {
+      throw new Error('scheduledRefreshContexts was not defined in config.py');
+    }
+
+    expect(await config.scheduledRefreshContexts()).toEqual([
+      {
+        securityContext: {
+          appid: 'test1', u: { prop1: 'value1' }
+        }
+      },
+      {
+        securityContext: {
+          appid: 'test2', u: { prop1: 'value2' }
+        }
+      },
+      {
+        securityContext: {
+          appid: 'test3', u: { prop1: 'value3' }
+        }
+      },
+    ]);
+  });
+
   test('repository factory', async () => {
     if (!config.repositoryFactory) {
       throw new Error('repositoryFactory was not defined in config.py');