relationalai 0.13.5__py3-none-any.whl → 1.0.0a2__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- relationalai/__init__.py +1 -256
- relationalai/config/__init__.py +56 -0
- relationalai/config/config.py +289 -0
- relationalai/config/config_fields.py +86 -0
- relationalai/config/connections/__init__.py +46 -0
- relationalai/config/connections/base.py +23 -0
- relationalai/config/connections/duckdb.py +29 -0
- relationalai/config/connections/snowflake.py +243 -0
- relationalai/config/external/__init__.py +17 -0
- relationalai/config/external/dbt_converter.py +101 -0
- relationalai/config/external/dbt_models.py +93 -0
- relationalai/config/external/snowflake_converter.py +41 -0
- relationalai/config/external/snowflake_models.py +85 -0
- relationalai/config/external/utils.py +19 -0
- relationalai/config/shims.py +1 -0
- relationalai/semantics/__init__.py +146 -22
- relationalai/semantics/backends/lqp/annotations.py +11 -0
- relationalai/semantics/backends/sql/sql_compiler.py +327 -0
- relationalai/semantics/frontend/base.py +1719 -0
- relationalai/semantics/frontend/core.py +179 -0
- relationalai/semantics/frontend/front_compiler.py +1316 -0
- relationalai/semantics/frontend/pprint.py +408 -0
- relationalai/semantics/metamodel/__init__.py +6 -40
- relationalai/semantics/metamodel/builtins.py +206 -772
- relationalai/semantics/metamodel/metamodel.py +465 -0
- relationalai/semantics/metamodel/metamodel_analyzer.py +519 -0
- relationalai/semantics/metamodel/pprint.py +414 -0
- relationalai/semantics/metamodel/rewriter.py +266 -0
- relationalai/semantics/metamodel/typer.py +1213 -0
- relationalai/semantics/std/__init__.py +60 -40
- relationalai/semantics/std/aggregates.py +148 -0
- relationalai/semantics/std/common.py +44 -0
- relationalai/semantics/std/constraints.py +37 -43
- relationalai/semantics/std/datetime.py +249 -135
- relationalai/semantics/std/decimals.py +45 -52
- relationalai/semantics/std/floats.py +13 -5
- relationalai/semantics/std/integers.py +26 -11
- relationalai/semantics/std/math.py +183 -112
- relationalai/semantics/std/numbers.py +86 -0
- relationalai/semantics/std/re.py +80 -62
- relationalai/semantics/std/strings.py +101 -46
- relationalai/shims/executor.py +179 -0
- relationalai/shims/helpers.py +126 -0
- relationalai/shims/hoister.py +221 -0
- relationalai/shims/mm2v0.py +1394 -0
- relationalai/tools/cli/__init__.py +6 -0
- relationalai/tools/cli/cli.py +90 -0
- relationalai/tools/cli/components/__init__.py +5 -0
- relationalai/tools/cli/components/progress_reader.py +1524 -0
- relationalai/tools/cli/components/utils.py +58 -0
- relationalai/tools/cli/config_template.py +45 -0
- relationalai/tools/cli/dev.py +19 -0
- relationalai/tools/debugger.py +289 -183
- relationalai/tools/typer_debugger.py +93 -0
- relationalai/util/dataclasses.py +43 -0
- relationalai/util/docutils.py +40 -0
- relationalai/util/error.py +199 -0
- relationalai/util/format.py +48 -109
- relationalai/util/naming.py +145 -0
- relationalai/util/python.py +35 -0
- relationalai/util/runtime.py +156 -0
- relationalai/util/schema.py +197 -0
- relationalai/util/source.py +185 -0
- relationalai/util/structures.py +163 -0
- relationalai/util/tracing.py +261 -0
- relationalai-1.0.0a2.dist-info/METADATA +44 -0
- relationalai-1.0.0a2.dist-info/RECORD +489 -0
- relationalai-1.0.0a2.dist-info/WHEEL +5 -0
- relationalai-1.0.0a2.dist-info/entry_points.txt +3 -0
- relationalai-1.0.0a2.dist-info/top_level.txt +2 -0
- v0/relationalai/__init__.py +216 -0
- v0/relationalai/clients/__init__.py +5 -0
- v0/relationalai/clients/azure.py +477 -0
- v0/relationalai/clients/client.py +912 -0
- v0/relationalai/clients/config.py +673 -0
- v0/relationalai/clients/direct_access_client.py +118 -0
- v0/relationalai/clients/hash_util.py +31 -0
- v0/relationalai/clients/local.py +571 -0
- v0/relationalai/clients/profile_polling.py +73 -0
- v0/relationalai/clients/result_helpers.py +420 -0
- v0/relationalai/clients/snowflake.py +3869 -0
- v0/relationalai/clients/types.py +113 -0
- v0/relationalai/clients/use_index_poller.py +980 -0
- v0/relationalai/clients/util.py +356 -0
- v0/relationalai/debugging.py +389 -0
- v0/relationalai/dsl.py +1749 -0
- v0/relationalai/early_access/builder/__init__.py +30 -0
- v0/relationalai/early_access/builder/builder/__init__.py +35 -0
- v0/relationalai/early_access/builder/snowflake/__init__.py +12 -0
- v0/relationalai/early_access/builder/std/__init__.py +25 -0
- v0/relationalai/early_access/builder/std/decimals/__init__.py +12 -0
- v0/relationalai/early_access/builder/std/integers/__init__.py +12 -0
- v0/relationalai/early_access/builder/std/math/__init__.py +12 -0
- v0/relationalai/early_access/builder/std/strings/__init__.py +14 -0
- v0/relationalai/early_access/devtools/__init__.py +12 -0
- v0/relationalai/early_access/devtools/benchmark_lqp/__init__.py +12 -0
- v0/relationalai/early_access/devtools/extract_lqp/__init__.py +12 -0
- v0/relationalai/early_access/dsl/adapters/orm/adapter_qb.py +427 -0
- v0/relationalai/early_access/dsl/adapters/orm/parser.py +636 -0
- v0/relationalai/early_access/dsl/adapters/owl/adapter.py +176 -0
- v0/relationalai/early_access/dsl/adapters/owl/parser.py +160 -0
- v0/relationalai/early_access/dsl/bindings/common.py +402 -0
- v0/relationalai/early_access/dsl/bindings/csv.py +170 -0
- v0/relationalai/early_access/dsl/bindings/legacy/binding_models.py +143 -0
- v0/relationalai/early_access/dsl/bindings/snowflake.py +64 -0
- v0/relationalai/early_access/dsl/codegen/binder.py +411 -0
- v0/relationalai/early_access/dsl/codegen/common.py +79 -0
- v0/relationalai/early_access/dsl/codegen/helpers.py +23 -0
- v0/relationalai/early_access/dsl/codegen/relations.py +700 -0
- v0/relationalai/early_access/dsl/codegen/weaver.py +417 -0
- v0/relationalai/early_access/dsl/core/builders/__init__.py +47 -0
- v0/relationalai/early_access/dsl/core/builders/logic.py +19 -0
- v0/relationalai/early_access/dsl/core/builders/scalar_constraint.py +11 -0
- v0/relationalai/early_access/dsl/core/constraints/predicate/atomic.py +455 -0
- v0/relationalai/early_access/dsl/core/constraints/predicate/universal.py +73 -0
- v0/relationalai/early_access/dsl/core/constraints/scalar.py +310 -0
- v0/relationalai/early_access/dsl/core/context.py +13 -0
- v0/relationalai/early_access/dsl/core/cset.py +132 -0
- v0/relationalai/early_access/dsl/core/exprs/__init__.py +116 -0
- v0/relationalai/early_access/dsl/core/exprs/relational.py +18 -0
- v0/relationalai/early_access/dsl/core/exprs/scalar.py +412 -0
- v0/relationalai/early_access/dsl/core/instances.py +44 -0
- v0/relationalai/early_access/dsl/core/logic/__init__.py +193 -0
- v0/relationalai/early_access/dsl/core/logic/aggregation.py +98 -0
- v0/relationalai/early_access/dsl/core/logic/exists.py +223 -0
- v0/relationalai/early_access/dsl/core/logic/helper.py +163 -0
- v0/relationalai/early_access/dsl/core/namespaces.py +32 -0
- v0/relationalai/early_access/dsl/core/relations.py +276 -0
- v0/relationalai/early_access/dsl/core/rules.py +112 -0
- v0/relationalai/early_access/dsl/core/std/__init__.py +45 -0
- v0/relationalai/early_access/dsl/core/temporal/recall.py +6 -0
- v0/relationalai/early_access/dsl/core/types/__init__.py +270 -0
- v0/relationalai/early_access/dsl/core/types/concepts.py +128 -0
- v0/relationalai/early_access/dsl/core/types/constrained/__init__.py +267 -0
- v0/relationalai/early_access/dsl/core/types/constrained/nominal.py +143 -0
- v0/relationalai/early_access/dsl/core/types/constrained/subtype.py +124 -0
- v0/relationalai/early_access/dsl/core/types/standard.py +92 -0
- v0/relationalai/early_access/dsl/core/types/unconstrained.py +50 -0
- v0/relationalai/early_access/dsl/core/types/variables.py +203 -0
- v0/relationalai/early_access/dsl/ir/compiler.py +318 -0
- v0/relationalai/early_access/dsl/ir/executor.py +260 -0
- v0/relationalai/early_access/dsl/ontologies/constraints.py +88 -0
- v0/relationalai/early_access/dsl/ontologies/export.py +30 -0
- v0/relationalai/early_access/dsl/ontologies/models.py +453 -0
- v0/relationalai/early_access/dsl/ontologies/python_printer.py +303 -0
- v0/relationalai/early_access/dsl/ontologies/readings.py +60 -0
- v0/relationalai/early_access/dsl/ontologies/relationships.py +322 -0
- v0/relationalai/early_access/dsl/ontologies/roles.py +87 -0
- v0/relationalai/early_access/dsl/ontologies/subtyping.py +55 -0
- v0/relationalai/early_access/dsl/orm/constraints.py +438 -0
- v0/relationalai/early_access/dsl/orm/measures/dimensions.py +200 -0
- v0/relationalai/early_access/dsl/orm/measures/initializer.py +16 -0
- v0/relationalai/early_access/dsl/orm/measures/measure_rules.py +275 -0
- v0/relationalai/early_access/dsl/orm/measures/measures.py +299 -0
- v0/relationalai/early_access/dsl/orm/measures/role_exprs.py +268 -0
- v0/relationalai/early_access/dsl/orm/models.py +256 -0
- v0/relationalai/early_access/dsl/orm/object_oriented_printer.py +344 -0
- v0/relationalai/early_access/dsl/orm/printer.py +469 -0
- v0/relationalai/early_access/dsl/orm/reasoners.py +480 -0
- v0/relationalai/early_access/dsl/orm/relations.py +19 -0
- v0/relationalai/early_access/dsl/orm/relationships.py +251 -0
- v0/relationalai/early_access/dsl/orm/types.py +42 -0
- v0/relationalai/early_access/dsl/orm/utils.py +79 -0
- v0/relationalai/early_access/dsl/orm/verb.py +204 -0
- v0/relationalai/early_access/dsl/physical_metadata/tables.py +133 -0
- v0/relationalai/early_access/dsl/relations.py +170 -0
- v0/relationalai/early_access/dsl/rulesets.py +69 -0
- v0/relationalai/early_access/dsl/schemas/__init__.py +450 -0
- v0/relationalai/early_access/dsl/schemas/builder.py +48 -0
- v0/relationalai/early_access/dsl/schemas/comp_names.py +51 -0
- v0/relationalai/early_access/dsl/schemas/components.py +203 -0
- v0/relationalai/early_access/dsl/schemas/contexts.py +156 -0
- v0/relationalai/early_access/dsl/schemas/exprs.py +89 -0
- v0/relationalai/early_access/dsl/schemas/fragments.py +464 -0
- v0/relationalai/early_access/dsl/serialization.py +79 -0
- v0/relationalai/early_access/dsl/serialize/exporter.py +163 -0
- v0/relationalai/early_access/dsl/snow/api.py +104 -0
- v0/relationalai/early_access/dsl/snow/common.py +76 -0
- v0/relationalai/early_access/dsl/state_mgmt/__init__.py +129 -0
- v0/relationalai/early_access/dsl/state_mgmt/state_charts.py +125 -0
- v0/relationalai/early_access/dsl/state_mgmt/transitions.py +130 -0
- v0/relationalai/early_access/dsl/types/__init__.py +40 -0
- v0/relationalai/early_access/dsl/types/concepts.py +12 -0
- v0/relationalai/early_access/dsl/types/entities.py +135 -0
- v0/relationalai/early_access/dsl/types/values.py +17 -0
- v0/relationalai/early_access/dsl/utils.py +102 -0
- v0/relationalai/early_access/graphs/__init__.py +13 -0
- v0/relationalai/early_access/lqp/__init__.py +12 -0
- v0/relationalai/early_access/lqp/compiler/__init__.py +12 -0
- v0/relationalai/early_access/lqp/constructors/__init__.py +18 -0
- v0/relationalai/early_access/lqp/executor/__init__.py +12 -0
- v0/relationalai/early_access/lqp/ir/__init__.py +12 -0
- v0/relationalai/early_access/lqp/passes/__init__.py +12 -0
- v0/relationalai/early_access/lqp/pragmas/__init__.py +12 -0
- v0/relationalai/early_access/lqp/primitives/__init__.py +12 -0
- v0/relationalai/early_access/lqp/types/__init__.py +12 -0
- v0/relationalai/early_access/lqp/utils/__init__.py +12 -0
- v0/relationalai/early_access/lqp/validators/__init__.py +12 -0
- v0/relationalai/early_access/metamodel/__init__.py +58 -0
- v0/relationalai/early_access/metamodel/builtins/__init__.py +12 -0
- v0/relationalai/early_access/metamodel/compiler/__init__.py +12 -0
- v0/relationalai/early_access/metamodel/dependency/__init__.py +12 -0
- v0/relationalai/early_access/metamodel/factory/__init__.py +17 -0
- v0/relationalai/early_access/metamodel/helpers/__init__.py +12 -0
- v0/relationalai/early_access/metamodel/ir/__init__.py +14 -0
- v0/relationalai/early_access/metamodel/rewrite/__init__.py +7 -0
- v0/relationalai/early_access/metamodel/typer/__init__.py +3 -0
- v0/relationalai/early_access/metamodel/typer/typer/__init__.py +12 -0
- v0/relationalai/early_access/metamodel/types/__init__.py +15 -0
- v0/relationalai/early_access/metamodel/util/__init__.py +15 -0
- v0/relationalai/early_access/metamodel/visitor/__init__.py +12 -0
- v0/relationalai/early_access/rel/__init__.py +12 -0
- v0/relationalai/early_access/rel/executor/__init__.py +12 -0
- v0/relationalai/early_access/rel/rel_utils/__init__.py +12 -0
- v0/relationalai/early_access/rel/rewrite/__init__.py +7 -0
- v0/relationalai/early_access/solvers/__init__.py +19 -0
- v0/relationalai/early_access/sql/__init__.py +11 -0
- v0/relationalai/early_access/sql/executor/__init__.py +3 -0
- v0/relationalai/early_access/sql/rewrite/__init__.py +3 -0
- v0/relationalai/early_access/tests/logging/__init__.py +12 -0
- v0/relationalai/early_access/tests/test_snapshot_base/__init__.py +12 -0
- v0/relationalai/early_access/tests/utils/__init__.py +12 -0
- v0/relationalai/environments/__init__.py +35 -0
- v0/relationalai/environments/base.py +381 -0
- v0/relationalai/environments/colab.py +14 -0
- v0/relationalai/environments/generic.py +71 -0
- v0/relationalai/environments/ipython.py +68 -0
- v0/relationalai/environments/jupyter.py +9 -0
- v0/relationalai/environments/snowbook.py +169 -0
- v0/relationalai/errors.py +2478 -0
- v0/relationalai/experimental/SF.py +38 -0
- v0/relationalai/experimental/inspect.py +47 -0
- v0/relationalai/experimental/pathfinder/__init__.py +158 -0
- v0/relationalai/experimental/pathfinder/api.py +160 -0
- v0/relationalai/experimental/pathfinder/automaton.py +584 -0
- v0/relationalai/experimental/pathfinder/bridge.py +226 -0
- v0/relationalai/experimental/pathfinder/compiler.py +416 -0
- v0/relationalai/experimental/pathfinder/datalog.py +214 -0
- v0/relationalai/experimental/pathfinder/diagnostics.py +56 -0
- v0/relationalai/experimental/pathfinder/filter.py +236 -0
- v0/relationalai/experimental/pathfinder/glushkov.py +439 -0
- v0/relationalai/experimental/pathfinder/options.py +265 -0
- v0/relationalai/experimental/pathfinder/rpq.py +344 -0
- v0/relationalai/experimental/pathfinder/transition.py +200 -0
- v0/relationalai/experimental/pathfinder/utils.py +26 -0
- v0/relationalai/experimental/paths/api.py +143 -0
- v0/relationalai/experimental/paths/benchmarks/grid_graph.py +37 -0
- v0/relationalai/experimental/paths/examples/basic_example.py +40 -0
- v0/relationalai/experimental/paths/examples/minimal_engine_warmup.py +3 -0
- v0/relationalai/experimental/paths/examples/movie_example.py +77 -0
- v0/relationalai/experimental/paths/examples/paths_benchmark.py +115 -0
- v0/relationalai/experimental/paths/examples/paths_example.py +116 -0
- v0/relationalai/experimental/paths/examples/pattern_to_automaton.py +28 -0
- v0/relationalai/experimental/paths/find_paths_via_automaton.py +85 -0
- v0/relationalai/experimental/paths/graph.py +185 -0
- v0/relationalai/experimental/paths/path_algorithms/find_paths.py +280 -0
- v0/relationalai/experimental/paths/path_algorithms/one_sided_ball_repetition.py +26 -0
- v0/relationalai/experimental/paths/path_algorithms/one_sided_ball_upto.py +111 -0
- v0/relationalai/experimental/paths/path_algorithms/single.py +59 -0
- v0/relationalai/experimental/paths/path_algorithms/two_sided_balls_repetition.py +39 -0
- v0/relationalai/experimental/paths/path_algorithms/two_sided_balls_upto.py +103 -0
- v0/relationalai/experimental/paths/path_algorithms/usp-old.py +130 -0
- v0/relationalai/experimental/paths/path_algorithms/usp-tuple.py +183 -0
- v0/relationalai/experimental/paths/path_algorithms/usp.py +150 -0
- v0/relationalai/experimental/paths/product_graph.py +93 -0
- v0/relationalai/experimental/paths/rpq/automaton.py +584 -0
- v0/relationalai/experimental/paths/rpq/diagnostics.py +56 -0
- v0/relationalai/experimental/paths/rpq/rpq.py +378 -0
- v0/relationalai/experimental/paths/tests/tests_limit_sp_max_length.py +90 -0
- v0/relationalai/experimental/paths/tests/tests_limit_sp_multiple.py +119 -0
- v0/relationalai/experimental/paths/tests/tests_limit_sp_single.py +104 -0
- v0/relationalai/experimental/paths/tests/tests_limit_walks_multiple.py +113 -0
- v0/relationalai/experimental/paths/tests/tests_limit_walks_single.py +149 -0
- v0/relationalai/experimental/paths/tests/tests_one_sided_ball_repetition_multiple.py +70 -0
- v0/relationalai/experimental/paths/tests/tests_one_sided_ball_repetition_single.py +64 -0
- v0/relationalai/experimental/paths/tests/tests_one_sided_ball_upto_multiple.py +115 -0
- v0/relationalai/experimental/paths/tests/tests_one_sided_ball_upto_single.py +75 -0
- v0/relationalai/experimental/paths/tests/tests_single_paths.py +152 -0
- v0/relationalai/experimental/paths/tests/tests_single_walks.py +208 -0
- v0/relationalai/experimental/paths/tests/tests_single_walks_undirected.py +297 -0
- v0/relationalai/experimental/paths/tests/tests_two_sided_balls_repetition_multiple.py +107 -0
- v0/relationalai/experimental/paths/tests/tests_two_sided_balls_repetition_single.py +76 -0
- v0/relationalai/experimental/paths/tests/tests_two_sided_balls_upto_multiple.py +76 -0
- v0/relationalai/experimental/paths/tests/tests_two_sided_balls_upto_single.py +110 -0
- v0/relationalai/experimental/paths/tests/tests_usp_nsp_multiple.py +229 -0
- v0/relationalai/experimental/paths/tests/tests_usp_nsp_single.py +108 -0
- v0/relationalai/experimental/paths/tree_agg.py +168 -0
- v0/relationalai/experimental/paths/utilities/iterators.py +27 -0
- v0/relationalai/experimental/paths/utilities/prefix_sum.py +91 -0
- v0/relationalai/experimental/solvers.py +1087 -0
- v0/relationalai/loaders/csv.py +195 -0
- v0/relationalai/loaders/loader.py +177 -0
- v0/relationalai/loaders/types.py +23 -0
- v0/relationalai/rel_emitter.py +373 -0
- v0/relationalai/rel_utils.py +185 -0
- v0/relationalai/semantics/__init__.py +29 -0
- v0/relationalai/semantics/devtools/benchmark_lqp.py +536 -0
- v0/relationalai/semantics/devtools/compilation_manager.py +294 -0
- v0/relationalai/semantics/devtools/extract_lqp.py +110 -0
- v0/relationalai/semantics/internal/internal.py +3785 -0
- v0/relationalai/semantics/internal/snowflake.py +325 -0
- v0/relationalai/semantics/lqp/builtins.py +16 -0
- v0/relationalai/semantics/lqp/compiler.py +22 -0
- v0/relationalai/semantics/lqp/constructors.py +68 -0
- v0/relationalai/semantics/lqp/executor.py +474 -0
- v0/relationalai/semantics/lqp/intrinsics.py +24 -0
- v0/relationalai/semantics/lqp/ir.py +124 -0
- v0/relationalai/semantics/lqp/model2lqp.py +877 -0
- v0/relationalai/semantics/lqp/passes.py +680 -0
- v0/relationalai/semantics/lqp/primitives.py +252 -0
- v0/relationalai/semantics/lqp/result_helpers.py +202 -0
- v0/relationalai/semantics/lqp/rewrite/__init__.py +18 -0
- v0/relationalai/semantics/lqp/rewrite/annotate_constraints.py +57 -0
- v0/relationalai/semantics/lqp/rewrite/cdc.py +216 -0
- v0/relationalai/semantics/lqp/rewrite/extract_common.py +338 -0
- v0/relationalai/semantics/lqp/rewrite/extract_keys.py +490 -0
- v0/relationalai/semantics/lqp/rewrite/function_annotations.py +114 -0
- v0/relationalai/semantics/lqp/rewrite/functional_dependencies.py +314 -0
- v0/relationalai/semantics/lqp/rewrite/quantify_vars.py +296 -0
- v0/relationalai/semantics/lqp/rewrite/splinter.py +76 -0
- v0/relationalai/semantics/lqp/types.py +101 -0
- v0/relationalai/semantics/lqp/utils.py +160 -0
- v0/relationalai/semantics/lqp/validators.py +57 -0
- v0/relationalai/semantics/metamodel/__init__.py +40 -0
- v0/relationalai/semantics/metamodel/builtins.py +776 -0
- v0/relationalai/semantics/metamodel/compiler.py +133 -0
- v0/relationalai/semantics/metamodel/dependency.py +862 -0
- v0/relationalai/semantics/metamodel/executor.py +61 -0
- v0/relationalai/semantics/metamodel/factory.py +287 -0
- v0/relationalai/semantics/metamodel/helpers.py +361 -0
- v0/relationalai/semantics/metamodel/ir.py +923 -0
- v0/relationalai/semantics/metamodel/rewrite/__init__.py +7 -0
- v0/relationalai/semantics/metamodel/rewrite/discharge_constraints.py +39 -0
- v0/relationalai/semantics/metamodel/rewrite/dnf_union_splitter.py +210 -0
- v0/relationalai/semantics/metamodel/rewrite/extract_nested_logicals.py +78 -0
- v0/relationalai/semantics/metamodel/rewrite/flatten.py +554 -0
- v0/relationalai/semantics/metamodel/rewrite/format_outputs.py +165 -0
- v0/relationalai/semantics/metamodel/typer/checker.py +353 -0
- v0/relationalai/semantics/metamodel/typer/typer.py +1395 -0
- v0/relationalai/semantics/metamodel/util.py +505 -0
- v0/relationalai/semantics/metamodel/visitor.py +944 -0
- v0/relationalai/semantics/reasoners/__init__.py +10 -0
- v0/relationalai/semantics/reasoners/graph/__init__.py +37 -0
- v0/relationalai/semantics/reasoners/graph/core.py +9019 -0
- v0/relationalai/semantics/reasoners/optimization/__init__.py +68 -0
- v0/relationalai/semantics/reasoners/optimization/common.py +88 -0
- v0/relationalai/semantics/reasoners/optimization/solvers_dev.py +568 -0
- v0/relationalai/semantics/reasoners/optimization/solvers_pb.py +1163 -0
- v0/relationalai/semantics/rel/builtins.py +40 -0
- v0/relationalai/semantics/rel/compiler.py +989 -0
- v0/relationalai/semantics/rel/executor.py +359 -0
- v0/relationalai/semantics/rel/rel.py +482 -0
- v0/relationalai/semantics/rel/rel_utils.py +276 -0
- v0/relationalai/semantics/snowflake/__init__.py +3 -0
- v0/relationalai/semantics/sql/compiler.py +2503 -0
- v0/relationalai/semantics/sql/executor/duck_db.py +52 -0
- v0/relationalai/semantics/sql/executor/result_helpers.py +64 -0
- v0/relationalai/semantics/sql/executor/snowflake.py +145 -0
- v0/relationalai/semantics/sql/rewrite/denormalize.py +222 -0
- v0/relationalai/semantics/sql/rewrite/double_negation.py +49 -0
- v0/relationalai/semantics/sql/rewrite/recursive_union.py +127 -0
- v0/relationalai/semantics/sql/rewrite/sort_output_query.py +246 -0
- v0/relationalai/semantics/sql/sql.py +504 -0
- v0/relationalai/semantics/std/__init__.py +54 -0
- v0/relationalai/semantics/std/constraints.py +43 -0
- v0/relationalai/semantics/std/datetime.py +363 -0
- v0/relationalai/semantics/std/decimals.py +62 -0
- v0/relationalai/semantics/std/floats.py +7 -0
- v0/relationalai/semantics/std/integers.py +22 -0
- v0/relationalai/semantics/std/math.py +141 -0
- v0/relationalai/semantics/std/pragmas.py +11 -0
- v0/relationalai/semantics/std/re.py +83 -0
- v0/relationalai/semantics/std/std.py +14 -0
- v0/relationalai/semantics/std/strings.py +63 -0
- v0/relationalai/semantics/tests/__init__.py +0 -0
- v0/relationalai/semantics/tests/test_snapshot_abstract.py +143 -0
- v0/relationalai/semantics/tests/test_snapshot_base.py +9 -0
- v0/relationalai/semantics/tests/utils.py +46 -0
- v0/relationalai/std/__init__.py +70 -0
- v0/relationalai/tools/__init__.py +0 -0
- v0/relationalai/tools/cli.py +1940 -0
- v0/relationalai/tools/cli_controls.py +1826 -0
- v0/relationalai/tools/cli_helpers.py +390 -0
- v0/relationalai/tools/debugger.py +183 -0
- v0/relationalai/tools/debugger_client.py +109 -0
- v0/relationalai/tools/debugger_server.py +302 -0
- v0/relationalai/tools/dev.py +685 -0
- v0/relationalai/tools/qb_debugger.py +425 -0
- v0/relationalai/util/clean_up_databases.py +95 -0
- v0/relationalai/util/format.py +123 -0
- v0/relationalai/util/list_databases.py +9 -0
- v0/relationalai/util/otel_configuration.py +25 -0
- v0/relationalai/util/otel_handler.py +484 -0
- v0/relationalai/util/snowflake_handler.py +88 -0
- v0/relationalai/util/span_format_test.py +43 -0
- v0/relationalai/util/span_tracker.py +207 -0
- v0/relationalai/util/spans_file_handler.py +72 -0
- v0/relationalai/util/tracing_handler.py +34 -0
- frontend/debugger/dist/.gitignore +0 -2
- frontend/debugger/dist/assets/favicon-Dy0ZgA6N.png +0 -0
- frontend/debugger/dist/assets/index-Cssla-O7.js +0 -208
- frontend/debugger/dist/assets/index-DlHsYx1V.css +0 -9
- frontend/debugger/dist/index.html +0 -17
- relationalai/clients/__init__.py +0 -18
- relationalai/clients/client.py +0 -946
- relationalai/clients/config.py +0 -673
- relationalai/clients/direct_access_client.py +0 -118
- relationalai/clients/exec_txn_poller.py +0 -153
- relationalai/clients/hash_util.py +0 -31
- relationalai/clients/local.py +0 -594
- relationalai/clients/profile_polling.py +0 -73
- relationalai/clients/resources/__init__.py +0 -8
- relationalai/clients/resources/azure/azure.py +0 -502
- relationalai/clients/resources/snowflake/__init__.py +0 -20
- relationalai/clients/resources/snowflake/cli_resources.py +0 -98
- relationalai/clients/resources/snowflake/direct_access_resources.py +0 -739
- relationalai/clients/resources/snowflake/engine_service.py +0 -381
- relationalai/clients/resources/snowflake/engine_state_handlers.py +0 -315
- relationalai/clients/resources/snowflake/error_handlers.py +0 -240
- relationalai/clients/resources/snowflake/export_procedure.py.jinja +0 -249
- relationalai/clients/resources/snowflake/resources_factory.py +0 -99
- relationalai/clients/resources/snowflake/snowflake.py +0 -3193
- relationalai/clients/resources/snowflake/use_index_poller.py +0 -1019
- relationalai/clients/resources/snowflake/use_index_resources.py +0 -188
- relationalai/clients/resources/snowflake/util.py +0 -387
- relationalai/clients/result_helpers.py +0 -420
- relationalai/clients/types.py +0 -118
- relationalai/clients/util.py +0 -356
- relationalai/debugging.py +0 -389
- relationalai/dsl.py +0 -1749
- relationalai/early_access/builder/__init__.py +0 -30
- relationalai/early_access/builder/builder/__init__.py +0 -35
- relationalai/early_access/builder/snowflake/__init__.py +0 -12
- relationalai/early_access/builder/std/__init__.py +0 -25
- relationalai/early_access/builder/std/decimals/__init__.py +0 -12
- relationalai/early_access/builder/std/integers/__init__.py +0 -12
- relationalai/early_access/builder/std/math/__init__.py +0 -12
- relationalai/early_access/builder/std/strings/__init__.py +0 -14
- relationalai/early_access/devtools/__init__.py +0 -12
- relationalai/early_access/devtools/benchmark_lqp/__init__.py +0 -12
- relationalai/early_access/devtools/extract_lqp/__init__.py +0 -12
- relationalai/early_access/dsl/adapters/orm/adapter_qb.py +0 -427
- relationalai/early_access/dsl/adapters/orm/parser.py +0 -636
- relationalai/early_access/dsl/adapters/owl/adapter.py +0 -176
- relationalai/early_access/dsl/adapters/owl/parser.py +0 -160
- relationalai/early_access/dsl/bindings/common.py +0 -402
- relationalai/early_access/dsl/bindings/csv.py +0 -170
- relationalai/early_access/dsl/bindings/legacy/binding_models.py +0 -143
- relationalai/early_access/dsl/bindings/snowflake.py +0 -64
- relationalai/early_access/dsl/codegen/binder.py +0 -411
- relationalai/early_access/dsl/codegen/common.py +0 -79
- relationalai/early_access/dsl/codegen/helpers.py +0 -23
- relationalai/early_access/dsl/codegen/relations.py +0 -700
- relationalai/early_access/dsl/codegen/weaver.py +0 -417
- relationalai/early_access/dsl/core/builders/__init__.py +0 -47
- relationalai/early_access/dsl/core/builders/logic.py +0 -19
- relationalai/early_access/dsl/core/builders/scalar_constraint.py +0 -11
- relationalai/early_access/dsl/core/constraints/predicate/atomic.py +0 -455
- relationalai/early_access/dsl/core/constraints/predicate/universal.py +0 -73
- relationalai/early_access/dsl/core/constraints/scalar.py +0 -310
- relationalai/early_access/dsl/core/context.py +0 -13
- relationalai/early_access/dsl/core/cset.py +0 -132
- relationalai/early_access/dsl/core/exprs/__init__.py +0 -116
- relationalai/early_access/dsl/core/exprs/relational.py +0 -18
- relationalai/early_access/dsl/core/exprs/scalar.py +0 -412
- relationalai/early_access/dsl/core/instances.py +0 -44
- relationalai/early_access/dsl/core/logic/__init__.py +0 -193
- relationalai/early_access/dsl/core/logic/aggregation.py +0 -98
- relationalai/early_access/dsl/core/logic/exists.py +0 -223
- relationalai/early_access/dsl/core/logic/helper.py +0 -163
- relationalai/early_access/dsl/core/namespaces.py +0 -32
- relationalai/early_access/dsl/core/relations.py +0 -276
- relationalai/early_access/dsl/core/rules.py +0 -112
- relationalai/early_access/dsl/core/std/__init__.py +0 -45
- relationalai/early_access/dsl/core/temporal/recall.py +0 -6
- relationalai/early_access/dsl/core/types/__init__.py +0 -270
- relationalai/early_access/dsl/core/types/concepts.py +0 -128
- relationalai/early_access/dsl/core/types/constrained/__init__.py +0 -267
- relationalai/early_access/dsl/core/types/constrained/nominal.py +0 -143
- relationalai/early_access/dsl/core/types/constrained/subtype.py +0 -124
- relationalai/early_access/dsl/core/types/standard.py +0 -92
- relationalai/early_access/dsl/core/types/unconstrained.py +0 -50
- relationalai/early_access/dsl/core/types/variables.py +0 -203
- relationalai/early_access/dsl/ir/compiler.py +0 -318
- relationalai/early_access/dsl/ir/executor.py +0 -260
- relationalai/early_access/dsl/ontologies/constraints.py +0 -88
- relationalai/early_access/dsl/ontologies/export.py +0 -30
- relationalai/early_access/dsl/ontologies/models.py +0 -453
- relationalai/early_access/dsl/ontologies/python_printer.py +0 -303
- relationalai/early_access/dsl/ontologies/readings.py +0 -60
- relationalai/early_access/dsl/ontologies/relationships.py +0 -322
- relationalai/early_access/dsl/ontologies/roles.py +0 -87
- relationalai/early_access/dsl/ontologies/subtyping.py +0 -55
- relationalai/early_access/dsl/orm/constraints.py +0 -438
- relationalai/early_access/dsl/orm/measures/dimensions.py +0 -200
- relationalai/early_access/dsl/orm/measures/initializer.py +0 -16
- relationalai/early_access/dsl/orm/measures/measure_rules.py +0 -275
- relationalai/early_access/dsl/orm/measures/measures.py +0 -299
- relationalai/early_access/dsl/orm/measures/role_exprs.py +0 -268
- relationalai/early_access/dsl/orm/models.py +0 -256
- relationalai/early_access/dsl/orm/object_oriented_printer.py +0 -344
- relationalai/early_access/dsl/orm/printer.py +0 -469
- relationalai/early_access/dsl/orm/reasoners.py +0 -480
- relationalai/early_access/dsl/orm/relations.py +0 -19
- relationalai/early_access/dsl/orm/relationships.py +0 -251
- relationalai/early_access/dsl/orm/types.py +0 -42
- relationalai/early_access/dsl/orm/utils.py +0 -79
- relationalai/early_access/dsl/orm/verb.py +0 -204
- relationalai/early_access/dsl/physical_metadata/tables.py +0 -133
- relationalai/early_access/dsl/relations.py +0 -170
- relationalai/early_access/dsl/rulesets.py +0 -69
- relationalai/early_access/dsl/schemas/__init__.py +0 -450
- relationalai/early_access/dsl/schemas/builder.py +0 -48
- relationalai/early_access/dsl/schemas/comp_names.py +0 -51
- relationalai/early_access/dsl/schemas/components.py +0 -203
- relationalai/early_access/dsl/schemas/contexts.py +0 -156
- relationalai/early_access/dsl/schemas/exprs.py +0 -89
- relationalai/early_access/dsl/schemas/fragments.py +0 -464
- relationalai/early_access/dsl/serialization.py +0 -79
- relationalai/early_access/dsl/serialize/exporter.py +0 -163
- relationalai/early_access/dsl/snow/api.py +0 -105
- relationalai/early_access/dsl/snow/common.py +0 -76
- relationalai/early_access/dsl/state_mgmt/__init__.py +0 -129
- relationalai/early_access/dsl/state_mgmt/state_charts.py +0 -125
- relationalai/early_access/dsl/state_mgmt/transitions.py +0 -130
- relationalai/early_access/dsl/types/__init__.py +0 -40
- relationalai/early_access/dsl/types/concepts.py +0 -12
- relationalai/early_access/dsl/types/entities.py +0 -135
- relationalai/early_access/dsl/types/values.py +0 -17
- relationalai/early_access/dsl/utils.py +0 -102
- relationalai/early_access/graphs/__init__.py +0 -13
- relationalai/early_access/lqp/__init__.py +0 -12
- relationalai/early_access/lqp/compiler/__init__.py +0 -12
- relationalai/early_access/lqp/constructors/__init__.py +0 -18
- relationalai/early_access/lqp/executor/__init__.py +0 -12
- relationalai/early_access/lqp/ir/__init__.py +0 -12
- relationalai/early_access/lqp/passes/__init__.py +0 -12
- relationalai/early_access/lqp/pragmas/__init__.py +0 -12
- relationalai/early_access/lqp/primitives/__init__.py +0 -12
- relationalai/early_access/lqp/types/__init__.py +0 -12
- relationalai/early_access/lqp/utils/__init__.py +0 -12
- relationalai/early_access/lqp/validators/__init__.py +0 -12
- relationalai/early_access/metamodel/__init__.py +0 -58
- relationalai/early_access/metamodel/builtins/__init__.py +0 -12
- relationalai/early_access/metamodel/compiler/__init__.py +0 -12
- relationalai/early_access/metamodel/dependency/__init__.py +0 -12
- relationalai/early_access/metamodel/factory/__init__.py +0 -17
- relationalai/early_access/metamodel/helpers/__init__.py +0 -12
- relationalai/early_access/metamodel/ir/__init__.py +0 -14
- relationalai/early_access/metamodel/rewrite/__init__.py +0 -7
- relationalai/early_access/metamodel/typer/__init__.py +0 -3
- relationalai/early_access/metamodel/typer/typer/__init__.py +0 -12
- relationalai/early_access/metamodel/types/__init__.py +0 -15
- relationalai/early_access/metamodel/util/__init__.py +0 -15
- relationalai/early_access/metamodel/visitor/__init__.py +0 -12
- relationalai/early_access/rel/__init__.py +0 -12
- relationalai/early_access/rel/executor/__init__.py +0 -12
- relationalai/early_access/rel/rel_utils/__init__.py +0 -12
- relationalai/early_access/rel/rewrite/__init__.py +0 -7
- relationalai/early_access/solvers/__init__.py +0 -19
- relationalai/early_access/sql/__init__.py +0 -11
- relationalai/early_access/sql/executor/__init__.py +0 -3
- relationalai/early_access/sql/rewrite/__init__.py +0 -3
- relationalai/early_access/tests/logging/__init__.py +0 -12
- relationalai/early_access/tests/test_snapshot_base/__init__.py +0 -12
- relationalai/early_access/tests/utils/__init__.py +0 -12
- relationalai/environments/__init__.py +0 -35
- relationalai/environments/base.py +0 -381
- relationalai/environments/colab.py +0 -14
- relationalai/environments/generic.py +0 -71
- relationalai/environments/ipython.py +0 -68
- relationalai/environments/jupyter.py +0 -9
- relationalai/environments/snowbook.py +0 -169
- relationalai/errors.py +0 -2496
- relationalai/experimental/SF.py +0 -38
- relationalai/experimental/inspect.py +0 -47
- relationalai/experimental/pathfinder/__init__.py +0 -158
- relationalai/experimental/pathfinder/api.py +0 -160
- relationalai/experimental/pathfinder/automaton.py +0 -584
- relationalai/experimental/pathfinder/bridge.py +0 -226
- relationalai/experimental/pathfinder/compiler.py +0 -416
- relationalai/experimental/pathfinder/datalog.py +0 -214
- relationalai/experimental/pathfinder/diagnostics.py +0 -56
- relationalai/experimental/pathfinder/filter.py +0 -236
- relationalai/experimental/pathfinder/glushkov.py +0 -439
- relationalai/experimental/pathfinder/options.py +0 -265
- relationalai/experimental/pathfinder/pathfinder-v0.7.0.rel +0 -1951
- relationalai/experimental/pathfinder/rpq.py +0 -344
- relationalai/experimental/pathfinder/transition.py +0 -200
- relationalai/experimental/pathfinder/utils.py +0 -26
- relationalai/experimental/paths/README.md +0 -107
- relationalai/experimental/paths/api.py +0 -143
- relationalai/experimental/paths/benchmarks/grid_graph.py +0 -37
- relationalai/experimental/paths/code_organization.md +0 -2
- relationalai/experimental/paths/examples/Movies.ipynb +0 -16328
- relationalai/experimental/paths/examples/basic_example.py +0 -40
- relationalai/experimental/paths/examples/minimal_engine_warmup.py +0 -3
- relationalai/experimental/paths/examples/movie_example.py +0 -77
- relationalai/experimental/paths/examples/movies_data/actedin.csv +0 -193
- relationalai/experimental/paths/examples/movies_data/directed.csv +0 -45
- relationalai/experimental/paths/examples/movies_data/follows.csv +0 -7
- relationalai/experimental/paths/examples/movies_data/movies.csv +0 -39
- relationalai/experimental/paths/examples/movies_data/person.csv +0 -134
- relationalai/experimental/paths/examples/movies_data/produced.csv +0 -16
- relationalai/experimental/paths/examples/movies_data/ratings.csv +0 -10
- relationalai/experimental/paths/examples/movies_data/wrote.csv +0 -11
- relationalai/experimental/paths/examples/paths_benchmark.py +0 -115
- relationalai/experimental/paths/examples/paths_example.py +0 -116
- relationalai/experimental/paths/examples/pattern_to_automaton.py +0 -28
- relationalai/experimental/paths/find_paths_via_automaton.py +0 -85
- relationalai/experimental/paths/graph.py +0 -185
- relationalai/experimental/paths/path_algorithms/find_paths.py +0 -280
- relationalai/experimental/paths/path_algorithms/one_sided_ball_repetition.py +0 -26
- relationalai/experimental/paths/path_algorithms/one_sided_ball_upto.py +0 -111
- relationalai/experimental/paths/path_algorithms/single.py +0 -59
- relationalai/experimental/paths/path_algorithms/two_sided_balls_repetition.py +0 -39
- relationalai/experimental/paths/path_algorithms/two_sided_balls_upto.py +0 -103
- relationalai/experimental/paths/path_algorithms/usp-old.py +0 -130
- relationalai/experimental/paths/path_algorithms/usp-tuple.py +0 -183
- relationalai/experimental/paths/path_algorithms/usp.py +0 -150
- relationalai/experimental/paths/product_graph.py +0 -93
- relationalai/experimental/paths/rpq/automaton.py +0 -584
- relationalai/experimental/paths/rpq/diagnostics.py +0 -56
- relationalai/experimental/paths/rpq/rpq.py +0 -378
- relationalai/experimental/paths/tests/tests_limit_sp_max_length.py +0 -90
- relationalai/experimental/paths/tests/tests_limit_sp_multiple.py +0 -119
- relationalai/experimental/paths/tests/tests_limit_sp_single.py +0 -104
- relationalai/experimental/paths/tests/tests_limit_walks_multiple.py +0 -113
- relationalai/experimental/paths/tests/tests_limit_walks_single.py +0 -149
- relationalai/experimental/paths/tests/tests_one_sided_ball_repetition_multiple.py +0 -70
- relationalai/experimental/paths/tests/tests_one_sided_ball_repetition_single.py +0 -64
- relationalai/experimental/paths/tests/tests_one_sided_ball_upto_multiple.py +0 -115
- relationalai/experimental/paths/tests/tests_one_sided_ball_upto_single.py +0 -75
- relationalai/experimental/paths/tests/tests_single_paths.py +0 -152
- relationalai/experimental/paths/tests/tests_single_walks.py +0 -208
- relationalai/experimental/paths/tests/tests_single_walks_undirected.py +0 -297
- relationalai/experimental/paths/tests/tests_two_sided_balls_repetition_multiple.py +0 -107
- relationalai/experimental/paths/tests/tests_two_sided_balls_repetition_single.py +0 -76
- relationalai/experimental/paths/tests/tests_two_sided_balls_upto_multiple.py +0 -76
- relationalai/experimental/paths/tests/tests_two_sided_balls_upto_single.py +0 -110
- relationalai/experimental/paths/tests/tests_usp_nsp_multiple.py +0 -229
- relationalai/experimental/paths/tests/tests_usp_nsp_single.py +0 -108
- relationalai/experimental/paths/tree_agg.py +0 -168
- relationalai/experimental/paths/utilities/iterators.py +0 -27
- relationalai/experimental/paths/utilities/prefix_sum.py +0 -91
- relationalai/experimental/solvers.py +0 -1095
- relationalai/loaders/csv.py +0 -195
- relationalai/loaders/loader.py +0 -177
- relationalai/loaders/types.py +0 -23
- relationalai/rel_emitter.py +0 -373
- relationalai/rel_utils.py +0 -185
- relationalai/semantics/designs/query_builder/identify_by.md +0 -106
- relationalai/semantics/devtools/benchmark_lqp.py +0 -535
- relationalai/semantics/devtools/compilation_manager.py +0 -294
- relationalai/semantics/devtools/extract_lqp.py +0 -110
- relationalai/semantics/internal/internal.py +0 -3785
- relationalai/semantics/internal/snowflake.py +0 -329
- relationalai/semantics/lqp/README.md +0 -34
- relationalai/semantics/lqp/algorithms.py +0 -173
- relationalai/semantics/lqp/builtins.py +0 -213
- relationalai/semantics/lqp/compiler.py +0 -22
- relationalai/semantics/lqp/constructors.py +0 -68
- relationalai/semantics/lqp/executor.py +0 -518
- relationalai/semantics/lqp/export_rewriter.py +0 -40
- relationalai/semantics/lqp/intrinsics.py +0 -24
- relationalai/semantics/lqp/ir.py +0 -150
- relationalai/semantics/lqp/model2lqp.py +0 -1056
- relationalai/semantics/lqp/passes.py +0 -38
- relationalai/semantics/lqp/primitives.py +0 -252
- relationalai/semantics/lqp/result_helpers.py +0 -266
- relationalai/semantics/lqp/rewrite/__init__.py +0 -32
- relationalai/semantics/lqp/rewrite/algorithm.py +0 -385
- relationalai/semantics/lqp/rewrite/annotate_constraints.py +0 -69
- relationalai/semantics/lqp/rewrite/cdc.py +0 -216
- relationalai/semantics/lqp/rewrite/constants_to_vars.py +0 -70
- relationalai/semantics/lqp/rewrite/deduplicate_vars.py +0 -104
- relationalai/semantics/lqp/rewrite/eliminate_data.py +0 -108
- relationalai/semantics/lqp/rewrite/extract_common.py +0 -340
- relationalai/semantics/lqp/rewrite/extract_keys.py +0 -577
- relationalai/semantics/lqp/rewrite/flatten_script.py +0 -301
- relationalai/semantics/lqp/rewrite/function_annotations.py +0 -114
- relationalai/semantics/lqp/rewrite/functional_dependencies.py +0 -348
- relationalai/semantics/lqp/rewrite/period_math.py +0 -77
- relationalai/semantics/lqp/rewrite/quantify_vars.py +0 -339
- relationalai/semantics/lqp/rewrite/splinter.py +0 -76
- relationalai/semantics/lqp/rewrite/unify_definitions.py +0 -323
- relationalai/semantics/lqp/types.py +0 -101
- relationalai/semantics/lqp/utils.py +0 -170
- relationalai/semantics/lqp/validators.py +0 -70
- relationalai/semantics/metamodel/compiler.py +0 -134
- relationalai/semantics/metamodel/dependency.py +0 -880
- relationalai/semantics/metamodel/executor.py +0 -78
- relationalai/semantics/metamodel/factory.py +0 -287
- relationalai/semantics/metamodel/helpers.py +0 -368
- relationalai/semantics/metamodel/ir.py +0 -924
- relationalai/semantics/metamodel/rewrite/__init__.py +0 -8
- relationalai/semantics/metamodel/rewrite/discharge_constraints.py +0 -39
- relationalai/semantics/metamodel/rewrite/dnf_union_splitter.py +0 -220
- relationalai/semantics/metamodel/rewrite/extract_nested_logicals.py +0 -78
- relationalai/semantics/metamodel/rewrite/flatten.py +0 -590
- relationalai/semantics/metamodel/rewrite/format_outputs.py +0 -256
- relationalai/semantics/metamodel/rewrite/handle_aggregations_and_ranks.py +0 -237
- relationalai/semantics/metamodel/typer/checker.py +0 -355
- relationalai/semantics/metamodel/typer/typer.py +0 -1396
- relationalai/semantics/metamodel/util.py +0 -506
- relationalai/semantics/metamodel/visitor.py +0 -945
- relationalai/semantics/reasoners/__init__.py +0 -10
- relationalai/semantics/reasoners/graph/README.md +0 -620
- relationalai/semantics/reasoners/graph/__init__.py +0 -37
- relationalai/semantics/reasoners/graph/core.py +0 -9019
- relationalai/semantics/reasoners/graph/design/beyond_demand_transform.md +0 -797
- relationalai/semantics/reasoners/graph/tests/README.md +0 -21
- relationalai/semantics/reasoners/optimization/__init__.py +0 -68
- relationalai/semantics/reasoners/optimization/common.py +0 -88
- relationalai/semantics/reasoners/optimization/solvers_dev.py +0 -568
- relationalai/semantics/reasoners/optimization/solvers_pb.py +0 -1407
- relationalai/semantics/rel/builtins.py +0 -40
- relationalai/semantics/rel/compiler.py +0 -994
- relationalai/semantics/rel/executor.py +0 -363
- relationalai/semantics/rel/rel.py +0 -482
- relationalai/semantics/rel/rel_utils.py +0 -276
- relationalai/semantics/snowflake/__init__.py +0 -3
- relationalai/semantics/sql/compiler.py +0 -2503
- relationalai/semantics/sql/executor/duck_db.py +0 -52
- relationalai/semantics/sql/executor/result_helpers.py +0 -64
- relationalai/semantics/sql/executor/snowflake.py +0 -149
- relationalai/semantics/sql/rewrite/denormalize.py +0 -222
- relationalai/semantics/sql/rewrite/double_negation.py +0 -49
- relationalai/semantics/sql/rewrite/recursive_union.py +0 -127
- relationalai/semantics/sql/rewrite/sort_output_query.py +0 -246
- relationalai/semantics/sql/sql.py +0 -504
- relationalai/semantics/std/pragmas.py +0 -11
- relationalai/semantics/std/std.py +0 -14
- relationalai/semantics/tests/lqp/algorithms.py +0 -345
- relationalai/semantics/tests/test_snapshot_abstract.py +0 -144
- relationalai/semantics/tests/test_snapshot_base.py +0 -9
- relationalai/semantics/tests/utils.py +0 -46
- relationalai/std/__init__.py +0 -70
- relationalai/tools/cli.py +0 -2089
- relationalai/tools/cli_controls.py +0 -1975
- relationalai/tools/cli_helpers.py +0 -802
- relationalai/tools/debugger_client.py +0 -109
- relationalai/tools/debugger_server.py +0 -302
- relationalai/tools/dev.py +0 -685
- relationalai/tools/notes +0 -7
- relationalai/tools/qb_debugger.py +0 -425
- relationalai/tools/txn_progress.py +0 -188
- relationalai/util/clean_up_databases.py +0 -95
- relationalai/util/list_databases.py +0 -9
- relationalai/util/otel_configuration.py +0 -26
- relationalai/util/otel_handler.py +0 -484
- relationalai/util/snowflake_handler.py +0 -88
- relationalai/util/span_format_test.py +0 -43
- relationalai/util/span_tracker.py +0 -207
- relationalai/util/spans_file_handler.py +0 -72
- relationalai/util/tracing_handler.py +0 -34
- relationalai-0.13.5.dist-info/METADATA +0 -74
- relationalai-0.13.5.dist-info/RECORD +0 -473
- relationalai-0.13.5.dist-info/WHEEL +0 -4
- relationalai-0.13.5.dist-info/entry_points.txt +0 -3
- relationalai-0.13.5.dist-info/licenses/LICENSE +0 -202
- relationalai_test_util/__init__.py +0 -4
- relationalai_test_util/fixtures.py +0 -233
- relationalai_test_util/snapshot.py +0 -252
- relationalai_test_util/traceback.py +0 -118
- /relationalai/{analysis → semantics/frontend}/__init__.py +0 -0
- /relationalai/{auth/__init__.py → semantics/metamodel/metamodel_compiler.py} +0 -0
- /relationalai/{early_access → shims}/__init__.py +0 -0
- {relationalai/early_access/dsl/adapters → v0/relationalai/analysis}/__init__.py +0 -0
- {relationalai → v0/relationalai}/analysis/mechanistic.py +0 -0
- {relationalai → v0/relationalai}/analysis/whynot.py +0 -0
- {relationalai/early_access/dsl/adapters/orm → v0/relationalai/auth}/__init__.py +0 -0
- {relationalai → v0/relationalai}/auth/jwt_generator.py +0 -0
- {relationalai → v0/relationalai}/auth/oauth_callback_server.py +0 -0
- {relationalai → v0/relationalai}/auth/token_handler.py +0 -0
- {relationalai → v0/relationalai}/auth/util.py +0 -0
- {relationalai/clients/resources/snowflake → v0/relationalai/clients}/cache_store.py +0 -0
- {relationalai → v0/relationalai}/compiler.py +0 -0
- {relationalai → v0/relationalai}/dependencies.py +0 -0
- {relationalai → v0/relationalai}/docutils.py +0 -0
- {relationalai/early_access/dsl/adapters/owl → v0/relationalai/early_access}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/__init__.py +0 -0
- {relationalai/early_access/dsl/bindings → v0/relationalai/early_access/dsl/adapters}/__init__.py +0 -0
- {relationalai/early_access/dsl/bindings/legacy → v0/relationalai/early_access/dsl/adapters/orm}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/adapters/orm/model.py +0 -0
- {relationalai/early_access/dsl/codegen → v0/relationalai/early_access/dsl/adapters/owl}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/adapters/owl/model.py +0 -0
- {relationalai/early_access/dsl/core/temporal → v0/relationalai/early_access/dsl/bindings}/__init__.py +0 -0
- {relationalai/early_access/dsl/ir → v0/relationalai/early_access/dsl/bindings/legacy}/__init__.py +0 -0
- {relationalai/early_access/dsl/ontologies → v0/relationalai/early_access/dsl/codegen}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/constants.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/core/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/core/constraints/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/core/constraints/predicate/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/core/stack.py +0 -0
- {relationalai/early_access/dsl/orm → v0/relationalai/early_access/dsl/core/temporal}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/core/utils.py +0 -0
- {relationalai/early_access/dsl/orm/measures → v0/relationalai/early_access/dsl/ir}/__init__.py +0 -0
- {relationalai/early_access/dsl/physical_metadata → v0/relationalai/early_access/dsl/ontologies}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/ontologies/raw_source.py +0 -0
- {relationalai/early_access/dsl/serialize → v0/relationalai/early_access/dsl/orm}/__init__.py +0 -0
- {relationalai/early_access/dsl/snow → v0/relationalai/early_access/dsl/orm/measures}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/orm/reasoner_errors.py +0 -0
- {relationalai/loaders → v0/relationalai/early_access/dsl/physical_metadata}/__init__.py +0 -0
- {relationalai/semantics/tests → v0/relationalai/early_access/dsl/serialize}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/serialize/binding_model.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/serialize/model.py +0 -0
- {relationalai/semantics/tests/lqp → v0/relationalai/early_access/dsl/snow}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/tests/__init__.py +0 -0
- {relationalai → v0/relationalai}/environments/ci.py +0 -0
- {relationalai → v0/relationalai}/environments/hex.py +0 -0
- {relationalai → v0/relationalai}/environments/terminal.py +0 -0
- {relationalai → v0/relationalai}/experimental/__init__.py +0 -0
- {relationalai → v0/relationalai}/experimental/graphs.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/__init__.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/benchmarks/__init__.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/path_algorithms/__init__.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/rpq/__init__.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/rpq/filter.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/rpq/glushkov.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/rpq/transition.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/utilities/__init__.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/utilities/utilities.py +0 -0
- {relationalai/tools → v0/relationalai/loaders}/__init__.py +0 -0
- {relationalai → v0/relationalai}/metagen.py +0 -0
- {relationalai → v0/relationalai}/metamodel.py +0 -0
- {relationalai → v0/relationalai}/rel.py +0 -0
- {relationalai → v0/relationalai}/semantics/devtools/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/internal/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/internal/annotations.py +0 -0
- {relationalai → v0/relationalai}/semantics/lqp/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/lqp/pragmas.py +0 -0
- {relationalai → v0/relationalai}/semantics/metamodel/dataflow.py +0 -0
- {relationalai → v0/relationalai}/semantics/metamodel/typer/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/metamodel/types.py +0 -0
- {relationalai → v0/relationalai}/semantics/reasoners/experimental/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/rel/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/sql/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/sql/executor/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/sql/rewrite/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/tests/logging.py +0 -0
- {relationalai → v0/relationalai}/std/aggregates.py +0 -0
- {relationalai → v0/relationalai}/std/dates.py +0 -0
- {relationalai → v0/relationalai}/std/graphs.py +0 -0
- {relationalai → v0/relationalai}/std/inspect.py +0 -0
- {relationalai → v0/relationalai}/std/math.py +0 -0
- {relationalai → v0/relationalai}/std/re.py +0 -0
- {relationalai → v0/relationalai}/std/strings.py +0 -0
- {relationalai → v0/relationalai}/tools/cleanup_snapshots.py +0 -0
- {relationalai → v0/relationalai}/tools/constants.py +0 -0
- {relationalai → v0/relationalai}/tools/query_utils.py +0 -0
- {relationalai → v0/relationalai}/tools/snapshot_viewer.py +0 -0
- {relationalai → v0/relationalai}/util/__init__.py +0 -0
- {relationalai → v0/relationalai}/util/constants.py +0 -0
- {relationalai → v0/relationalai}/util/graph.py +0 -0
- {relationalai → v0/relationalai}/util/timeout.py +0 -0
relationalai/rel_emitter.py
DELETED
|
@@ -1,373 +0,0 @@
|
|
|
1
|
-
from __future__ import annotations
|
|
2
|
-
import json
|
|
3
|
-
from typing import Any, List, Set, cast
|
|
4
|
-
from datetime import datetime, date
|
|
5
|
-
import pandas as pd
|
|
6
|
-
from zoneinfo import ZoneInfo
|
|
7
|
-
|
|
8
|
-
from relationalai.errors import RAIException
|
|
9
|
-
from relationalai.rel_utils import RESERVED_WORDS, sanitize_identifier
|
|
10
|
-
|
|
11
|
-
from .metamodel import Builtins, Action, ActionType, Namer, Property, Type, Var, Task, Value
|
|
12
|
-
from . import metamodel as m
|
|
13
|
-
from . import compiler as c
|
|
14
|
-
import re
|
|
15
|
-
|
|
16
|
-
gather_vars = m.Utils.gather_vars
|
|
17
|
-
|
|
18
|
-
#--------------------------------------------------
|
|
19
|
-
# Emitter
|
|
20
|
-
#--------------------------------------------------
|
|
21
|
-
|
|
22
|
-
rel_infix = [">", "<", ">=", "<=", "=", "!=", "+", "-", "*", "/", "^", "~="]
|
|
23
|
-
rel_sanitize_re = re.compile(r'[^\w:\[\]\^" ,]|^(?=\d)', re.UNICODE)
|
|
24
|
-
unsafe_symbol_pattern = re.compile(r"[^a-zA-Z0-9_]", re.UNICODE)
|
|
25
|
-
|
|
26
|
-
def sanitize(input_string, is_rel_name_or_symbol = False):
|
|
27
|
-
# Replace non-alphanumeric characters with underscores
|
|
28
|
-
if is_rel_name_or_symbol and "[" in input_string:
|
|
29
|
-
string_parts = input_string.split('[', 1)
|
|
30
|
-
sanitized_name_or_symbol = sanitize_identifier(string_parts[0])
|
|
31
|
-
sanitized_rest = re.sub(rel_sanitize_re, "_", string_parts[1])
|
|
32
|
-
sanitized = f"{sanitized_name_or_symbol}[{sanitized_rest}"
|
|
33
|
-
else:
|
|
34
|
-
if "::" in input_string: # TODO: This is a temp solution to avoid sanitizing the namespace
|
|
35
|
-
sanitized = re.sub(rel_sanitize_re, "_", input_string)
|
|
36
|
-
else:
|
|
37
|
-
sanitized = sanitize_identifier(input_string)
|
|
38
|
-
|
|
39
|
-
# Check if the resulting string is a keyword and append an underscore if it is
|
|
40
|
-
if sanitized in RESERVED_WORDS:
|
|
41
|
-
sanitized += "_"
|
|
42
|
-
|
|
43
|
-
return sanitized
|
|
44
|
-
|
|
45
|
-
class Emitter(c.Emitter):
|
|
46
|
-
|
|
47
|
-
def __init__(self, config):
|
|
48
|
-
super().__init__()
|
|
49
|
-
self.namer = Namer()
|
|
50
|
-
self.stack:List[Set[Var]] = []
|
|
51
|
-
self.mapped = {}
|
|
52
|
-
self.config = config
|
|
53
|
-
|
|
54
|
-
def reset(self):
|
|
55
|
-
self.namer.reset()
|
|
56
|
-
self.stack = []
|
|
57
|
-
self.mapped = {}
|
|
58
|
-
|
|
59
|
-
#--------------------------------------------------
|
|
60
|
-
# Emit
|
|
61
|
-
#--------------------------------------------------
|
|
62
|
-
|
|
63
|
-
def emit(self, task: Task|Var):
|
|
64
|
-
self.mapped.clear()
|
|
65
|
-
code = ""
|
|
66
|
-
try:
|
|
67
|
-
if isinstance(task, Task):
|
|
68
|
-
code = getattr(self, task.behavior.value)(task)
|
|
69
|
-
elif isinstance(task, Var):
|
|
70
|
-
code = self.emit_var(task)
|
|
71
|
-
except Exception as e:
|
|
72
|
-
# print("EMIT FAILED:", e)
|
|
73
|
-
raise e
|
|
74
|
-
return code
|
|
75
|
-
|
|
76
|
-
#--------------------------------------------------
|
|
77
|
-
# Helpers
|
|
78
|
-
#--------------------------------------------------
|
|
79
|
-
|
|
80
|
-
def sanitize(self, input_string, is_rel_name = False):
|
|
81
|
-
return sanitize(input_string, is_rel_name)
|
|
82
|
-
|
|
83
|
-
#--------------------------------------------------
|
|
84
|
-
# Vars
|
|
85
|
-
#--------------------------------------------------
|
|
86
|
-
|
|
87
|
-
def emit_val(self, value: Value, var:Var|None=None):
|
|
88
|
-
if value is True:
|
|
89
|
-
return 'boolean_true'
|
|
90
|
-
if value is False:
|
|
91
|
-
return 'boolean_false'
|
|
92
|
-
if isinstance(value, list):
|
|
93
|
-
return f"{', '.join([self.emit_var(v) for v in value])}"
|
|
94
|
-
if isinstance(value, Task) and (value.items or not value.name):
|
|
95
|
-
return self.to_inline_relation(value)
|
|
96
|
-
if isinstance(value, Task) and value.name:
|
|
97
|
-
return f"{{{value.name}}}"
|
|
98
|
-
if isinstance(value, bytes):
|
|
99
|
-
if var and var.type.isa(Builtins.ValueType):
|
|
100
|
-
return f"^{self.sanitize(var.name).capitalize()}Type[0x{value.hex()}]"
|
|
101
|
-
return f"0x{value.hex()}"
|
|
102
|
-
if isinstance(value, Property):
|
|
103
|
-
return "{" + self.sanitize(value.name) + "}"
|
|
104
|
-
if isinstance(value, Type):
|
|
105
|
-
return "{" + self.sanitize(value.name, value.isa(Builtins.Relation)) + "}"
|
|
106
|
-
if isinstance(value, pd.Timestamp):
|
|
107
|
-
t = value.tz_localize('UTC')
|
|
108
|
-
return t.isoformat()
|
|
109
|
-
if isinstance(value, datetime):
|
|
110
|
-
t = value.astimezone(ZoneInfo('UTC'))
|
|
111
|
-
return t.isoformat(timespec='milliseconds')
|
|
112
|
-
if isinstance(value, date):
|
|
113
|
-
t = value
|
|
114
|
-
return t.isoformat()
|
|
115
|
-
if isinstance(value, str):
|
|
116
|
-
# % is the string interpolation character in rel
|
|
117
|
-
return json.dumps(value).replace("%", "\\%")
|
|
118
|
-
return json.dumps(value)
|
|
119
|
-
|
|
120
|
-
def emit_var(self, var: Var|Value):
|
|
121
|
-
if not isinstance(var, Var):
|
|
122
|
-
return self.emit_val(var)
|
|
123
|
-
if var in self.mapped:
|
|
124
|
-
return self.mapped[var]
|
|
125
|
-
if var.type.isa(Builtins.Symbol):
|
|
126
|
-
if unsafe_symbol_pattern.search(cast(str, var.value)):
|
|
127
|
-
return f":\"{var.value}\""
|
|
128
|
-
return f":{self.sanitize(var.value)}"
|
|
129
|
-
if var.value is not None:
|
|
130
|
-
return self.emit_val(var.value, var)
|
|
131
|
-
if var.name == "_":
|
|
132
|
-
return "_"
|
|
133
|
-
name = self.namer.get(var)
|
|
134
|
-
return f"_{self.sanitize(name)}"
|
|
135
|
-
|
|
136
|
-
def to_name(self, value: Value|Var|None) -> str:
|
|
137
|
-
if value is not None and hasattr(value, "name"):
|
|
138
|
-
name = getattr(value, "name") # @NOTE: Working around pyright
|
|
139
|
-
if name:
|
|
140
|
-
return name
|
|
141
|
-
|
|
142
|
-
raise RAIException(f"Unable to determine the appropriate name for value of type: {type(value)} ({value})")
|
|
143
|
-
|
|
144
|
-
#--------------------------------------------------
|
|
145
|
-
# Sequence
|
|
146
|
-
#--------------------------------------------------
|
|
147
|
-
|
|
148
|
-
def sequence_action(self, action: Action):
|
|
149
|
-
if action.entity.value == Builtins.RawCode:
|
|
150
|
-
return str([*action.bindings.values()][0].value)
|
|
151
|
-
elif isinstance(action.entity.value, Task):
|
|
152
|
-
return self.emit(action.entity.value)
|
|
153
|
-
else:
|
|
154
|
-
raise Exception(f"TODO: Rel emit for action type {action.action}")
|
|
155
|
-
|
|
156
|
-
def sequence(self, task: Task):
|
|
157
|
-
items = [ self.sequence_action(i) for i in task.items ]
|
|
158
|
-
return "\n\n".join(items)
|
|
159
|
-
|
|
160
|
-
#--------------------------------------------------
|
|
161
|
-
# Query helpers
|
|
162
|
-
#--------------------------------------------------
|
|
163
|
-
|
|
164
|
-
def to_relation(self, action: Action, vars: set, body:List[str], in_head = False):
|
|
165
|
-
root = cast(Type, action.entity.value)
|
|
166
|
-
rel_name = root.name
|
|
167
|
-
if not rel_name and isinstance(root, Task):
|
|
168
|
-
rel_name = f"T{root.id}"
|
|
169
|
-
elif not rel_name:
|
|
170
|
-
rel_name = f"R_{root.id}"
|
|
171
|
-
args = []
|
|
172
|
-
for var in action.bindings.values():
|
|
173
|
-
emitted = self.emit_var(var)
|
|
174
|
-
vars.add(var)
|
|
175
|
-
if in_head and emitted in args:
|
|
176
|
-
name = self.namer.get(var)
|
|
177
|
-
orig = emitted
|
|
178
|
-
emitted = self.sanitize(f"_{self.namer.get_safe_name(name)}")
|
|
179
|
-
body.append(f"{emitted} = {orig}")
|
|
180
|
-
elif in_head and var.value is not None and var.isa(Builtins.ValueType):
|
|
181
|
-
name = self.namer.get(var)
|
|
182
|
-
emitted = self.sanitize(f"_{self.namer.get_safe_name(name)}")
|
|
183
|
-
body.append(f"{emitted} = {self.emit_val(var.value, var)}")
|
|
184
|
-
elif in_head and isinstance(var.value, bool):
|
|
185
|
-
name = self.namer.get(var)
|
|
186
|
-
orig = emitted
|
|
187
|
-
emitted = self.sanitize(f"_{self.namer.get_safe_name(name)}")
|
|
188
|
-
body.append(f"{emitted} = {orig}")
|
|
189
|
-
args.append(emitted)
|
|
190
|
-
if rel_name in rel_infix:
|
|
191
|
-
if len(args) == 2:
|
|
192
|
-
return f"{args[0]} {rel_name} {args[1]}"
|
|
193
|
-
else:
|
|
194
|
-
return f"{args[2]} = {args[0]} {rel_name} {args[1]}"
|
|
195
|
-
rel_name = self.sanitize(rel_name, True)
|
|
196
|
-
if isinstance(root, Task) and root.isa(Builtins.Quantifier):
|
|
197
|
-
# TODO: handle quantifier grouping
|
|
198
|
-
args = args[1:]
|
|
199
|
-
rel_name = root.name.lower()
|
|
200
|
-
if root.isa(Builtins.Exists):
|
|
201
|
-
rel_name = "not empty"
|
|
202
|
-
if action.action == ActionType.Persist:
|
|
203
|
-
rel_name = f"insert[:{rel_name}]"
|
|
204
|
-
elif action.action == ActionType.Unpersist:
|
|
205
|
-
rel_name = f"delete[:{rel_name}]"
|
|
206
|
-
partial_appl = rel_name.find('[')
|
|
207
|
-
if partial_appl >= 0:
|
|
208
|
-
final_rel_name = rel_name[:partial_appl]
|
|
209
|
-
if len(args):
|
|
210
|
-
final_args = f"{rel_name[partial_appl+1:-1]}, {', '.join(args)}"
|
|
211
|
-
else:
|
|
212
|
-
final_args = ', '.join(args)
|
|
213
|
-
final = f"{final_rel_name}({final_args})"
|
|
214
|
-
else:
|
|
215
|
-
final = f"{rel_name}({', '.join(args)})"
|
|
216
|
-
return final
|
|
217
|
-
|
|
218
|
-
def to_return(self, action: Action, vars: set, body:List[str]):
|
|
219
|
-
args = []
|
|
220
|
-
for var in action.bindings.values():
|
|
221
|
-
emitted = self.emit_var(var)
|
|
222
|
-
vars.add(var)
|
|
223
|
-
if emitted in args:
|
|
224
|
-
name = self.namer.get(var)
|
|
225
|
-
orig = emitted
|
|
226
|
-
emitted = self.sanitize(f"_{self.namer.get_safe_name(name)}")
|
|
227
|
-
body.append(f"{emitted} = {orig}")
|
|
228
|
-
elif var.value is not None and var.isa(Builtins.ValueType):
|
|
229
|
-
name = self.namer.get(var)
|
|
230
|
-
emitted = self.sanitize(f"_{self.namer.get_safe_name(name)}")
|
|
231
|
-
body.append(f"{emitted} = {self.emit_val(var.value, var)}")
|
|
232
|
-
elif isinstance(var.value, bool):
|
|
233
|
-
name = self.namer.get(var)
|
|
234
|
-
orig = emitted
|
|
235
|
-
emitted = self.sanitize(f"_{self.namer.get_safe_name(name)}")
|
|
236
|
-
body.append(f"{emitted} = {orig}")
|
|
237
|
-
args.append(emitted)
|
|
238
|
-
return f"output({', '.join(args)})"
|
|
239
|
-
|
|
240
|
-
def to_set(self, action: Action, vars: set):
|
|
241
|
-
vals:Any = [v for v in action.bindings.values()]
|
|
242
|
-
mid_point = len(vals) // 2
|
|
243
|
-
vars_str = ", ".join([self.emit_var(i) for i in vals[mid_point:]])
|
|
244
|
-
rows = []
|
|
245
|
-
for ix in range(len(vals[0].value)):
|
|
246
|
-
row_vals = vals[:mid_point]
|
|
247
|
-
product_str = ",".join([self.emit_var(col.value[ix]) for col in row_vals])
|
|
248
|
-
if len(row_vals) > 1:
|
|
249
|
-
rows.append(f"({product_str})")
|
|
250
|
-
else:
|
|
251
|
-
rows.append(product_str)
|
|
252
|
-
row_str = "; ".join(rows)
|
|
253
|
-
return f"{{{row_str}}}({vars_str})"
|
|
254
|
-
|
|
255
|
-
def to_inline_set(self, action: Action, vars: set):
|
|
256
|
-
vals:Any = [v for v in action.bindings.values()]
|
|
257
|
-
rows = []
|
|
258
|
-
for ix in range(len(vals[0].value)):
|
|
259
|
-
row_vals = vals[:-1]
|
|
260
|
-
product_str = ",".join([self.emit_var(col.value[ix]) for col in row_vals])
|
|
261
|
-
if len(row_vals) > 1:
|
|
262
|
-
rows.append(f"({product_str})")
|
|
263
|
-
else:
|
|
264
|
-
rows.append(product_str)
|
|
265
|
-
row_str = "; ".join(rows)
|
|
266
|
-
return (vals[-1], f"{{{row_str}}}")
|
|
267
|
-
|
|
268
|
-
def to_inline_relation(self, task:Task, existing_vars=set()):
|
|
269
|
-
return f"{{{self.query(task, True)}}}"
|
|
270
|
-
|
|
271
|
-
#--------------------------------------------------
|
|
272
|
-
# Query
|
|
273
|
-
#--------------------------------------------------
|
|
274
|
-
|
|
275
|
-
def query(self, task: Task, inline = False):
|
|
276
|
-
supporting_rules = []
|
|
277
|
-
head = ""
|
|
278
|
-
body = []
|
|
279
|
-
body_vars = set()
|
|
280
|
-
head_vars = set()
|
|
281
|
-
|
|
282
|
-
task_vars = set(gather_vars(task.items))
|
|
283
|
-
self.stack.append(task_vars)
|
|
284
|
-
|
|
285
|
-
for i in task.items:
|
|
286
|
-
if i.action in [ActionType.Get, ActionType.Call]:
|
|
287
|
-
if i.entity.value == Builtins.RawData:
|
|
288
|
-
body.append(self.to_set(i, body_vars))
|
|
289
|
-
elif i.entity.value == Builtins.InlineRawData:
|
|
290
|
-
(v, data) = self.to_inline_set(i, body_vars)
|
|
291
|
-
self.mapped[v] = data
|
|
292
|
-
head_vars.add(v)
|
|
293
|
-
elif i.entity.value == Builtins.Install:
|
|
294
|
-
(item,) = i.bindings.values()
|
|
295
|
-
supporting_rules.append(f"declare {self.sanitize(self.to_name(item.value), True)}")
|
|
296
|
-
if isinstance(item.value, Type) and item.value.isa(Builtins.ValueType):
|
|
297
|
-
vtype = self.sanitize(item.value.name, True).capitalize() + "Type"
|
|
298
|
-
supporting_rules.append(f"value type {vtype} {{ UInt128 }}")
|
|
299
|
-
supporting_rules.append(f"@inline def pyrel_unwrap(y in {vtype}, x): ^{vtype}(x, y)")
|
|
300
|
-
supporting_rules.append(f"def ::std::common::uuid_string(x in {vtype}, y): exists((z) | pyrel_unwrap(x, z) and uuid_string(z, y))")
|
|
301
|
-
for op in ["<", ">"]:
|
|
302
|
-
supporting_rules.append(f"def ::std::common::({op})[x in {vtype}, y in {vtype}]: exists((a, b) | ^{vtype}(a, x) and ^{vtype}(b, y) and a {op} b)")
|
|
303
|
-
if self.config.get("compiler.use_monotype_operators", False):
|
|
304
|
-
supporting_rules.append(f"def ::std::monotype::({op})[x in {vtype}, y in {vtype}]: exists((a, b) | ^{vtype}(a, x) and ^{vtype}(b, y) and a {op} b)")
|
|
305
|
-
for op in ["minimum", "maximum"]:
|
|
306
|
-
supporting_rules.append(f"def ::std::common::{op}(x in {vtype}, y in {vtype}, z in {vtype}): exists((a, b) | ^{vtype}(minimum[a, b], z) and ^{vtype}(a, x) and ^{vtype}(b, y))")
|
|
307
|
-
else:
|
|
308
|
-
body.append(self.to_relation(i, body_vars, body))
|
|
309
|
-
elif i.action == ActionType.Bind and i.entity.value == Builtins.Return:
|
|
310
|
-
head_rel = self.to_return(i, head_vars, body)
|
|
311
|
-
head = f"def {head_rel}:\n "
|
|
312
|
-
elif i.action in [ActionType.Bind, ActionType.Persist, ActionType.Unpersist]:
|
|
313
|
-
if not inline:
|
|
314
|
-
head_rel = self.to_relation(i, head_vars, body, True)
|
|
315
|
-
annotations = set(["from_pyrel"])
|
|
316
|
-
# If optimized rel is requested, add desired attribute
|
|
317
|
-
opt = self.config.get(name="describe_optimization", strict=False)
|
|
318
|
-
if opt == "simple" :
|
|
319
|
-
annotations.add(Builtins.InspectOptimizedSimple.name.lower())
|
|
320
|
-
elif opt == "verbose" :
|
|
321
|
-
annotations.add(Builtins.InspectOptimizedVerbose.name.lower())
|
|
322
|
-
ent = i.entity.value
|
|
323
|
-
if isinstance(ent, Type) or isinstance(ent, Property):
|
|
324
|
-
for parent in ent.parents:
|
|
325
|
-
if parent.isa(Builtins.Annotation):
|
|
326
|
-
annotations.add(parent.name.lower())
|
|
327
|
-
if annotations:
|
|
328
|
-
annotation_str = " ".join([f"@{a}" for a in annotations])
|
|
329
|
-
head = f"{annotation_str}\ndef {head_rel}:\n "
|
|
330
|
-
else:
|
|
331
|
-
head = f"def {head_rel}:\n "
|
|
332
|
-
else:
|
|
333
|
-
props = list(i.bindings.values())
|
|
334
|
-
head_vars.update(props)
|
|
335
|
-
head = f"({', '.join([self.emit_var(v) for v in props])}): "
|
|
336
|
-
elif i.action == ActionType.Construct:
|
|
337
|
-
inputs = [*i.bindings.values()]
|
|
338
|
-
input_str =', '.join([self.emit_var(v) for v in inputs[:-1]])
|
|
339
|
-
output = self.emit_var(inputs[-1])
|
|
340
|
-
body_vars.update(inputs)
|
|
341
|
-
body.append(f"{output} = ^{self.sanitize(self.to_name(i.entity.value)).capitalize()}Type[{input_str}]")
|
|
342
|
-
else:
|
|
343
|
-
raise Exception(f"TODO: Rel emit for action type {i.action}")
|
|
344
|
-
body_str = " and\n ".join(body)
|
|
345
|
-
existing_vars = set()
|
|
346
|
-
for vars in self.stack[:-1]:
|
|
347
|
-
existing_vars.update(vars)
|
|
348
|
-
from_vars = [self.emit_var(v) for v in (body_vars - head_vars - existing_vars) if v.value is None]
|
|
349
|
-
if len(from_vars) and not inline:
|
|
350
|
-
body_str = f"exists(({', '.join(from_vars)}) |\n {body_str})"
|
|
351
|
-
if head and len(from_vars) and inline:
|
|
352
|
-
body_str = f"exists(({', '.join(from_vars)}) | {body_str})"
|
|
353
|
-
|
|
354
|
-
|
|
355
|
-
self.stack.pop()
|
|
356
|
-
|
|
357
|
-
if not head and inline:
|
|
358
|
-
return body_str
|
|
359
|
-
elif not head:
|
|
360
|
-
head = f"def T{task.id}():\n "
|
|
361
|
-
|
|
362
|
-
support_str = ""
|
|
363
|
-
if len(supporting_rules):
|
|
364
|
-
support_str = "\n\n".join(supporting_rules) + "\n\n"
|
|
365
|
-
|
|
366
|
-
if support_str and not head_vars and not body_str:
|
|
367
|
-
return support_str
|
|
368
|
-
|
|
369
|
-
if not body_str:
|
|
370
|
-
body_str = "{()}"
|
|
371
|
-
|
|
372
|
-
|
|
373
|
-
return f"{support_str}{head}{body_str}"
|
relationalai/rel_utils.py
DELETED
|
@@ -1,185 +0,0 @@
|
|
|
1
|
-
from __future__ import annotations
|
|
2
|
-
from typing import Any, Iterable, List, Sequence
|
|
3
|
-
|
|
4
|
-
import pandas as pd
|
|
5
|
-
|
|
6
|
-
from relationalai.clients.types import TransactionAsyncResponse
|
|
7
|
-
from relationalai.clients.util import IdentityParser, ParseError
|
|
8
|
-
from relationalai.dsl import safe_symbol
|
|
9
|
-
from relationalai.errors import RAIException, RAIExceptionSet, RelQueryError
|
|
10
|
-
|
|
11
|
-
#-------------------------------------------------------------------------------
|
|
12
|
-
# Emitting
|
|
13
|
-
#-------------------------------------------------------------------------------
|
|
14
|
-
|
|
15
|
-
class Char(str):
|
|
16
|
-
def __new__(cls, value):
|
|
17
|
-
if value is None:
|
|
18
|
-
raise ValueError("Char cannot be None")
|
|
19
|
-
if len(value) != 1:
|
|
20
|
-
raise ValueError("A Char must be a single character")
|
|
21
|
-
return str.__new__(cls, value)
|
|
22
|
-
|
|
23
|
-
def emit_literal(v: Any):
|
|
24
|
-
"""Emit `v` as its equivalent literal representation in rel."""
|
|
25
|
-
if isinstance(v, Char):
|
|
26
|
-
sanitized = v.replace("'", "\\'")
|
|
27
|
-
return f"'{sanitized}'"
|
|
28
|
-
if isinstance(v, str):
|
|
29
|
-
sanitized = v.replace('"', '\\"').replace("%", "\\%")
|
|
30
|
-
return f'"{sanitized}"'
|
|
31
|
-
if isinstance(v, tuple):
|
|
32
|
-
return ", ".join(emit_literal(item) for item in v)
|
|
33
|
-
return v
|
|
34
|
-
|
|
35
|
-
def emit_nested_relation(prefix: str, obj: dict|None = None, keys: Iterable[str]|None = None, raw = False) -> str:
|
|
36
|
-
"""Emit a set of defs encoding `obj` in GNF on `prefix`."""
|
|
37
|
-
obj = obj or {}
|
|
38
|
-
result = ""
|
|
39
|
-
for k in keys or obj.keys():
|
|
40
|
-
v = obj.get(k, None)
|
|
41
|
-
if isinstance(v, dict):
|
|
42
|
-
result += emit_nested_relation(f"{prefix}{safe_symbol(k)}, ", v)
|
|
43
|
-
elif v is not None:
|
|
44
|
-
result += f"def {prefix}{safe_symbol(k)}]: {emit_literal(v) if not raw else v}\n"
|
|
45
|
-
return result
|
|
46
|
-
|
|
47
|
-
# SEE: REL: https://docs.relational.ai/rel/ref/lexical-symbols#keywords
|
|
48
|
-
# CORE REL: https://docs.google.com/document/d/12LUQdRed7P5EqQI1D7AYG4Q5gno9uKqy32i3kvAWPCA
|
|
49
|
-
RESERVED_WORDS = [
|
|
50
|
-
"and",
|
|
51
|
-
"as",
|
|
52
|
-
"bound",
|
|
53
|
-
"declare",
|
|
54
|
-
"def",
|
|
55
|
-
"else",
|
|
56
|
-
"end",
|
|
57
|
-
"entity",
|
|
58
|
-
"exists",
|
|
59
|
-
"false",
|
|
60
|
-
"for",
|
|
61
|
-
"forall",
|
|
62
|
-
"from",
|
|
63
|
-
"ic",
|
|
64
|
-
"if",
|
|
65
|
-
"iff",
|
|
66
|
-
"implies",
|
|
67
|
-
"in",
|
|
68
|
-
"module",
|
|
69
|
-
"namespace",
|
|
70
|
-
"not",
|
|
71
|
-
"or",
|
|
72
|
-
"requires",
|
|
73
|
-
"then",
|
|
74
|
-
"true",
|
|
75
|
-
"use",
|
|
76
|
-
"where",
|
|
77
|
-
"with",
|
|
78
|
-
"xor"
|
|
79
|
-
]
|
|
80
|
-
|
|
81
|
-
def sanitize_identifier(name: str) -> str:
|
|
82
|
-
"""
|
|
83
|
-
Return a string safe to use as a top level identifier in rel, such as a variable or relation name.
|
|
84
|
-
|
|
85
|
-
Args:
|
|
86
|
-
name (str): The input identifier string.
|
|
87
|
-
|
|
88
|
-
Returns:
|
|
89
|
-
str: The sanitized identifier string.
|
|
90
|
-
"""
|
|
91
|
-
|
|
92
|
-
if not name:
|
|
93
|
-
return name
|
|
94
|
-
|
|
95
|
-
safe_name = ''.join(c if c.isalnum() else '_' for c in name)
|
|
96
|
-
if safe_name[0].isdigit():
|
|
97
|
-
safe_name = '_' + safe_name
|
|
98
|
-
if safe_name in RESERVED_WORDS:
|
|
99
|
-
safe_name = safe_name + "_" # preferring the pythonic pattern of `from_` vs `_from`
|
|
100
|
-
return safe_name
|
|
101
|
-
|
|
102
|
-
def to_fqn_relation_name(fqn: str) -> str:
|
|
103
|
-
"""
|
|
104
|
-
Wrapper around `sanitize_identifier` which also lowers the string if no double quotes are present
|
|
105
|
-
(for backwards compatibility), otherwise leaves it as is.
|
|
106
|
-
|
|
107
|
-
Args:
|
|
108
|
-
fqn (str): The fully qualified name ('db.schema.object').
|
|
109
|
-
|
|
110
|
-
Returns:
|
|
111
|
-
str: A string safe to use as a relation name in rel and matching the format in SF NA.
|
|
112
|
-
"""
|
|
113
|
-
# In this case we do not want to upper case the non unique/double quoted parts of the identifier
|
|
114
|
-
# This is required currently for for backwards compatibility
|
|
115
|
-
parser = IdentityParser(fqn, force_upper_case = False)
|
|
116
|
-
if not parser.is_complete:
|
|
117
|
-
raise ParseError(f"Failed to parse {fqn}. Incomplete relation identifier. Must be fully qualified e.g. 'db.schema.object'.")
|
|
118
|
-
if not parser.identity:
|
|
119
|
-
raise ParseError(f"Failed to parse {fqn}. Empty identifier.")
|
|
120
|
-
|
|
121
|
-
_name = parser.identity if parser.has_double_quoted_identifier else parser.identity.lower()
|
|
122
|
-
return sanitize_identifier(_name)
|
|
123
|
-
|
|
124
|
-
#-------------------------------------------------------------------------------
|
|
125
|
-
# Result Handling
|
|
126
|
-
#-------------------------------------------------------------------------------
|
|
127
|
-
|
|
128
|
-
_known_problems: list[Any] = []
|
|
129
|
-
def ignore_known_problems(problems: Sequence[Any]):
|
|
130
|
-
"""Filter out issues already in `problems` when checking responses for rel problems."""
|
|
131
|
-
global _known_problems
|
|
132
|
-
_known_problems.clear()
|
|
133
|
-
_known_problems.extend(problems)
|
|
134
|
-
|
|
135
|
-
def assert_no_problems(res: TransactionAsyncResponse):
|
|
136
|
-
"""Throw a vaguely-readable exception if rel problems were reported with the given transaction."""
|
|
137
|
-
new_problems = []
|
|
138
|
-
for problem in res.problems or []:
|
|
139
|
-
if problem not in _known_problems:
|
|
140
|
-
new_problems.append(problem)
|
|
141
|
-
|
|
142
|
-
if new_problems:
|
|
143
|
-
errs: List[RAIException] = [RelQueryError(problem) for problem in new_problems]
|
|
144
|
-
if len(errs) > 1:
|
|
145
|
-
raise RAIExceptionSet(errs) from None
|
|
146
|
-
else:
|
|
147
|
-
raise errs[0] from None
|
|
148
|
-
|
|
149
|
-
def maybe_scalarize(v):
|
|
150
|
-
if getattr(v, "__len__", None) and len(v) == 1:
|
|
151
|
-
return v[0]
|
|
152
|
-
return v
|
|
153
|
-
|
|
154
|
-
def process_gnf_results(gnf: Sequence, *key_names: str):
|
|
155
|
-
"""Process GNF results into a nested object keyed by the key(s) in `key_names`"""
|
|
156
|
-
assert len(key_names) > 0, "Must supply a name for the GNF key(s)"
|
|
157
|
-
key_size = len(key_names)
|
|
158
|
-
|
|
159
|
-
obj = {}
|
|
160
|
-
|
|
161
|
-
for rel in gnf:
|
|
162
|
-
sig = rel["relationId"].split("/")
|
|
163
|
-
if sig[1] != ":output":
|
|
164
|
-
continue
|
|
165
|
-
|
|
166
|
-
table: pd.DataFrame = rel["table"].to_pandas()
|
|
167
|
-
|
|
168
|
-
key_cols = table.columns[0:key_size]
|
|
169
|
-
|
|
170
|
-
for _, row in table.iterrows():
|
|
171
|
-
raw_keys = tuple(row[col] for col in key_cols)
|
|
172
|
-
keys = maybe_scalarize(raw_keys)
|
|
173
|
-
vals = [row[col] for col in table.columns[key_size:]]
|
|
174
|
-
|
|
175
|
-
entry = obj.setdefault(keys, {k: v for k, v in zip(key_names, raw_keys)})
|
|
176
|
-
|
|
177
|
-
cur = 4
|
|
178
|
-
field = sig[cur - 2][1:]
|
|
179
|
-
while len(sig) > cur and sig[cur][0] == ":":
|
|
180
|
-
entry = entry.setdefault(field, {})
|
|
181
|
-
field = sig[cur][1:]
|
|
182
|
-
cur += 2
|
|
183
|
-
entry[field] = maybe_scalarize(vals)
|
|
184
|
-
|
|
185
|
-
return obj
|
|
@@ -1,106 +0,0 @@
|
|
|
1
|
-
# Concept.identify_by
|
|
2
|
-
|
|
3
|
-
We want to support reference schemes in the query builder directly, we do this by speficying a set of relationships that act as identifying information for a concept.
|
|
4
|
-
|
|
5
|
-
## constructor variant
|
|
6
|
-
|
|
7
|
-
```python
|
|
8
|
-
SSN = Concept("SSN", extends=[String])
|
|
9
|
-
person = Concept("Person", identify_by={
|
|
10
|
-
"ssn": SSN # can be a Concept, in which case we create a relationship {Person} has {ssn:SSN} behind the scenes
|
|
11
|
-
})
|
|
12
|
-
|
|
13
|
-
EmployeeNumber = Concept("EmployeeNumber", extends=[String])
|
|
14
|
-
employee = Concept("Employee", extends=[Person], identify_by={
|
|
15
|
-
"employee_number": Relation("{Employee} is identified by {EmployeeNumber}"),
|
|
16
|
-
})
|
|
17
|
-
```
|
|
18
|
-
|
|
19
|
-
## function variant
|
|
20
|
-
|
|
21
|
-
```python
|
|
22
|
-
Person = Concept("Person")
|
|
23
|
-
Person.ssn = Relation("{Person} has {SSN}")
|
|
24
|
-
Person.identify_by(Person.ssn)
|
|
25
|
-
```
|
|
26
|
-
|
|
27
|
-
## How this impacts `Concept.new`
|
|
28
|
-
|
|
29
|
-
Once a concept has identifying relationships, values for them must always be provided when calling Concept.new. So given our definitions of Person above:
|
|
30
|
-
|
|
31
|
-
```python
|
|
32
|
-
Person.new() # raises an error, ssn is required
|
|
33
|
-
Person.new(name="foo") # raises an error, ssn is required
|
|
34
|
-
Person.new(ssn="123-45-6789") # works
|
|
35
|
-
Person.new(ssn="123-45-6789", name="foo") # works
|
|
36
|
-
```
|
|
37
|
-
|
|
38
|
-
References schemes are inherited from parent concepts, so because Employee extends Person, we must provide both ssn and employee_number when creating an Employee:
|
|
39
|
-
|
|
40
|
-
```python
|
|
41
|
-
Employee.new(employee_number="123456") # raises an error, ssn is required
|
|
42
|
-
Employee.new(ssn="123-45-6789") # raises an error, employee_number is required
|
|
43
|
-
Employee.new(ssn="123-45-6789", employee_number="123456") # works
|
|
44
|
-
```
|
|
45
|
-
Internally, when calling Concept.new with a reference scheme, we must not only fill the values of the provided relationships, but also add this new identity to the population of the concept and if there are inherited reference schemes, we create the internal mappings from the child identity to the parent one.
|
|
46
|
-
|
|
47
|
-
NOTE: the primary identity is always the highest ancestor in the hierarchy with a reference scheme. This means that while we hash employee_number to create a lookup to person, the identity of the employee will always be the hash of ssn. The QB deals with this mapping under the hood.
|
|
48
|
-
|
|
49
|
-
```python
|
|
50
|
-
Employee.new(ssn="123-45-6789", employee_number="123456")
|
|
51
|
-
|
|
52
|
-
# during compilation will turn into the equivalent of:
|
|
53
|
-
|
|
54
|
-
define(
|
|
55
|
-
p := construct(Person, ssn="123-45-6789"),
|
|
56
|
-
Person(p)
|
|
57
|
-
person_ssn(p, "123-45-6789"),
|
|
58
|
-
Employee(p),
|
|
59
|
-
employee_number(p, "123456"),
|
|
60
|
-
e := construct(Employee, employee_number="123456"),
|
|
61
|
-
employee_to_person(e, p),
|
|
62
|
-
)
|
|
63
|
-
```
|
|
64
|
-
## How this impacts `Concept.filter_by`
|
|
65
|
-
|
|
66
|
-
There are times where we might be working with an external source that only has a child identifier and not the parent one. While we require the parent identifier to be provided to add to the population, if we've already done that elsewhere, it's reasonable to get a reference to the parent identity from the child one. To do that we can use `Concept.filter_by`:
|
|
67
|
-
|
|
68
|
-
```python
|
|
69
|
-
define(
|
|
70
|
-
e := Employee.filter_by(employee_number="123456") # works assuming something has previousl added this employee_number to the population
|
|
71
|
-
e.salary(1000)
|
|
72
|
-
)
|
|
73
|
-
|
|
74
|
-
# during compilation will turn into the equivalent of:
|
|
75
|
-
define(
|
|
76
|
-
e := construct(Employee, employee_number="123456"),
|
|
77
|
-
employee_to_person(e, p),
|
|
78
|
-
employee_salary(p, 1000)
|
|
79
|
-
)
|
|
80
|
-
```
|
|
81
|
-
|
|
82
|
-
As stated previously, all references must map back to the parent identity and this employee id is just used to lookup the person id.
|
|
83
|
-
|
|
84
|
-
## Implied constraints
|
|
85
|
-
|
|
86
|
-
Any relationship that is used to identify a concept is mandatory (which is enforced directly by `.new()`) and also implies that the combination of identifying relationships must be unique. We can create those constraints like so when defining the reference scheme:
|
|
87
|
-
|
|
88
|
-
```python
|
|
89
|
-
# after adding the relationships to the concept
|
|
90
|
-
rels = [getattr(self, rel) for rel in identify_by.keys()]
|
|
91
|
-
self.require(*rels) # make the relationships mandatory
|
|
92
|
-
self.require(distinct(*rels)) # ensure the relationships are unique (needs to be implemented)
|
|
93
|
-
```
|
|
94
|
-
|
|
95
|
-
If you were to write that by hand, it'd look something like this for Employee:
|
|
96
|
-
|
|
97
|
-
```python
|
|
98
|
-
Employee.require(Employee.ssn, Employee.employee_number)
|
|
99
|
-
Employee.require(distinct(Employee.ssn, Employee.employee_number))
|
|
100
|
-
```
|
|
101
|
-
|
|
102
|
-
## Implementation notes
|
|
103
|
-
|
|
104
|
-
We want to store the reference schemes on the concept itself and during the QB -> IR compilation, take them into account when we encounter either a `ConceptNew` (for the `.new(..)` case) or a `ConceptConstruct` (for the `.filter_by(..)` case). At that point in the compiler, we'd add the necessary mappings and handle parent reference schemes.
|
|
105
|
-
|
|
106
|
-
Since we allow for multiple inheritance, the "primary" reference scheme is determined by doing depth first search on the parent types and taking the first ancestor with a reference scheme that we encounter. This means a user can control this by ordering their parents in `extends=[..]`.
|