feat(native): Cube.py - support pre_aggregations_schema (#7130)

Author: ovr

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

@@ -53,6 +53,7 @@ class Configuration:
     repository_factory: Callable
     schema_version: Callable[[RequestContext], str]
     semantic_layer_sync: Callable
+    pre_aggregations_schema: Callable[[RequestContext], str]
 
     def __init__(self):
         self.schema_path = None
@@ -80,6 +81,7 @@ def __init__(self):
         self.repository_factory = None
         self.schema_version = None
         self.semantic_layer_sync = None
+        self.pre_aggregations_schema = None
 
     def set_schema_path(self, schema_path: str):
         self.schema_path = schema_path
@@ -150,5 +152,8 @@ def set_schema_version(self, schema_version: Callable[[RequestContext], str]):
     def set_semantic_layer_sync(self, semantic_layer_sync: Callable):
         self.semantic_layer_sync = semantic_layer_sync
 
+    def set_pre_aggregations_schema(self, pre_aggregations_schema: Callable[[RequestContext], str]):
+        self.pre_aggregations_schema = pre_aggregations_schema
+
 
 settings = Configuration()

packages/cubejs-backend-native/src/python/cube_config.rs

@@ -47,6 +47,7 @@ impl CubeConfigPy {
         self.function_attr(config_module, "repository_factory")?;
         self.function_attr(config_module, "semantic_layer_sync")?;
         self.function_attr(config_module, "schema_version")?;
+        self.function_attr(config_module, "pre_aggregations_schema")?;
 
         Ok(())
     }

packages/cubejs-server-core/src/core/server.ts

@@ -507,7 +507,7 @@ export class CubejsServerCore {
           ),
           externalDialectClass: this.options.externalDialectFactory && this.options.externalDialectFactory(context),
           schemaVersion: currentSchemaVersion,
-          preAggregationsSchema: this.preAggregationsSchema(context),
+          preAggregationsSchema: await this.preAggregationsSchema(context),
           context,
           allowJsDuplicatePropsInSchema: this.options.allowJsDuplicatePropsInSchema,
           allowNodeRequire: this.options.allowNodeRequire,

packages/cubejs-server-core/src/core/types.ts

@@ -127,7 +127,7 @@ export type ContextToOrchestratorIdFn = (context: RequestContext) => string | Pr
 
 export type OrchestratorOptionsFn = (context: RequestContext) => OrchestratorOptions;
 
-export type PreAggregationsSchemaFn = (context: RequestContext) => string;
+export type PreAggregationsSchemaFn = (context: RequestContext) => string | Promise<string>;
 
 // internal
 export type DriverOptions = {