pyfect-dev 0.1.0__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.
- pyfect/__init__.py +729 -0
- pyfect/_all.py +242 -0
- pyfect/_background.py +28 -0
- pyfect/_binary_codecs.py +124 -0
- pyfect/_cached.py +165 -0
- pyfect/_catch.py +128 -0
- pyfect/_catch_reason.py +574 -0
- pyfect/_concurrency.py +291 -0
- pyfect/_context_effects.py +163 -0
- pyfect/_context_managers.py +76 -0
- pyfect/_core.py +368 -0
- pyfect/_digits.py +71 -0
- pyfect/_effect_args.py +46 -0
- pyfect/_effect_base.py +71 -0
- pyfect/_effect_concurrency.py +133 -0
- pyfect/_effect_eager.py +103 -0
- pyfect/_effect_errors.py +344 -0
- pyfect/_effect_folding.py +110 -0
- pyfect/_effect_hash.py +140 -0
- pyfect/_effect_logging.py +57 -0
- pyfect/_effect_metrics.py +102 -0
- pyfect/_effect_resources.py +179 -0
- pyfect/_effect_services.py +25 -0
- pyfect/_effect_timing.py +199 -0
- pyfect/_effect_tracing.py +99 -0
- pyfect/_equal.py +195 -0
- pyfect/_extras.py +43 -0
- pyfect/_fiber.py +969 -0
- pyfect/_fiber_all.py +134 -0
- pyfect/_fiber_collections.py +106 -0
- pyfect/_fn.py +706 -0
- pyfect/_hamt.py +174 -0
- pyfect/_interrupt_reason.py +33 -0
- pyfect/_isaac.py +149 -0
- pyfect/_iterables.py +231 -0
- pyfect/_js.py +263 -0
- pyfect/_js_date.py +87 -0
- pyfect/_json_text.py +286 -0
- pyfect/_json_value.py +9 -0
- pyfect/_layer_core.py +403 -0
- pyfect/_layer_for_life.py +33 -0
- pyfect/_layer_tracing.py +70 -0
- pyfect/_locked_ref.py +147 -0
- pyfect/_loop_thread.py +152 -0
- pyfect/_metric_ref.py +29 -0
- pyfect/_one_or_many.py +14 -0
- pyfect/_os_wait.py +118 -0
- pyfect/_path_posix.py +129 -0
- pyfect/_primitives.py +434 -0
- pyfect/_public.py +36 -0
- pyfect/_reasons.py +64 -0
- pyfect/_released.py +67 -0
- pyfect/_root.py +87 -0
- pyfect/_round_robin.py +73 -0
- pyfect/_some.py +17 -0
- pyfect/_stack.py +52 -0
- pyfect/_stack_frame.py +114 -0
- pyfect/_string_hash.py +35 -0
- pyfect/_subscriber_queue.py +47 -0
- pyfect/_tags.py +14 -0
- pyfect/_tracer_ref.py +21 -0
- pyfect/_tx_ops.py +77 -0
- pyfect/_unsubscripted.py +23 -0
- pyfect/_url.py +54 -0
- pyfect/_waiters.py +127 -0
- pyfect/_wake_tracking.py +98 -0
- pyfect/_wire_codec.py +123 -0
- pyfect/ai/__init__.py +73 -0
- pyfect/ai/_approvals.py +249 -0
- pyfect/ai/_decision_answers.py +185 -0
- pyfect/ai/_generated_client.py +298 -0
- pyfect/ai/_generated_support.py +84 -0
- pyfect/ai/_mcp/__init__.py +1 -0
- pyfect/ai/_mcp/batch.py +68 -0
- pyfect/ai/_mcp/client_transport.py +412 -0
- pyfect/ai/_mcp/client_wire.py +284 -0
- pyfect/ai/_mcp/core.py +490 -0
- pyfect/ai/_mcp/dispatch.py +1048 -0
- pyfect/ai/_mcp/http.py +462 -0
- pyfect/ai/_mcp/http_admission.py +269 -0
- pyfect/ai/_mcp/http_status.py +27 -0
- pyfect/ai/_mcp/input_wire.py +226 -0
- pyfect/ai/_mcp/log_levels.py +55 -0
- pyfect/ai/_mcp/param_headers.py +63 -0
- pyfect/ai/_mcp/projection.py +248 -0
- pyfect/ai/_mcp/register.py +683 -0
- pyfect/ai/_mcp/reverse_policy.py +155 -0
- pyfect/ai/_mcp/routing_header.py +24 -0
- pyfect/ai/_mcp/sessions.py +114 -0
- pyfect/ai/_mcp/stdio.py +110 -0
- pyfect/ai/_mcp/subscriptions.py +134 -0
- pyfect/ai/_mcp/tool_schema.py +35 -0
- pyfect/ai/_noul_answers.py +64 -0
- pyfect/ai/_openai_all_of.py +137 -0
- pyfect/ai/_openai_errors.py +233 -0
- pyfect/ai/_provider.py +439 -0
- pyfect/ai/_provider_tools.py +129 -0
- pyfect/ai/_stream_resolution.py +231 -0
- pyfect/ai/_structured_output_checks.py +138 -0
- pyfect/ai/_structured_output_codec.py +229 -0
- pyfect/ai/_structured_output_json.py +109 -0
- pyfect/ai/_tool_call_ids.py +80 -0
- pyfect/ai/ai_error.py +785 -0
- pyfect/ai/anthropic/__init__.py +28 -0
- pyfect/ai/anthropic/_errors.py +164 -0
- pyfect/ai/anthropic/_prompt.py +375 -0
- pyfect/ai/anthropic/_response.py +443 -0
- pyfect/ai/anthropic/_schema.py +206 -0
- pyfect/ai/anthropic/_stream.py +377 -0
- pyfect/ai/anthropic/_tools.py +248 -0
- pyfect/ai/anthropic/client.py +289 -0
- pyfect/ai/anthropic/config.py +32 -0
- pyfect/ai/anthropic/error.py +27 -0
- pyfect/ai/anthropic/language_model.py +366 -0
- pyfect/ai/anthropic/telemetry.py +51 -0
- pyfect/ai/anthropic/tool.py +570 -0
- pyfect/ai/anthropic_structured_output.py +138 -0
- pyfect/ai/chat.py +548 -0
- pyfect/ai/cloudflare/__init__.py +15 -0
- pyfect/ai/cloudflare/_errors.py +108 -0
- pyfect/ai/cloudflare/client.py +174 -0
- pyfect/ai/cloudflare/decision_model.py +117 -0
- pyfect/ai/cloudflare/schema.py +144 -0
- pyfect/ai/decision.py +290 -0
- pyfect/ai/decision_model.py +283 -0
- pyfect/ai/embedding_model.py +177 -0
- pyfect/ai/id_generator.py +79 -0
- pyfect/ai/language_model.py +1077 -0
- pyfect/ai/mcp_client.py +682 -0
- pyfect/ai/mcp_client_error.py +223 -0
- pyfect/ai/mcp_protocol.py +205 -0
- pyfect/ai/mcp_schema.py +1300 -0
- pyfect/ai/mcp_server.py +404 -0
- pyfect/ai/model.py +56 -0
- pyfect/ai/openai/__init__.py +49 -0
- pyfect/ai/openai/_errors.py +89 -0
- pyfect/ai/openai/_prompt.py +527 -0
- pyfect/ai/openai/_requests.py +34 -0
- pyfect/ai/openai/_response.py +569 -0
- pyfect/ai/openai/_schema.py +769 -0
- pyfect/ai/openai/_stream.py +771 -0
- pyfect/ai/openai/_tools.py +128 -0
- pyfect/ai/openai/_web_socket.py +295 -0
- pyfect/ai/openai/client.py +296 -0
- pyfect/ai/openai/client_generated.py +123 -0
- pyfect/ai/openai/config.py +34 -0
- pyfect/ai/openai/decision_model.py +177 -0
- pyfect/ai/openai/embedding_model.py +116 -0
- pyfect/ai/openai/error.py +8 -0
- pyfect/ai/openai/generated/__init__.py +7213 -0
- pyfect/ai/openai/generated/_recursive.py +18 -0
- pyfect/ai/openai/generated/_schemas_0.py +1015 -0
- pyfect/ai/openai/generated/_schemas_1.py +1016 -0
- pyfect/ai/openai/generated/_schemas_2.py +1017 -0
- pyfect/ai/openai/generated/_schemas_3.py +615 -0
- pyfect/ai/openai/generated/client.py +5298 -0
- pyfect/ai/openai/generated/types.py +8123 -0
- pyfect/ai/openai/language_model.py +304 -0
- pyfect/ai/openai/telemetry.py +19 -0
- pyfect/ai/openai/tool.py +391 -0
- pyfect/ai/openai_compat/__init__.py +25 -0
- pyfect/ai/openai_compat/_prompt.py +191 -0
- pyfect/ai/openai_compat/_request.py +153 -0
- pyfect/ai/openai_compat/_response.py +407 -0
- pyfect/ai/openai_compat/_schema.py +97 -0
- pyfect/ai/openai_compat/client.py +234 -0
- pyfect/ai/openai_compat/config.py +32 -0
- pyfect/ai/openai_compat/embedding_model.py +139 -0
- pyfect/ai/openai_compat/error.py +7 -0
- pyfect/ai/openai_compat/language_model.py +245 -0
- pyfect/ai/openai_compat/telemetry.py +51 -0
- pyfect/ai/openai_structured_output.py +118 -0
- pyfect/ai/openrouter/__init__.py +36 -0
- pyfect/ai/openrouter/_errors.py +186 -0
- pyfect/ai/openrouter/_prompt.py +272 -0
- pyfect/ai/openrouter/_response.py +226 -0
- pyfect/ai/openrouter/_stream.py +320 -0
- pyfect/ai/openrouter/client.py +283 -0
- pyfect/ai/openrouter/config.py +36 -0
- pyfect/ai/openrouter/decision_model.py +161 -0
- pyfect/ai/openrouter/error.py +23 -0
- pyfect/ai/openrouter/generated/__init__.py +3029 -0
- pyfect/ai/openrouter/generated/_recursive.py +15 -0
- pyfect/ai/openrouter/generated/_schemas_0.py +1015 -0
- pyfect/ai/openrouter/generated/_schemas_1.py +522 -0
- pyfect/ai/openrouter/generated/client.py +1445 -0
- pyfect/ai/openrouter/generated/types.py +4769 -0
- pyfect/ai/openrouter/language_model.py +345 -0
- pyfect/ai/openrouter/schema.py +165 -0
- pyfect/ai/prompt.py +664 -0
- pyfect/ai/response.py +768 -0
- pyfect/ai/response_id_tracker.py +82 -0
- pyfect/ai/telemetry.py +126 -0
- pyfect/ai/tokenizer.py +61 -0
- pyfect/ai/tool.py +536 -0
- pyfect/ai/toolkit.py +490 -0
- pyfect/ai/typesafe/__init__.py +17 -0
- pyfect/ai/typesafe/_errors.py +120 -0
- pyfect/ai/typesafe/client.py +146 -0
- pyfect/ai/typesafe/config.py +34 -0
- pyfect/ai/typesafe/decision_model.py +90 -0
- pyfect/ai/typesafe/schema.py +140 -0
- pyfect/batching/__init__.py +58 -0
- pyfect/batching/_api.py +71 -0
- pyfect/batching/_engine.py +229 -0
- pyfect/batching/_from_effect.py +76 -0
- pyfect/batching/_persisted.py +98 -0
- pyfect/batching/_request.py +133 -0
- pyfect/batching/_resolver.py +477 -0
- pyfect/batching/_with_cache.py +90 -0
- pyfect/big_decimal.py +500 -0
- pyfect/brand.py +114 -0
- pyfect/byte_size.py +354 -0
- pyfect/cache/__init__.py +29 -0
- pyfect/cache/_cache.py +426 -0
- pyfect/cache/_expiry.py +69 -0
- pyfect/cache/_scoped_cache.py +546 -0
- pyfect/case.py +174 -0
- pyfect/cause.py +462 -0
- pyfect/channel/__init__.py +155 -0
- pyfect/channel/_base.py +92 -0
- pyfect/channel/_catch.py +158 -0
- pyfect/channel/_catch_reason.py +572 -0
- pyfect/channel/_channel.py +158 -0
- pyfect/channel/_combinators.py +220 -0
- pyfect/channel/_concurrency_methods.py +217 -0
- pyfect/channel/_concurrent_map.py +118 -0
- pyfect/channel/_constructors.py +349 -0
- pyfect/channel/_context.py +232 -0
- pyfect/channel/_error_methods.py +336 -0
- pyfect/channel/_errors.py +180 -0
- pyfect/channel/_filters.py +225 -0
- pyfect/channel/_lifecycle.py +71 -0
- pyfect/channel/_lifecycle_methods.py +206 -0
- pyfect/channel/_loop.py +100 -0
- pyfect/channel/_merge.py +246 -0
- pyfect/channel/_repeat.py +154 -0
- pyfect/channel/_run.py +149 -0
- pyfect/channel/_run_methods.py +341 -0
- pyfect/channel/_sinks.py +88 -0
- pyfect/channel/_sources.py +238 -0
- pyfect/channel/_text.py +176 -0
- pyfect/channel/_transform_methods.py +415 -0
- pyfect/channel/_upstream.py +67 -0
- pyfect/channel_schema.py +131 -0
- pyfect/cli/__init__.py +79 -0
- pyfect/cli/_ansi.py +82 -0
- pyfect/cli/_command.py +469 -0
- pyfect/cli/_config.py +164 -0
- pyfect/cli/_environment.py +10 -0
- pyfect/cli/_help.py +195 -0
- pyfect/cli/_lexer.py +81 -0
- pyfect/cli/_parser.py +465 -0
- pyfect/cli/_settings.py +85 -0
- pyfect/cli/_suggest.py +33 -0
- pyfect/cli/_width.py +134 -0
- pyfect/cli/_wizard.py +317 -0
- pyfect/cli/argument.py +212 -0
- pyfect/cli/cli_config.py +40 -0
- pyfect/cli/cli_error.py +242 -0
- pyfect/cli/cli_output.py +305 -0
- pyfect/cli/command.py +440 -0
- pyfect/cli/completions/__init__.py +50 -0
- pyfect/cli/completions/_completer.ps1 +190 -0
- pyfect/cli/completions/_descriptor.py +182 -0
- pyfect/cli/completions/bash.py +300 -0
- pyfect/cli/completions/fish.py +164 -0
- pyfect/cli/completions/powershell.py +156 -0
- pyfect/cli/completions/zsh.py +189 -0
- pyfect/cli/flag.py +246 -0
- pyfect/cli/global_flag.py +99 -0
- pyfect/cli/help_doc.py +63 -0
- pyfect/cli/param.py +789 -0
- pyfect/cli/primitive.py +308 -0
- pyfect/cli/prompt/__init__.py +75 -0
- pyfect/cli/prompt/_auto_complete.py +205 -0
- pyfect/cli/prompt/_choices.py +104 -0
- pyfect/cli/prompt/_confirm.py +195 -0
- pyfect/cli/prompt/_core.py +512 -0
- pyfect/cli/prompt/_date.py +173 -0
- pyfect/cli/prompt/_date_parts.py +387 -0
- pyfect/cli/prompt/_file.py +372 -0
- pyfect/cli/prompt/_multi_select.py +204 -0
- pyfect/cli/prompt/_number.py +257 -0
- pyfect/cli/prompt/_select.py +117 -0
- pyfect/cli/prompt/_text.py +239 -0
- pyfect/cli/terminal_capabilities.py +118 -0
- pyfect/clock.py +150 -0
- pyfect/cluster/__init__.py +155 -0
- pyfect/cluster/_cluster_abandon.py +36 -0
- pyfect/cluster/_cluster_layer.py +122 -0
- pyfect/cluster/_entity_activation.py +16 -0
- pyfect/cluster/_entity_manager/__init__.py +22 -0
- pyfect/cluster/_entity_manager/_activation.py +540 -0
- pyfect/cluster/_entity_manager/_core.py +186 -0
- pyfect/cluster/_entity_manager/_decode.py +60 -0
- pyfect/cluster/_entity_manager/_manager.py +423 -0
- pyfect/cluster/_entity_manager/_state.py +113 -0
- pyfect/cluster/_entity_reaper.py +94 -0
- pyfect/cluster/_hash.py +13 -0
- pyfect/cluster/_interruptors.py +88 -0
- pyfect/cluster/_resource_map.py +179 -0
- pyfect/cluster/_resource_ref.py +266 -0
- pyfect/cluster/_shard_lock.py +14 -0
- pyfect/cluster/cluster_cron.py +230 -0
- pyfect/cluster/cluster_error.py +81 -0
- pyfect/cluster/cluster_http.py +210 -0
- pyfect/cluster/cluster_metrics.py +54 -0
- pyfect/cluster/cluster_schema.py +137 -0
- pyfect/cluster/cluster_socket.py +197 -0
- pyfect/cluster/cluster_workflow_engine/__init__.py +24 -0
- pyfect/cluster/cluster_workflow_engine/_engine.py +1099 -0
- pyfect/cluster/cluster_workflow_engine/_entities.py +298 -0
- pyfect/cluster/deliver_at.py +31 -0
- pyfect/cluster/entity/__init__.py +39 -0
- pyfect/cluster/entity/_entity.py +332 -0
- pyfect/cluster/entity/_handlers.py +168 -0
- pyfect/cluster/entity/_keep_alive.py +74 -0
- pyfect/cluster/entity/_queue.py +103 -0
- pyfect/cluster/entity/_request.py +34 -0
- pyfect/cluster/entity/_services.py +23 -0
- pyfect/cluster/entity/_test_client.py +237 -0
- pyfect/cluster/entity_address.py +25 -0
- pyfect/cluster/entity_id.py +20 -0
- pyfect/cluster/entity_proxy.py +161 -0
- pyfect/cluster/entity_proxy_server.py +103 -0
- pyfect/cluster/entity_resource.py +122 -0
- pyfect/cluster/entity_type.py +20 -0
- pyfect/cluster/envelope.py +307 -0
- pyfect/cluster/http_runner.py +254 -0
- pyfect/cluster/k8s_http_client.py +281 -0
- pyfect/cluster/k8s_types.py +1394 -0
- pyfect/cluster/machine_id.py +21 -0
- pyfect/cluster/message.py +219 -0
- pyfect/cluster/message_storage/__init__.py +67 -0
- pyfect/cluster/message_storage/_encoded.py +393 -0
- pyfect/cluster/message_storage/_memory.py +386 -0
- pyfect/cluster/message_storage/_noop.py +109 -0
- pyfect/cluster/message_storage/_service.py +388 -0
- pyfect/cluster/reply.py +236 -0
- pyfect/cluster/runner.py +32 -0
- pyfect/cluster/runner_address.py +23 -0
- pyfect/cluster/runner_health.py +93 -0
- pyfect/cluster/runner_server.py +286 -0
- pyfect/cluster/runner_storage.py +212 -0
- pyfect/cluster/runners/__init__.py +55 -0
- pyfect/cluster/runners/_local.py +460 -0
- pyfect/cluster/runners/_noop.py +57 -0
- pyfect/cluster/runners/_rpc.py +290 -0
- pyfect/cluster/runners/_service.py +113 -0
- pyfect/cluster/shard_id.py +39 -0
- pyfect/cluster/sharding/__init__.py +12 -0
- pyfect/cluster/sharding/_acquisition.py +490 -0
- pyfect/cluster/sharding/_clients.py +271 -0
- pyfect/cluster/sharding/_make.py +388 -0
- pyfect/cluster/sharding/_routing.py +296 -0
- pyfect/cluster/sharding/_runners_sync.py +198 -0
- pyfect/cluster/sharding/_service.py +91 -0
- pyfect/cluster/sharding/_singletons.py +183 -0
- pyfect/cluster/sharding/_state.py +296 -0
- pyfect/cluster/sharding/_storage_loop.py +421 -0
- pyfect/cluster/sharding_config.py +230 -0
- pyfect/cluster/sharding_registration_event.py +28 -0
- pyfect/cluster/single_runner.py +49 -0
- pyfect/cluster/singleton.py +36 -0
- pyfect/cluster/singleton_address.py +20 -0
- pyfect/cluster/snowflake.py +133 -0
- pyfect/cluster/socket_runner.py +67 -0
- pyfect/cluster/sql_message_storage.py +1001 -0
- pyfect/cluster/sql_runner_storage.py +841 -0
- pyfect/cluster/test_runner.py +23 -0
- pyfect/config/__init__.py +93 -0
- pyfect/config/_all.py +96 -0
- pyfect/config/_collections.py +291 -0
- pyfect/config/_config.py +261 -0
- pyfect/config/_constructors.py +106 -0
- pyfect/config/_error.py +27 -0
- pyfect/config/_leaf.py +92 -0
- pyfect/config/_scalars.py +193 -0
- pyfect/config/_schema.py +340 -0
- pyfect/config_provider/__init__.py +60 -0
- pyfect/config_provider/_dir.py +84 -0
- pyfect/config_provider/_dotenv.py +127 -0
- pyfect/config_provider/_env.py +79 -0
- pyfect/config_provider/_layers.py +80 -0
- pyfect/config_provider/_node.py +90 -0
- pyfect/config_provider/_provider.py +118 -0
- pyfect/config_provider/_unknown.py +62 -0
- pyfect/console/__init__.py +63 -0
- pyfect/console/_effects.py +134 -0
- pyfect/console/_live.py +156 -0
- pyfect/console/_service.py +49 -0
- pyfect/console/_table.py +90 -0
- pyfect/context.py +414 -0
- pyfect/cron/__init__.py +30 -0
- pyfect/cron/_model.py +183 -0
- pyfect/cron/_search.py +268 -0
- pyfect/cron/_text.py +202 -0
- pyfect/cron/_wall.py +42 -0
- pyfect/crypto.py +257 -0
- pyfect/date_time/__init__.py +77 -0
- pyfect/date_time/_calendar.py +209 -0
- pyfect/date_time/_current.py +118 -0
- pyfect/date_time/_epoch.py +22 -0
- pyfect/date_time/_iso.py +80 -0
- pyfect/date_time/_zones.py +135 -0
- pyfect/date_time/intl.py +144 -0
- pyfect/deferred.py +145 -0
- pyfect/devtools/__init__.py +24 -0
- pyfect/devtools/client.py +323 -0
- pyfect/devtools/layers.py +32 -0
- pyfect/devtools/schema.py +301 -0
- pyfect/devtools/server.py +72 -0
- pyfect/duration.py +295 -0
- pyfect/effect.py +565 -0
- pyfect/effectable.py +37 -0
- pyfect/encoding/__init__.py +62 -0
- pyfect/encoding/_archive.py +197 -0
- pyfect/encoding/_json_array_scanner.py +477 -0
- pyfect/encoding/base64.py +23 -0
- pyfect/encoding/base64_url.py +19 -0
- pyfect/encoding/encoding_error.py +17 -0
- pyfect/encoding/hex.py +28 -0
- pyfect/encoding/ini.py +188 -0
- pyfect/encoding/json_array.py +246 -0
- pyfect/encoding/ndjson.py +461 -0
- pyfect/encoding/schema_binary/__init__.py +257 -0
- pyfect/encoding/schema_binary/_channels.py +173 -0
- pyfect/encoding/schema_binary/_compile.py +584 -0
- pyfect/encoding/schema_binary/_decode.py +848 -0
- pyfect/encoding/schema_binary/_dictionary.py +90 -0
- pyfect/encoding/schema_binary/_encode.py +838 -0
- pyfect/encoding/schema_binary/_fingerprint.py +168 -0
- pyfect/encoding/schema_binary/_layout.py +350 -0
- pyfect/encoding/schema_binary/_literals.py +47 -0
- pyfect/encoding/schema_binary/_parser.py +163 -0
- pyfect/encoding/schema_binary/_reader.py +148 -0
- pyfect/encoding/schema_binary/_target.py +310 -0
- pyfect/encoding/schema_binary/_values.py +133 -0
- pyfect/encoding/schema_binary/_wire.py +253 -0
- pyfect/encoding/sse.py +476 -0
- pyfect/encoding/tar/__init__.py +396 -0
- pyfect/encoding/tar/_decoder.py +220 -0
- pyfect/encoding/tar/_encoder.py +83 -0
- pyfect/encoding/tar/_header.py +494 -0
- pyfect/encoding/tar/_model.py +82 -0
- pyfect/encoding/toml.py +375 -0
- pyfect/encoding/xml/__init__.py +222 -0
- pyfect/encoding/xml/_chars.py +119 -0
- pyfect/encoding/xml/_collector.py +106 -0
- pyfect/encoding/xml/_inputs.py +126 -0
- pyfect/encoding/xml/_model.py +114 -0
- pyfect/encoding/xml/_parser.py +694 -0
- pyfect/encoding/xml/_scanner.py +180 -0
- pyfect/encoding/xml/_writer.py +238 -0
- pyfect/encoding/yaml.py +739 -0
- pyfect/encoding/zip/__init__.py +392 -0
- pyfect/encoding/zip/_decoder.py +378 -0
- pyfect/encoding/zip/_format.py +122 -0
- pyfect/encoding/zip/_inflate.py +65 -0
- pyfect/encoding/zip/_model.py +86 -0
- pyfect/encoding/zip/_writer.py +311 -0
- pyfect/error_reporter.py +240 -0
- pyfect/errors.py +321 -0
- pyfect/eventlog/__init__.py +77 -0
- pyfect/eventlog/_conflicts.py +94 -0
- pyfect/eventlog/_crypto.py +64 -0
- pyfect/eventlog/_identity.py +26 -0
- pyfect/eventlog/_identity_derivation.py +79 -0
- pyfect/eventlog/_transferable_bytes.py +55 -0
- pyfect/eventlog/_uuid.py +33 -0
- pyfect/eventlog/event.py +115 -0
- pyfect/eventlog/event_group.py +55 -0
- pyfect/eventlog/event_journal.py +630 -0
- pyfect/eventlog/event_log.py +1039 -0
- pyfect/eventlog/event_log_encryption.py +219 -0
- pyfect/eventlog/event_log_message.py +357 -0
- pyfect/eventlog/event_log_remote.py +465 -0
- pyfect/eventlog/event_log_server.py +442 -0
- pyfect/eventlog/event_log_server_encrypted.py +322 -0
- pyfect/eventlog/event_log_server_unencrypted.py +814 -0
- pyfect/eventlog/event_log_session_auth.py +250 -0
- pyfect/eventlog/event_row.py +423 -0
- pyfect/eventlog/sql_event_journal.py +692 -0
- pyfect/eventlog/sql_event_log_server_encrypted.py +440 -0
- pyfect/eventlog/sql_event_log_server_unencrypted.py +568 -0
- pyfect/execution_plan/__init__.py +42 -0
- pyfect/execution_plan/_events.py +128 -0
- pyfect/execution_plan/_model.py +108 -0
- pyfect/execution_plan/_run.py +176 -0
- pyfect/exit.py +205 -0
- pyfect/fiber_handle.py +305 -0
- pyfect/fiber_map.py +350 -0
- pyfect/fiber_set.py +232 -0
- pyfect/file_system/__init__.py +69 -0
- pyfect/file_system/_copy.py +116 -0
- pyfect/file_system/_errors.py +32 -0
- pyfect/file_system/_file.py +177 -0
- pyfect/file_system/_live.py +351 -0
- pyfect/file_system/_model.py +72 -0
- pyfect/file_system/_noop.py +189 -0
- pyfect/file_system/_service.py +336 -0
- pyfect/file_system/_stat.py +69 -0
- pyfect/file_system/_watch.py +107 -0
- pyfect/filters.py +245 -0
- pyfect/graph/__init__.py +101 -0
- pyfect/graph/_all_pairs.py +150 -0
- pyfect/graph/_combine.py +259 -0
- pyfect/graph/_compact.py +143 -0
- pyfect/graph/_components.py +200 -0
- pyfect/graph/_cycles.py +144 -0
- pyfect/graph/_enumerate.py +168 -0
- pyfect/graph/_errors.py +51 -0
- pyfect/graph/_flow.py +164 -0
- pyfect/graph/_forest.py +109 -0
- pyfect/graph/_graph.py +269 -0
- pyfect/graph/_lowlink.py +98 -0
- pyfect/graph/_matching.py +101 -0
- pyfect/graph/_mutate.py +211 -0
- pyfect/graph/_query.py +138 -0
- pyfect/graph/_records.py +119 -0
- pyfect/graph/_render.py +109 -0
- pyfect/graph/_shortest.py +274 -0
- pyfect/graph/_snapshot.py +50 -0
- pyfect/graph/_store.py +136 -0
- pyfect/graph/_traverse.py +289 -0
- pyfect/graph/_view.py +384 -0
- pyfect/graph/_walker.py +70 -0
- pyfect/graphql/__init__.py +87 -0
- pyfect/graphql/_codec.py +117 -0
- pyfect/graphql/_http.py +309 -0
- pyfect/graphql/_schemas.py +39 -0
- pyfect/graphql/_slices.py +110 -0
- pyfect/graphql/_template.py +111 -0
- pyfect/graphql/_types.py +115 -0
- pyfect/graphql/_ws.py +395 -0
- pyfect/graphql/_ws_messages.py +120 -0
- pyfect/graphql/client.py +428 -0
- pyfect/graphql/client_error.py +285 -0
- pyfect/graphql/group.py +142 -0
- pyfect/graphql/middleware.py +98 -0
- pyfect/graphql/operation.py +409 -0
- pyfect/graphql/protocol.py +119 -0
- pyfect/graphql/resolver.py +574 -0
- pyfect/graphql/template.py +61 -0
- pyfect/hash_map.py +317 -0
- pyfect/hash_ring.py +206 -0
- pyfect/hash_set.py +163 -0
- pyfect/http/__init__.py +125 -0
- pyfect/http/_asgi_listener.py +272 -0
- pyfect/http/_authority.py +63 -0
- pyfect/http/_brotli.py +27 -0
- pyfect/http/_compression.py +52 -0
- pyfect/http/_field.py +18 -0
- pyfect/http/_foreign_path.py +27 -0
- pyfect/http/_form_data.py +34 -0
- pyfect/http/_httpx_options.py +120 -0
- pyfect/http/_in_memory.py +76 -0
- pyfect/http/_request_state.py +59 -0
- pyfect/http/_vary.py +14 -0
- pyfect/http/abort.py +65 -0
- pyfect/http/asgi/__init__.py +59 -0
- pyfect/http/asgi/_app.py +155 -0
- pyfect/http/asgi/_http.py +201 -0
- pyfect/http/asgi/_mount.py +263 -0
- pyfect/http/asgi/_scope.py +86 -0
- pyfect/http/asgi/_serve.py +72 -0
- pyfect/http/asgi/_types.py +13 -0
- pyfect/http/asgi/_web_socket.py +325 -0
- pyfect/http/body.py +437 -0
- pyfect/http/client/__init__.py +111 -0
- pyfect/http/client/_circuit_breaker.py +113 -0
- pyfect/http/client/_cookie_jar.py +120 -0
- pyfect/http/client/_make.py +252 -0
- pyfect/http/client/_rate_limit.py +423 -0
- pyfect/http/client/_redirects.py +93 -0
- pyfect/http/client/_retry.py +115 -0
- pyfect/http/client/_service.py +497 -0
- pyfect/http/cookies.py +444 -0
- pyfect/http/error.py +188 -0
- pyfect/http/etag.py +96 -0
- pyfect/http/file_like.py +49 -0
- pyfect/http/find_my_way/__init__.py +10 -0
- pyfect/http/find_my_way/_nodes.py +186 -0
- pyfect/http/find_my_way/_paths.py +43 -0
- pyfect/http/find_my_way/_query_string.py +67 -0
- pyfect/http/find_my_way/_router.py +401 -0
- pyfect/http/headers.py +343 -0
- pyfect/http/http_effect.py +198 -0
- pyfect/http/httpx_client.py +291 -0
- pyfect/http/hypercorn_server.py +122 -0
- pyfect/http/incoming_message.py +140 -0
- pyfect/http/live_server/__init__.py +191 -0
- pyfect/http/live_server/_body.py +164 -0
- pyfect/http/live_server/_exchange.py +321 -0
- pyfect/http/live_server/_head.py +194 -0
- pyfect/http/live_server/_inbound.py +156 -0
- pyfect/http/live_server/_listener.py +211 -0
- pyfect/http/live_server/_serve.py +225 -0
- pyfect/http/live_server/_timers.py +20 -0
- pyfect/http/live_server/_upgrade.py +108 -0
- pyfect/http/live_server/_websocket.py +286 -0
- pyfect/http/live_server/_ws_frames.py +242 -0
- pyfect/http/method.py +30 -0
- pyfect/http/middleware/__init__.py +61 -0
- pyfect/http/middleware/_compression.py +156 -0
- pyfect/http/middleware/_cors.py +129 -0
- pyfect/http/middleware/_decompression.py +274 -0
- pyfect/http/middleware/_forwarded.py +113 -0
- pyfect/http/middleware/_logger.py +73 -0
- pyfect/http/middleware/_search_params.py +34 -0
- pyfect/http/middleware/_tracer.py +153 -0
- pyfect/http/middleware/_types.py +16 -0
- pyfect/http/mime/__init__.py +69 -0
- pyfect/http/mime/_types.py +361 -0
- pyfect/http/multipart.py +613 -0
- pyfect/http/multipart_parser/__init__.py +348 -0
- pyfect/http/multipart_parser/_content_type.py +109 -0
- pyfect/http/multipart_parser/_headers.py +206 -0
- pyfect/http/multipart_parser/_search.py +85 -0
- pyfect/http/origin.py +146 -0
- pyfect/http/platform.py +593 -0
- pyfect/http/prometheus.py +34 -0
- pyfect/http/request.py +384 -0
- pyfect/http/response.py +474 -0
- pyfect/http/router/__init__.py +82 -0
- pyfect/http/router/_api.py +462 -0
- pyfect/http/router/_in_memory.py +76 -0
- pyfect/http/router/_markers.py +47 -0
- pyfect/http/router/_middleware.py +181 -0
- pyfect/http/router/_middleware_stack.py +51 -0
- pyfect/http/router/_provided.py +31 -0
- pyfect/http/router/_route.py +114 -0
- pyfect/http/router/_router.py +308 -0
- pyfect/http/schema.py +276 -0
- pyfect/http/server.py +199 -0
- pyfect/http/server_error.py +261 -0
- pyfect/http/server_request.py +729 -0
- pyfect/http/server_respondable.py +77 -0
- pyfect/http/server_response.py +569 -0
- pyfect/http/static_server/__init__.py +257 -0
- pyfect/http/static_server/_conditional.py +72 -0
- pyfect/http/static_server/_mime.py +55 -0
- pyfect/http/static_server/_range.py +53 -0
- pyfect/http/status.py +51 -0
- pyfect/http/template.py +161 -0
- pyfect/http/trace_context.py +91 -0
- pyfect/http/url.py +442 -0
- pyfect/http/url_params.py +304 -0
- pyfect/http/uvicorn_server.py +119 -0
- pyfect/http/web.py +171 -0
- pyfect/http/web_handler.py +463 -0
- pyfect/http/wire.py +209 -0
- pyfect/http_api/__init__.py +83 -0
- pyfect/http_api/_assets/scalar.js +8 -0
- pyfect/http_api/_assets/swagger-ui.css +3 -0
- pyfect/http_api/_assets/swagger-ui.js +4 -0
- pyfect/http_api/_client_method.py +220 -0
- pyfect/http_api/_client_payload.py +144 -0
- pyfect/http_api/_client_response.py +377 -0
- pyfect/http_api/_handlers.py +282 -0
- pyfect/http_api/_html.py +38 -0
- pyfect/http_api/_media_type.py +6 -0
- pyfect/http_api/_openapi_responses.py +164 -0
- pyfect/http_api/_openapi_schemas.py +163 -0
- pyfect/http_api/_path.py +102 -0
- pyfect/http_api/_payload.py +260 -0
- pyfect/http_api/_response.py +363 -0
- pyfect/http_api/_security.py +174 -0
- pyfect/http_api/api.py +265 -0
- pyfect/http_api/builder.py +447 -0
- pyfect/http_api/client.py +474 -0
- pyfect/http_api/endpoint.py +969 -0
- pyfect/http_api/error.py +280 -0
- pyfect/http_api/group.py +215 -0
- pyfect/http_api/middleware.py +232 -0
- pyfect/http_api/openapi.py +396 -0
- pyfect/http_api/scalar.py +143 -0
- pyfect/http_api/schema.py +713 -0
- pyfect/http_api/security.py +91 -0
- pyfect/http_api/swagger.py +69 -0
- pyfect/http_api/test.py +105 -0
- pyfect/json_patch.py +255 -0
- pyfect/json_pointer.py +100 -0
- pyfect/json_schema/__init__.py +66 -0
- pyfect/json_schema/_api.py +96 -0
- pyfect/json_schema/_converter.py +258 -0
- pyfect/json_schema/_draft04.py +198 -0
- pyfect/json_schema/_draft07.py +203 -0
- pyfect/json_schema/_keywords.py +202 -0
- pyfect/json_schema/_model.py +201 -0
- pyfect/json_schema/_openapi.py +208 -0
- pyfect/latch.py +127 -0
- pyfect/layer.py +479 -0
- pyfect/layer_map.py +279 -0
- pyfect/layer_ref.py +208 -0
- pyfect/logger/__init__.py +91 -0
- pyfect/logger/_batched.py +49 -0
- pyfect/logger/_console.py +116 -0
- pyfect/logger/_file.py +31 -0
- pyfect/logger/_formats.py +104 -0
- pyfect/logger/_layer.py +50 -0
- pyfect/logger/_level.py +53 -0
- pyfect/logger/_log.py +142 -0
- pyfect/logger/_model.py +163 -0
- pyfect/logger/_refs.py +78 -0
- pyfect/logger/_stdlib.py +54 -0
- pyfect/logger/_text.py +263 -0
- pyfect/logger/_tracer.py +35 -0
- pyfect/managed_runtime.py +393 -0
- pyfect/metric/__init__.py +119 -0
- pyfect/metric/_boundaries.py +27 -0
- pyfect/metric/_kinds.py +417 -0
- pyfect/metric/_metric.py +302 -0
- pyfect/metric/_prometheus.py +150 -0
- pyfect/metric/_refs.py +78 -0
- pyfect/metric/_runtime.py +75 -0
- pyfect/metric/_snapshot.py +97 -0
- pyfect/metric/_state.py +74 -0
- pyfect/msgspec.py +98 -0
- pyfect/mutable_hash_map.py +140 -0
- pyfect/mutable_hash_set.py +95 -0
- pyfect/net/__init__.py +18 -0
- pyfect/net/_errors.py +29 -0
- pyfect/net/address_resolver.py +179 -0
- pyfect/net/dns/__init__.py +92 -0
- pyfect/net/dns/_error.py +43 -0
- pyfect/net/dns/_records.py +335 -0
- pyfect/net/dns/_service.py +287 -0
- pyfect/net/dns/_system.py +89 -0
- pyfect/net/host.py +311 -0
- pyfect/net/ip_interface.py +202 -0
- pyfect/net/ip_network.py +250 -0
- pyfect/net/net_address/__init__.py +245 -0
- pyfect/net/net_address/_inet.py +412 -0
- pyfect/net/net_address/_ip.py +348 -0
- pyfect/net/net_address/_mac.py +119 -0
- pyfect/optic/__init__.py +50 -0
- pyfect/optic/_builtins.py +90 -0
- pyfect/optic/_data.py +97 -0
- pyfect/optic/_optic.py +422 -0
- pyfect/optic/_steps.py +151 -0
- pyfect/option.py +362 -0
- pyfect/otel/__init__.py +22 -0
- pyfect/otel/_attributes.py +67 -0
- pyfect/otel/_producer.py +322 -0
- pyfect/otel/_shutdown.py +26 -0
- pyfect/otel/_span.py +292 -0
- pyfect/otel/_wrapper.py +92 -0
- pyfect/otel/logger.py +154 -0
- pyfect/otel/metrics.py +126 -0
- pyfect/otel/resource.py +111 -0
- pyfect/otel/sdk.py +195 -0
- pyfect/otel/tracer.py +157 -0
- pyfect/partitioned_semaphore.py +211 -0
- pyfect/path.py +304 -0
- pyfect/persistence/__init__.py +77 -0
- pyfect/persistence/_columns.py +32 -0
- pyfect/persistence/_error.py +14 -0
- pyfect/persistence/_seconds.py +68 -0
- pyfect/persistence/_sql_keys.py +47 -0
- pyfect/persistence/circuit_breaker/__init__.py +120 -0
- pyfect/persistence/circuit_breaker/_breaker.py +317 -0
- pyfect/persistence/circuit_breaker/_errors.py +67 -0
- pyfect/persistence/circuit_breaker/_memory.py +243 -0
- pyfect/persistence/circuit_breaker/_redis.py +509 -0
- pyfect/persistence/circuit_breaker/_store.py +60 -0
- pyfect/persistence/circuit_breaker/_types.py +191 -0
- pyfect/persistence/key_value_store/__init__.py +51 -0
- pyfect/persistence/key_value_store/_layers.py +216 -0
- pyfect/persistence/key_value_store/_schema_store.py +91 -0
- pyfect/persistence/key_value_store/_sql.py +197 -0
- pyfect/persistence/key_value_store/_store.py +212 -0
- pyfect/persistence/persistable.py +164 -0
- pyfect/persistence/persisted_cache.py +159 -0
- pyfect/persistence/persisted_queue/__init__.py +75 -0
- pyfect/persistence/persisted_queue/_error.py +42 -0
- pyfect/persistence/persisted_queue/_mailbox.py +101 -0
- pyfect/persistence/persisted_queue/_memory.py +182 -0
- pyfect/persistence/persisted_queue/_queue.py +272 -0
- pyfect/persistence/persisted_queue/_redis.py +455 -0
- pyfect/persistence/persisted_queue/_redis_scripts.py +395 -0
- pyfect/persistence/persisted_queue/_sql.py +334 -0
- pyfect/persistence/persisted_queue/_sql_migrations.py +319 -0
- pyfect/persistence/persisted_queue/_sql_statements.py +441 -0
- pyfect/persistence/persisted_queue/_store.py +147 -0
- pyfect/persistence/persistence/__init__.py +84 -0
- pyfect/persistence/persistence/_kvs.py +135 -0
- pyfect/persistence/persistence/_memory.py +62 -0
- pyfect/persistence/persistence/_redis.py +162 -0
- pyfect/persistence/persistence/_service.py +276 -0
- pyfect/persistence/persistence/_sql.py +499 -0
- pyfect/persistence/rate_limiter/__init__.py +82 -0
- pyfect/persistence/rate_limiter/_errors.py +59 -0
- pyfect/persistence/rate_limiter/_limiter.py +325 -0
- pyfect/persistence/rate_limiter/_memory.py +217 -0
- pyfect/persistence/rate_limiter/_redis.py +485 -0
- pyfect/persistence/rate_limiter/_store.py +89 -0
- pyfect/persistence/redis/__init__.py +52 -0
- pyfect/persistence/redis/_error.py +12 -0
- pyfect/persistence/redis/_make.py +234 -0
- pyfect/persistence/redis/_script.py +78 -0
- pyfect/persistence/redis/_service.py +104 -0
- pyfect/persistence/redis_py.py +303 -0
- pyfect/persistence/user_directories.py +195 -0
- pyfect/platform.py +26 -0
- pyfect/platform_error.py +256 -0
- pyfect/pool/__init__.py +36 -0
- pyfect/pool/_pool.py +707 -0
- pyfect/pool/_state.py +161 -0
- pyfect/pool/_ttl.py +105 -0
- pyfect/pool/_waiters.py +116 -0
- pyfect/primary_key.py +35 -0
- pyfect/process/__init__.py +82 -0
- pyfect/process/_child.py +213 -0
- pyfect/process/_command.py +358 -0
- pyfect/process/_handle.py +77 -0
- pyfect/process/_live.py +455 -0
- pyfect/process/_os.py +78 -0
- pyfect/process/_pipes.py +291 -0
- pyfect/process/_spawner.py +75 -0
- pyfect/pubsub/__init__.py +33 -0
- pyfect/pubsub/_atomic.py +181 -0
- pyfect/pubsub/_hub.py +214 -0
- pyfect/pubsub/_pubsub.py +214 -0
- pyfect/pubsub/_replay.py +107 -0
- pyfect/pubsub/_subscription.py +226 -0
- pyfect/pull.py +132 -0
- pyfect/py.typed +0 -0
- pyfect/pydantic.py +188 -0
- pyfect/queue/__init__.py +33 -0
- pyfect/queue/_queue.py +525 -0
- pyfect/queue/_state.py +111 -0
- pyfect/queue/_views.py +55 -0
- pyfect/random.py +146 -0
- pyfect/rc_map.py +363 -0
- pyfect/rc_ref.py +271 -0
- pyfect/reactivity/__init__.py +53 -0
- pyfect/reactivity/_async_result_schema.py +227 -0
- pyfect/reactivity/_hydration.py +18 -0
- pyfect/reactivity/_live.py +21 -0
- pyfect/reactivity/_reactivity.py +256 -0
- pyfect/reactivity/_result_effect.py +17 -0
- pyfect/reactivity/_time.py +11 -0
- pyfect/reactivity/async_result.py +679 -0
- pyfect/reactivity/atom/__init__.py +143 -0
- pyfect/reactivity/atom/_combinators.py +223 -0
- pyfect/reactivity/atom/_constructors.py +80 -0
- pyfect/reactivity/atom/_conversions.py +79 -0
- pyfect/reactivity/atom/_effect.py +252 -0
- pyfect/reactivity/atom/_fn.py +341 -0
- pyfect/reactivity/atom/_kvs.py +101 -0
- pyfect/reactivity/atom/_optimistic.py +162 -0
- pyfect/reactivity/atom/_runtime.py +323 -0
- pyfect/reactivity/atom/_serializable.py +65 -0
- pyfect/reactivity/atom/_subscription_ref.py +102 -0
- pyfect/reactivity/atom/_swr.py +69 -0
- pyfect/reactivity/atom/_types.py +435 -0
- pyfect/reactivity/atom_http_api.py +667 -0
- pyfect/reactivity/atom_ref.py +260 -0
- pyfect/reactivity/atom_registry/__init__.py +219 -0
- pyfect/reactivity/atom_registry/_batch.py +96 -0
- pyfect/reactivity/atom_registry/_dispatcher.py +60 -0
- pyfect/reactivity/atom_registry/_lifetime.py +202 -0
- pyfect/reactivity/atom_registry/_node.py +423 -0
- pyfect/reactivity/atom_registry/_registry.py +380 -0
- pyfect/reactivity/atom_rpc.py +283 -0
- pyfect/reactivity/hydration.py +200 -0
- pyfect/redacted.py +83 -0
- pyfect/ref.py +129 -0
- pyfect/resource.py +98 -0
- pyfect/result.py +306 -0
- pyfect/rpc/__init__.py +64 -0
- pyfect/rpc/_encoded_schema.py +86 -0
- pyfect/rpc/_json_rpc.py +246 -0
- pyfect/rpc/_methods.py +177 -0
- pyfect/rpc/_rpc.py +498 -0
- pyfect/rpc/client/__init__.py +100 -0
- pyfect/rpc/client/_core.py +449 -0
- pyfect/rpc/client/_http.py +257 -0
- pyfect/rpc/client/_protocol.py +171 -0
- pyfect/rpc/client/_serialized.py +321 -0
- pyfect/rpc/client/_transports.py +328 -0
- pyfect/rpc/client/_worker.py +283 -0
- pyfect/rpc/client_error.py +35 -0
- pyfect/rpc/group.py +296 -0
- pyfect/rpc/message.py +299 -0
- pyfect/rpc/middleware.py +182 -0
- pyfect/rpc/schema.py +89 -0
- pyfect/rpc/serialization.py +366 -0
- pyfect/rpc/server/__init__.py +181 -0
- pyfect/rpc/server/_core.py +549 -0
- pyfect/rpc/server/_protocol.py +123 -0
- pyfect/rpc/server/_serialized.py +363 -0
- pyfect/rpc/server/_stdio.py +101 -0
- pyfect/rpc/server/_transports.py +427 -0
- pyfect/rpc/server/_worker.py +108 -0
- pyfect/rpc/test.py +74 -0
- pyfect/rpc/worker.py +92 -0
- pyfect/run.py +393 -0
- pyfect/schedule/__init__.py +91 -0
- pyfect/schedule/_combinators.py +168 -0
- pyfect/schedule/_constructors.py +223 -0
- pyfect/schedule/_cron.py +49 -0
- pyfect/schedule/_drive.py +191 -0
- pyfect/schedule/_model.py +405 -0
- pyfect/scheduler.py +575 -0
- pyfect/schema/__init__.py +866 -0
- pyfect/schema/_adapters.py +39 -0
- pyfect/schema/_arbitrary_constraint.py +242 -0
- pyfect/schema/_arbitrary_graph.py +325 -0
- pyfect/schema/_arbitrary_links.py +179 -0
- pyfect/schema/_arbitrary_number.py +112 -0
- pyfect/schema/_arrays.py +287 -0
- pyfect/schema/_assignable.py +118 -0
- pyfect/schema/_boolean_literals.py +31 -0
- pyfect/schema/_checks_number.py +216 -0
- pyfect/schema/_checks_order.py +140 -0
- pyfect/schema/_checks_size.py +277 -0
- pyfect/schema/_checks_string.py +241 -0
- pyfect/schema/_class.py +569 -0
- pyfect/schema/_codec_derivation.py +227 -0
- pyfect/schema/_codec_links.py +221 -0
- pyfect/schema/_codecs.py +316 -0
- pyfect/schema/_collections.py +172 -0
- pyfect/schema/_constructor_default.py +89 -0
- pyfect/schema/_data.py +440 -0
- pyfect/schema/_data_codecs.py +145 -0
- pyfect/schema/_decimal_regexp.py +243 -0
- pyfect/schema/_declare.py +68 -0
- pyfect/schema/_derive.py +292 -0
- pyfect/schema/_duration_codecs.py +27 -0
- pyfect/schema/_ecma_pattern.py +416 -0
- pyfect/schema/_error.py +26 -0
- pyfect/schema/_error_instance.py +222 -0
- pyfect/schema/_graph.py +271 -0
- pyfect/schema/_instances.py +240 -0
- pyfect/schema/_jsonschema/__init__.py +90 -0
- pyfect/schema/_jsonschema/_annotations.py +114 -0
- pyfect/schema/_jsonschema/_checks.py +253 -0
- pyfect/schema/_jsonschema/_compile.py +346 -0
- pyfect/schema/_jsonschema/_exact.py +19 -0
- pyfect/schema/_jsonschema/_merge.py +211 -0
- pyfect/schema/_jsonschema/_objects.py +231 -0
- pyfect/schema/_jsonschema/_references.py +221 -0
- pyfect/schema/_jsonschema/_template.py +69 -0
- pyfect/schema/_literals.py +147 -0
- pyfect/schema/_middleware.py +209 -0
- pyfect/schema/_net.py +522 -0
- pyfect/schema/_options.py +49 -0
- pyfect/schema/_parse.py +351 -0
- pyfect/schema/_parsed_text.py +37 -0
- pyfect/schema/_primitives.py +48 -0
- pyfect/schema/_record.py +41 -0
- pyfect/schema/_refine.py +81 -0
- pyfect/schema/_schema.py +158 -0
- pyfect/schema/_semver.py +144 -0
- pyfect/schema/_strategy/__init__.py +258 -0
- pyfect/schema/_strategy/_big_decimal.py +203 -0
- pyfect/schema/_strategy/_composites.py +119 -0
- pyfect/schema/_strategy/_leaves.py +94 -0
- pyfect/schema/_strategy/_net.py +133 -0
- pyfect/schema/_struct.py +87 -0
- pyfect/schema/_struct_keys.py +128 -0
- pyfect/schema/_tagged.py +254 -0
- pyfect/schema/_template_literal.py +172 -0
- pyfect/schema/_to_codec_array_from_single.py +115 -0
- pyfect/schema/_to_codec_go_json.py +174 -0
- pyfect/schema/_to_codec_json.py +73 -0
- pyfect/schema/_to_codec_string_tree.py +115 -0
- pyfect/schema/_to_differ_json_patch.py +44 -0
- pyfect/schema/_to_encoder_xml.py +119 -0
- pyfect/schema/_to_equivalence.py +186 -0
- pyfect/schema/_to_formatter.py +180 -0
- pyfect/schema/_to_iso.py +73 -0
- pyfect/schema/_to_strategy.py +31 -0
- pyfect/schema/_transform.py +210 -0
- pyfect/schema/_tree.py +27 -0
- pyfect/schema/_type_walk.py +53 -0
- pyfect/schema/_undefined.py +58 -0
- pyfect/schema/_value_codecs.py +84 -0
- pyfect/schema/_values.py +123 -0
- pyfect/schema/_void.py +86 -0
- pyfect/schema/ast/__init__.py +227 -0
- pyfect/schema/ast/_annotations.py +43 -0
- pyfect/schema/ast/_arrays.py +210 -0
- pyfect/schema/ast/_base.py +295 -0
- pyfect/schema/ast/_candidates.py +309 -0
- pyfect/schema/ast/_check.py +181 -0
- pyfect/schema/ast/_class.py +159 -0
- pyfect/schema/ast/_constructors.py +65 -0
- pyfect/schema/ast/_declaration.py +97 -0
- pyfect/schema/ast/_encoding.py +111 -0
- pyfect/schema/ast/_json.py +129 -0
- pyfect/schema/ast/_keywords.py +125 -0
- pyfect/schema/ast/_literal.py +157 -0
- pyfect/schema/ast/_objects.py +356 -0
- pyfect/schema/ast/_record.py +36 -0
- pyfect/schema/ast/_result.py +116 -0
- pyfect/schema/ast/_steps.py +209 -0
- pyfect/schema/ast/_suspend.py +65 -0
- pyfect/schema/ast/_template_literal.py +256 -0
- pyfect/schema/ast/_union.py +172 -0
- pyfect/schema/getter/__init__.py +136 -0
- pyfect/schema/getter/_binary.py +104 -0
- pyfect/schema/getter/_coerce.py +196 -0
- pyfect/schema/getter/_getter.py +243 -0
- pyfect/schema/getter/_make.py +151 -0
- pyfect/schema/getter/_text.py +141 -0
- pyfect/schema/getter/_tree_record.py +170 -0
- pyfect/schema/issue.py +420 -0
- pyfect/schema/model/__init__.py +120 -0
- pyfect/schema/model/_class.py +245 -0
- pyfect/schema/model/_dates.py +103 -0
- pyfect/schema/model/_fields.py +77 -0
- pyfect/schema/model/_toolkit.py +114 -0
- pyfect/schema/model/_uuid.py +69 -0
- pyfect/schema/representation/__init__.py +424 -0
- pyfect/schema/representation/_builtin_revivers.py +331 -0
- pyfect/schema/representation/_code.py +106 -0
- pyfect/schema/representation/_code_builtins.py +281 -0
- pyfect/schema/representation/_code_document.py +563 -0
- pyfect/schema/representation/_code_sort.py +111 -0
- pyfect/schema/representation/_from.py +278 -0
- pyfect/schema/representation/_import/__init__.py +78 -0
- pyfect/schema/representation/_import/_finalize.py +121 -0
- pyfect/schema/representation/_import/_intersect.py +467 -0
- pyfect/schema/representation/_import/_js_order.py +66 -0
- pyfect/schema/representation/_import/_keywords.py +224 -0
- pyfect/schema/representation/_import/_nodes.py +199 -0
- pyfect/schema/representation/_import/_translate.py +455 -0
- pyfect/schema/representation/_json.py +520 -0
- pyfect/schema/representation/_json_schema.py +178 -0
- pyfect/schema/representation/_model.py +284 -0
- pyfect/schema/representation/_revivers.py +89 -0
- pyfect/schema/representation/_to.py +249 -0
- pyfect/schema/transformation/__init__.py +99 -0
- pyfect/schema/transformation/_builtins.py +314 -0
- pyfect/schema/transformation/_durations.py +150 -0
- pyfect/schema/transformation/_transformation.py +177 -0
- pyfect/schema/variant_schema/__init__.py +69 -0
- pyfect/schema/variant_schema/_class.py +188 -0
- pyfect/schema/variant_schema/_defaults.py +30 -0
- pyfect/schema/variant_schema/_field.py +32 -0
- pyfect/schema/variant_schema/_struct.py +130 -0
- pyfect/schema/variant_schema/_toolkit.py +160 -0
- pyfect/scope.py +374 -0
- pyfect/scoped_ref.py +124 -0
- pyfect/semaphore.py +276 -0
- pyfect/semver.py +403 -0
- pyfect/semver_range.py +420 -0
- pyfect/sink/__init__.py +131 -0
- pyfect/sink/_combinators.py +151 -0
- pyfect/sink/_constructors.py +187 -0
- pyfect/sink/_context.py +87 -0
- pyfect/sink/_folds.py +336 -0
- pyfect/sink/_for_each.py +87 -0
- pyfect/sink/_sink.py +265 -0
- pyfect/sink/_steps.py +34 -0
- pyfect/sink/_taking.py +183 -0
- pyfect/socket/__init__.py +146 -0
- pyfect/socket/_accepted.py +65 -0
- pyfect/socket/_connection.py +384 -0
- pyfect/socket/_errors.py +135 -0
- pyfect/socket/_loop.py +66 -0
- pyfect/socket/_net.py +281 -0
- pyfect/socket/_socket.py +245 -0
- pyfect/socket/_tls.py +88 -0
- pyfect/socket/_tls_session.py +117 -0
- pyfect/socket/_web_socket.py +555 -0
- pyfect/socket/datagram/__init__.py +110 -0
- pyfect/socket/datagram/_core.py +460 -0
- pyfect/socket/datagram/_errors.py +102 -0
- pyfect/socket/datagram/_live.py +675 -0
- pyfect/socket/datagram/_model.py +217 -0
- pyfect/socket/server.py +258 -0
- pyfect/socket/ws.py +484 -0
- pyfect/sql/__init__.py +60 -0
- pyfect/sql/_loop.py +134 -0
- pyfect/sql/clickhouse/__init__.py +74 -0
- pyfect/sql/clickhouse/_connection.py +237 -0
- pyfect/sql/clickhouse/_rows.py +87 -0
- pyfect/sql/clickhouse/client.py +322 -0
- pyfect/sql/clickhouse/config.py +241 -0
- pyfect/sql/clickhouse/driver.py +434 -0
- pyfect/sql/clickhouse/error.py +106 -0
- pyfect/sql/clickhouse/migrator.py +47 -0
- pyfect/sql/clickhouse/params.py +168 -0
- pyfect/sql/client/__init__.py +35 -0
- pyfect/sql/client/_client.py +309 -0
- pyfect/sql/client/_sqlite_acquirers.py +130 -0
- pyfect/sql/client/_transaction.py +410 -0
- pyfect/sql/connection.py +87 -0
- pyfect/sql/d1/__init__.py +48 -0
- pyfect/sql/d1/api.py +270 -0
- pyfect/sql/d1/client.py +344 -0
- pyfect/sql/d1/config.py +55 -0
- pyfect/sql/error.py +311 -0
- pyfect/sql/libsql/__init__.py +58 -0
- pyfect/sql/libsql/_connection.py +110 -0
- pyfect/sql/libsql/_hrana.py +258 -0
- pyfect/sql/libsql/_local.py +251 -0
- pyfect/sql/libsql/_targets.py +28 -0
- pyfect/sql/libsql/_values.py +46 -0
- pyfect/sql/libsql/client.py +225 -0
- pyfect/sql/libsql/config.py +170 -0
- pyfect/sql/libsql/migrator.py +43 -0
- pyfect/sql/libsql/sdk.py +123 -0
- pyfect/sql/migrator.py +434 -0
- pyfect/sql/model.py +324 -0
- pyfect/sql/mssql/__init__.py +79 -0
- pyfect/sql/mssql/_connection.py +169 -0
- pyfect/sql/mssql/_driver.py +103 -0
- pyfect/sql/mssql/_names.py +9 -0
- pyfect/sql/mssql/_pool.py +125 -0
- pyfect/sql/mssql/_tds.py +447 -0
- pyfect/sql/mssql/_tds_params.py +155 -0
- pyfect/sql/mssql/_worker.py +155 -0
- pyfect/sql/mssql/client.py +344 -0
- pyfect/sql/mssql/config.py +118 -0
- pyfect/sql/mssql/error.py +111 -0
- pyfect/sql/mssql/migrator.py +43 -0
- pyfect/sql/mssql/parameter.py +44 -0
- pyfect/sql/mssql/procedure.py +119 -0
- pyfect/sql/mssql/types.py +137 -0
- pyfect/sql/mysql/__init__.py +60 -0
- pyfect/sql/mysql/_asyncmy.py +168 -0
- pyfect/sql/mysql/_bind.py +76 -0
- pyfect/sql/mysql/_connection.py +156 -0
- pyfect/sql/mysql/_pool.py +63 -0
- pyfect/sql/mysql/_session.py +262 -0
- pyfect/sql/mysql/client.py +244 -0
- pyfect/sql/mysql/config.py +108 -0
- pyfect/sql/mysql/error.py +104 -0
- pyfect/sql/mysql/migrator.py +44 -0
- pyfect/sql/pg/__init__.py +82 -0
- pyfect/sql/pg/_resolve.py +284 -0
- pyfect/sql/pg/_session.py +380 -0
- pyfect/sql/pg/client.py +398 -0
- pyfect/sql/pg/config.py +142 -0
- pyfect/sql/pg/connection.py +585 -0
- pyfect/sql/pg/error.py +131 -0
- pyfect/sql/pg/migrator.py +139 -0
- pyfect/sql/pg/pool.py +265 -0
- pyfect/sql/resolver.py +294 -0
- pyfect/sql/result.py +24 -0
- pyfect/sql/schema.py +116 -0
- pyfect/sql/sqlite/__init__.py +26 -0
- pyfect/sql/sqlite/_connection.py +230 -0
- pyfect/sql/sqlite/_worker.py +71 -0
- pyfect/sql/sqlite/client.py +248 -0
- pyfect/sql/sqlite/migrator.py +43 -0
- pyfect/sql/statement/__init__.py +144 -0
- pyfect/sql/statement/_compiler.py +384 -0
- pyfect/sql/statement/_segments.py +487 -0
- pyfect/sql/statement/_statement.py +526 -0
- pyfect/sql/statement/_transforms.py +93 -0
- pyfect/sql/stream.py +88 -0
- pyfect/stdio.py +206 -0
- pyfect/stream/__init__.py +151 -0
- pyfect/stream/_async_iterable.py +129 -0
- pyfect/stream/_base.py +69 -0
- pyfect/stream/_bridge_methods.py +139 -0
- pyfect/stream/_catch.py +135 -0
- pyfect/stream/_catch_reason.py +558 -0
- pyfect/stream/_channel_methods.py +61 -0
- pyfect/stream/_chunk_methods.py +169 -0
- pyfect/stream/_chunking.py +310 -0
- pyfect/stream/_concat.py +65 -0
- pyfect/stream/_constructors.py +341 -0
- pyfect/stream/_context.py +147 -0
- pyfect/stream/_error_methods.py +254 -0
- pyfect/stream/_execution_plan.py +199 -0
- pyfect/stream/_fanout.py +154 -0
- pyfect/stream/_fanout_methods.py +183 -0
- pyfect/stream/_filter_methods.py +203 -0
- pyfect/stream/_grouping.py +168 -0
- pyfect/stream/_many.py +65 -0
- pyfect/stream/_map_methods.py +233 -0
- pyfect/stream/_mapping.py +126 -0
- pyfect/stream/_run.py +114 -0
- pyfect/stream/_run_methods.py +114 -0
- pyfect/stream/_service_methods.py +36 -0
- pyfect/stream/_sink_methods.py +151 -0
- pyfect/stream/_sinking.py +277 -0
- pyfect/stream/_sources.py +124 -0
- pyfect/stream/_stream.py +162 -0
- pyfect/stream/_taking.py +330 -0
- pyfect/stream/_text.py +43 -0
- pyfect/stream/_timing.py +294 -0
- pyfect/stream/_timing_methods.py +252 -0
- pyfect/stream/_zip_methods.py +279 -0
- pyfect/stream/_zipping.py +213 -0
- pyfect/subscription_ref.py +66 -0
- pyfect/synchronized_ref.py +38 -0
- pyfect/take.py +36 -0
- pyfect/terminal/__init__.py +41 -0
- pyfect/terminal/_keys.py +113 -0
- pyfect/terminal/_live.py +223 -0
- pyfect/terminal/_service.py +73 -0
- pyfect/testing/__init__.py +122 -0
- pyfect/testing/_assertions.py +92 -0
- pyfect/testing/_clock.py +309 -0
- pyfect/testing/_console.py +164 -0
- pyfect/testing/_crypto.py +44 -0
- pyfect/testing/_decorators.py +211 -0
- pyfect/testing/_flaky.py +34 -0
- pyfect/testing/_layers.py +153 -0
- pyfect/testing/_prop.py +255 -0
- pyfect/testing/_runners.py +84 -0
- pyfect/testing/_runtime.py +174 -0
- pyfect/testing/_schema.py +238 -0
- pyfect/testing/plugin.py +59 -0
- pyfect/tracer/__init__.py +107 -0
- pyfect/tracer/_model.py +176 -0
- pyfect/tracer/_native.py +172 -0
- pyfect/tracer/_refs.py +87 -0
- pyfect/tracer/_spans.py +269 -0
- pyfect/transaction.py +340 -0
- pyfect/trie.py +317 -0
- pyfect/tx_chunk.py +127 -0
- pyfect/tx_deferred.py +75 -0
- pyfect/tx_hash_map.py +265 -0
- pyfect/tx_hash_set.py +151 -0
- pyfect/tx_priority_queue.py +224 -0
- pyfect/tx_pubsub.py +194 -0
- pyfect/tx_queue/__init__.py +27 -0
- pyfect/tx_queue/_queue.py +359 -0
- pyfect/tx_queue/_views.py +60 -0
- pyfect/tx_reentrant_lock.py +201 -0
- pyfect/tx_ref.py +79 -0
- pyfect/tx_semaphore.py +136 -0
- pyfect/tx_subscription_ref.py +118 -0
- pyfect/workers/__init__.py +62 -0
- pyfect/workers/_frames.py +82 -0
- pyfect/workers/_mailbox.py +51 -0
- pyfect/workers/_port.py +106 -0
- pyfect/workers/_process.py +155 -0
- pyfect/workers/_thread.py +100 -0
- pyfect/workers/python_worker.py +133 -0
- pyfect/workers/python_worker_runner.py +112 -0
- pyfect/workers/transferable.py +103 -0
- pyfect/workers/worker.py +273 -0
- pyfect/workers/worker_error.py +86 -0
- pyfect/workers/worker_runner.py +85 -0
- pyfect/workflow/__init__.py +78 -0
- pyfect/workflow/_crypto.py +23 -0
- pyfect/workflow/_encoded.py +381 -0
- pyfect/workflow/_memory.py +424 -0
- pyfect/workflow/_result.py +43 -0
- pyfect/workflow/_services.py +241 -0
- pyfect/workflow/_suspension.py +153 -0
- pyfect/workflow/activity.py +427 -0
- pyfect/workflow/durable_clock.py +72 -0
- pyfect/workflow/durable_deferred.py +403 -0
- pyfect/workflow/durable_queue.py +259 -0
- pyfect/workflow/workflow.py +509 -0
- pyfect/workflow/workflow_engine.py +42 -0
- pyfect/workflow/workflow_proxy.py +161 -0
- pyfect/workflow/workflow_proxy_server.py +146 -0
- pyfect_dev-0.1.0.dist-info/METADATA +296 -0
- pyfect_dev-0.1.0.dist-info/RECORD +1265 -0
- pyfect_dev-0.1.0.dist-info/WHEEL +4 -0
- pyfect_dev-0.1.0.dist-info/entry_points.txt +3 -0
- pyfect_dev-0.1.0.dist-info/licenses/LICENSE +21 -0
|
@@ -0,0 +1,155 @@
|
|
|
1
|
+
"""Parameters as pytds sends them: each ``DataType`` as a pytds SQL type,
|
|
2
|
+
sized as tedious sizes it, and each value in the form its serializer
|
|
3
|
+
takes."""
|
|
4
|
+
|
|
5
|
+
from __future__ import annotations
|
|
6
|
+
|
|
7
|
+
import uuid
|
|
8
|
+
from collections.abc import Callable
|
|
9
|
+
from datetime import UTC, date, datetime
|
|
10
|
+
from decimal import Decimal
|
|
11
|
+
from typing import Any, cast
|
|
12
|
+
|
|
13
|
+
from pytds import tds_types # pyright: ignore[reportMissingTypeStubs]
|
|
14
|
+
|
|
15
|
+
from pyfect.sql.mssql._driver import TypedValue
|
|
16
|
+
from pyfect.sql.mssql.types import DataTypeName, ParameterOptions
|
|
17
|
+
|
|
18
|
+
_DEFAULT_PRECISION = 18
|
|
19
|
+
_DEFAULT_SCALE = 0
|
|
20
|
+
_DEFAULT_TIME_SCALE = 7
|
|
21
|
+
|
|
22
|
+
|
|
23
|
+
def _utf16_units(text: str) -> int:
|
|
24
|
+
"""``text``'s length as JavaScript counts it (UTF-16 code units), the
|
|
25
|
+
unit SQL Server sizes ``NVARCHAR`` in."""
|
|
26
|
+
return len(text.encode("utf-16-le", "surrogatepass")) // 2
|
|
27
|
+
|
|
28
|
+
|
|
29
|
+
def _length(options: ParameterOptions, value: object, maximum: int) -> int | None:
|
|
30
|
+
"""The length tedious gives the type: the options', else the value's
|
|
31
|
+
(at least 1), else the type's ``maximum`` (a null, as an output
|
|
32
|
+
parameter's); ``None`` for ``MAX``."""
|
|
33
|
+
if options.length == "max":
|
|
34
|
+
return None
|
|
35
|
+
if options.length is not None:
|
|
36
|
+
return options.length
|
|
37
|
+
if value is None:
|
|
38
|
+
return maximum
|
|
39
|
+
if isinstance(value, memoryview):
|
|
40
|
+
return cast("memoryview[int]", value).nbytes or 1
|
|
41
|
+
if isinstance(value, bytes | bytearray):
|
|
42
|
+
return len(value) or 1
|
|
43
|
+
size = _utf16_units(value if isinstance(value, str) else str(value))
|
|
44
|
+
return size or 1
|
|
45
|
+
|
|
46
|
+
|
|
47
|
+
def _variable(
|
|
48
|
+
sized: Callable[[int], Any], unbounded: Callable[[], Any], maximum: int
|
|
49
|
+
) -> Callable[[ParameterOptions, object], Any]:
|
|
50
|
+
def make(options: ParameterOptions, value: object) -> Any:
|
|
51
|
+
length = _length(options, value, maximum)
|
|
52
|
+
return unbounded() if length is None or length > maximum else sized(length)
|
|
53
|
+
|
|
54
|
+
return make
|
|
55
|
+
|
|
56
|
+
|
|
57
|
+
def _fixed(sized: Callable[[int], Any], maximum: int) -> Callable[[ParameterOptions, object], Any]:
|
|
58
|
+
return lambda options, value: sized(_length(options, value, maximum) or maximum)
|
|
59
|
+
|
|
60
|
+
|
|
61
|
+
def _plain(make: Callable[[], Any]) -> Callable[[ParameterOptions, object], Any]:
|
|
62
|
+
return lambda _options, _value: make()
|
|
63
|
+
|
|
64
|
+
|
|
65
|
+
def _decimal(options: ParameterOptions, _value: object) -> Any:
|
|
66
|
+
return tds_types.DecimalType(
|
|
67
|
+
precision=options.precision or _DEFAULT_PRECISION,
|
|
68
|
+
scale=_DEFAULT_SCALE if options.scale is None else options.scale,
|
|
69
|
+
)
|
|
70
|
+
|
|
71
|
+
|
|
72
|
+
def _scaled(make: Callable[[int], Any]) -> Callable[[ParameterOptions, object], Any]:
|
|
73
|
+
return lambda options, _value: make(
|
|
74
|
+
_DEFAULT_TIME_SCALE if options.scale is None else options.scale
|
|
75
|
+
)
|
|
76
|
+
|
|
77
|
+
|
|
78
|
+
_SQL_TYPES: dict[DataTypeName, Callable[[ParameterOptions, object], Any]] = {
|
|
79
|
+
"Bit": _plain(tds_types.BitType),
|
|
80
|
+
"TinyInt": _plain(tds_types.TinyIntType),
|
|
81
|
+
"SmallInt": _plain(tds_types.SmallIntType),
|
|
82
|
+
"Int": _plain(tds_types.IntType),
|
|
83
|
+
"BigInt": _plain(tds_types.BigIntType),
|
|
84
|
+
"Real": _plain(tds_types.RealType),
|
|
85
|
+
"Float": _plain(tds_types.FloatType),
|
|
86
|
+
"SmallMoney": _plain(tds_types.SmallMoneyType),
|
|
87
|
+
"Money": _plain(tds_types.MoneyType),
|
|
88
|
+
"Decimal": _decimal,
|
|
89
|
+
"Numeric": _decimal,
|
|
90
|
+
"Char": _fixed(tds_types.CharType, 8000),
|
|
91
|
+
"VarChar": _variable(tds_types.VarCharType, tds_types.VarCharMaxType, 8000),
|
|
92
|
+
"NChar": _fixed(tds_types.NCharType, 4000),
|
|
93
|
+
"NVarChar": _variable(tds_types.NVarCharType, tds_types.NVarCharMaxType, 4000),
|
|
94
|
+
"Text": _plain(tds_types.TextType),
|
|
95
|
+
"NText": _plain(tds_types.NTextType),
|
|
96
|
+
"Binary": _fixed(tds_types.BinaryType, 8000),
|
|
97
|
+
"VarBinary": _variable(tds_types.VarBinaryType, tds_types.VarBinaryMaxType, 8000),
|
|
98
|
+
"Image": _plain(tds_types.ImageType),
|
|
99
|
+
"UniqueIdentifier": _plain(tds_types.UniqueIdentifierType),
|
|
100
|
+
"Xml": _plain(tds_types.XmlType),
|
|
101
|
+
"Date": _plain(tds_types.DateType),
|
|
102
|
+
"Time": _scaled(lambda scale: tds_types.TimeType(precision=scale)),
|
|
103
|
+
"SmallDateTime": _plain(tds_types.SmallDateTimeType),
|
|
104
|
+
"DateTime": _plain(tds_types.DateTimeType),
|
|
105
|
+
"DateTime2": _scaled(lambda scale: tds_types.DateTime2Type(precision=scale)),
|
|
106
|
+
"DateTimeOffset": _scaled(lambda scale: tds_types.DateTimeOffsetType(precision=scale)),
|
|
107
|
+
}
|
|
108
|
+
|
|
109
|
+
_TEXT: frozenset[DataTypeName] = frozenset(
|
|
110
|
+
{"Char", "VarChar", "NChar", "NVarChar", "Text", "NText", "Xml"}
|
|
111
|
+
)
|
|
112
|
+
_BINARY: frozenset[DataTypeName] = frozenset({"Binary", "VarBinary", "Image"})
|
|
113
|
+
_DECIMAL: frozenset[DataTypeName] = frozenset({"Decimal", "Numeric", "SmallMoney", "Money"})
|
|
114
|
+
_UTC_WALL_CLOCK: frozenset[DataTypeName] = frozenset({"SmallDateTime", "DateTime", "DateTime2"})
|
|
115
|
+
|
|
116
|
+
|
|
117
|
+
def _value(name: DataTypeName, value: object) -> object: # noqa: PLR0911
|
|
118
|
+
"""``value`` as the serializer of ``name`` takes it. As tedious (with
|
|
119
|
+
``useUTC``), an aware datetime is sent as its UTC wall clock, and a
|
|
120
|
+
naive one as a UTC time for ``DateTimeOffset``."""
|
|
121
|
+
if value is None:
|
|
122
|
+
return None
|
|
123
|
+
if name in _TEXT and not isinstance(value, str):
|
|
124
|
+
return str(value)
|
|
125
|
+
if name in _BINARY and isinstance(value, bytearray):
|
|
126
|
+
return bytes(value)
|
|
127
|
+
if name in _BINARY and isinstance(value, memoryview):
|
|
128
|
+
return cast("memoryview[int]", value).tobytes()
|
|
129
|
+
if name in _DECIMAL and isinstance(value, int | float) and not isinstance(value, bool):
|
|
130
|
+
return Decimal(str(value))
|
|
131
|
+
if name == "UniqueIdentifier" and isinstance(value, str):
|
|
132
|
+
return uuid.UUID(value)
|
|
133
|
+
if isinstance(value, datetime):
|
|
134
|
+
if name in _UTC_WALL_CLOCK and value.tzinfo is not None:
|
|
135
|
+
return value.astimezone(UTC).replace(tzinfo=None)
|
|
136
|
+
if name == "DateTimeOffset" and value.tzinfo is None:
|
|
137
|
+
return value.replace(tzinfo=UTC)
|
|
138
|
+
if name == "Date":
|
|
139
|
+
return value.date()
|
|
140
|
+
if name in _UTC_WALL_CLOCK and type(value) is date:
|
|
141
|
+
return datetime(value.year, value.month, value.day)
|
|
142
|
+
return value
|
|
143
|
+
|
|
144
|
+
|
|
145
|
+
def sql_type(parameter: TypedValue) -> Any:
|
|
146
|
+
"""The pytds SQL type ``parameter`` is declared as."""
|
|
147
|
+
return _SQL_TYPES[parameter.type.name](parameter.options, parameter.value)
|
|
148
|
+
|
|
149
|
+
|
|
150
|
+
def value(parameter: TypedValue) -> object:
|
|
151
|
+
"""The value of ``parameter`` as pytds writes it."""
|
|
152
|
+
return _value(parameter.type.name, parameter.value)
|
|
153
|
+
|
|
154
|
+
|
|
155
|
+
__all__ = ["sql_type", "value"]
|
|
@@ -0,0 +1,155 @@
|
|
|
1
|
+
"""The thread a SQL Server connection's blocking calls run on.
|
|
2
|
+
|
|
3
|
+
Each connection gets one worker thread: its calls run there in the order
|
|
4
|
+
sent while the fiber that sent one waits without blocking others.
|
|
5
|
+
Interrupting the fiber drops a call that has not started; one that has
|
|
6
|
+
gets ``cancel`` (called while it runs, never after) and the fiber waits
|
|
7
|
+
for it to end, so the connection is never used by two calls at once. A
|
|
8
|
+
call given ``release`` (one that opens something, with nothing else to
|
|
9
|
+
protect yet) is left to finish on the thread instead. When a call ends,
|
|
10
|
+
its ``Cancellable`` forgets any cancel asked for it, under the same lock,
|
|
11
|
+
so a cancel that came after its request ended never reaches the next
|
|
12
|
+
call's. A call after ``shutdown`` fails with an ``SqlError`` saying so.
|
|
13
|
+
"""
|
|
14
|
+
# pyright: reportPrivateUsage=false
|
|
15
|
+
|
|
16
|
+
from __future__ import annotations
|
|
17
|
+
|
|
18
|
+
import threading
|
|
19
|
+
from collections.abc import Callable
|
|
20
|
+
from concurrent.futures import Future, ThreadPoolExecutor
|
|
21
|
+
from typing import Literal, Protocol
|
|
22
|
+
|
|
23
|
+
from pyfect._core import Effect, _FailCause, _Succeed, _Suspend
|
|
24
|
+
from pyfect._released import Lifetime, Released
|
|
25
|
+
from pyfect.cause import die, fail, interrupt
|
|
26
|
+
from pyfect.process._os import Latch
|
|
27
|
+
from pyfect.sql.error import SqlError, UnknownError
|
|
28
|
+
|
|
29
|
+
type _Phase = Literal["queued", "running", "dropped", "abandoned", "done"]
|
|
30
|
+
|
|
31
|
+
|
|
32
|
+
def _outcome[A](
|
|
33
|
+
thunk: Callable[[], A], catch: Callable[[Exception], SqlError]
|
|
34
|
+
) -> Effect[A, SqlError]:
|
|
35
|
+
"""``thunk()`` as an effect; never raises."""
|
|
36
|
+
try:
|
|
37
|
+
return _Succeed(thunk())
|
|
38
|
+
except Exception as exc:
|
|
39
|
+
try:
|
|
40
|
+
return _FailCause(fail(catch(exc)))
|
|
41
|
+
except BaseException as defect:
|
|
42
|
+
return _FailCause(die(defect))
|
|
43
|
+
except BaseException as defect:
|
|
44
|
+
return _FailCause(die(defect))
|
|
45
|
+
|
|
46
|
+
|
|
47
|
+
class Cancellable(Protocol):
|
|
48
|
+
"""What an interrupted call is cancelled through."""
|
|
49
|
+
|
|
50
|
+
def cancel(self) -> None:
|
|
51
|
+
"""Stop the call's request: in flight, or about to be sent."""
|
|
52
|
+
...
|
|
53
|
+
|
|
54
|
+
def forget_cancel(self) -> None:
|
|
55
|
+
"""The call ended: a cancel asked for it no longer applies."""
|
|
56
|
+
...
|
|
57
|
+
|
|
58
|
+
|
|
59
|
+
class _Call[A]:
|
|
60
|
+
"""One call's progress; ``phase`` changes under ``lock``."""
|
|
61
|
+
|
|
62
|
+
__slots__ = ("lock", "phase", "settled")
|
|
63
|
+
|
|
64
|
+
def __init__(self) -> None:
|
|
65
|
+
self.lock = threading.Lock()
|
|
66
|
+
self.phase: _Phase = "queued"
|
|
67
|
+
self.settled: Latch[Effect[A, SqlError]] = Latch(waits_for="work")
|
|
68
|
+
|
|
69
|
+
def end(self) -> bool:
|
|
70
|
+
"""The call ended; whether its fiber had stopped waiting for it."""
|
|
71
|
+
abandoned = self.phase == "abandoned"
|
|
72
|
+
self.phase = "done"
|
|
73
|
+
return abandoned
|
|
74
|
+
|
|
75
|
+
|
|
76
|
+
class Worker:
|
|
77
|
+
"""One thread running a connection's calls in order."""
|
|
78
|
+
|
|
79
|
+
__slots__ = ("_executor", "_lifetime")
|
|
80
|
+
|
|
81
|
+
def __init__(self) -> None:
|
|
82
|
+
self._executor = ThreadPoolExecutor(1, thread_name_prefix="pyfect-mssql")
|
|
83
|
+
self._lifetime = Lifetime("SQL Server connection")
|
|
84
|
+
|
|
85
|
+
def call[A](
|
|
86
|
+
self,
|
|
87
|
+
thunk: Callable[[], A],
|
|
88
|
+
catch: Callable[[Exception], SqlError],
|
|
89
|
+
cancel: Cancellable | None = None,
|
|
90
|
+
release: Callable[[A], None] | None = None,
|
|
91
|
+
) -> Effect[A, SqlError]:
|
|
92
|
+
"""Run ``thunk`` on the thread; an exception it raises fails with
|
|
93
|
+
``catch`` of it. Interrupted while it runs, ``cancel`` runs and the
|
|
94
|
+
fiber waits for ``thunk`` to end. With ``release`` the fiber does not
|
|
95
|
+
wait: a value ``thunk`` makes all the same goes to ``release``."""
|
|
96
|
+
|
|
97
|
+
def start() -> Effect[A, SqlError]:
|
|
98
|
+
call: _Call[A] = _Call()
|
|
99
|
+
|
|
100
|
+
def run() -> None:
|
|
101
|
+
with call.lock:
|
|
102
|
+
if call.phase == "dropped":
|
|
103
|
+
call.settled.set(_FailCause(interrupt()))
|
|
104
|
+
return
|
|
105
|
+
call.phase = "running"
|
|
106
|
+
outcome = _outcome(thunk, catch)
|
|
107
|
+
with call.lock:
|
|
108
|
+
abandoned = call.end()
|
|
109
|
+
if cancel is not None:
|
|
110
|
+
cancel.forget_cancel()
|
|
111
|
+
call.settled.set(outcome)
|
|
112
|
+
if abandoned:
|
|
113
|
+
made(outcome)
|
|
114
|
+
|
|
115
|
+
def made(outcome: Effect[A, SqlError]) -> None:
|
|
116
|
+
if release is not None and isinstance(outcome, _Succeed):
|
|
117
|
+
release(outcome.value)
|
|
118
|
+
|
|
119
|
+
def released(done: Future[None]) -> None:
|
|
120
|
+
error = done.exception()
|
|
121
|
+
if isinstance(error, Released):
|
|
122
|
+
reason = UnknownError(cause=error, message=str(error))
|
|
123
|
+
call.settled.set(_FailCause(fail(SqlError(reason=reason))))
|
|
124
|
+
|
|
125
|
+
self._lifetime.submit(lambda: self._executor.submit(run)).add_done_callback(released)
|
|
126
|
+
|
|
127
|
+
def interrupted() -> Effect[None]:
|
|
128
|
+
with call.lock:
|
|
129
|
+
if call.phase == "queued":
|
|
130
|
+
call.phase = "dropped"
|
|
131
|
+
return _Succeed(None)
|
|
132
|
+
if call.phase == "running":
|
|
133
|
+
if release is not None:
|
|
134
|
+
call.phase = "abandoned"
|
|
135
|
+
return _Succeed(None)
|
|
136
|
+
if cancel is not None:
|
|
137
|
+
cancel.cancel()
|
|
138
|
+
|
|
139
|
+
return call.settled.wait().map(made)
|
|
140
|
+
|
|
141
|
+
return (
|
|
142
|
+
call.settled.wait()
|
|
143
|
+
.flat_map(lambda outcome: outcome)
|
|
144
|
+
.on_interrupt(_Suspend(interrupted))
|
|
145
|
+
)
|
|
146
|
+
|
|
147
|
+
return _Suspend(start)
|
|
148
|
+
|
|
149
|
+
def shutdown(self) -> None:
|
|
150
|
+
"""Stop the thread once its queued calls ran."""
|
|
151
|
+
self._lifetime.close()
|
|
152
|
+
self._executor.shutdown(wait=False)
|
|
153
|
+
|
|
154
|
+
|
|
155
|
+
__all__ = ["Cancellable", "Worker"]
|
|
@@ -0,0 +1,344 @@
|
|
|
1
|
+
"""The SQL Server client (Effect's ``@effect/sql-mssql`` ``MssqlClient``):
|
|
2
|
+
an ``SqlClient`` on a pool of pytds connections, with typed parameters
|
|
3
|
+
(``param``) and stored procedure calls (``call``).
|
|
4
|
+
|
|
5
|
+
``make`` opens the pool and checks it with ``SELECT 1`` (within
|
|
6
|
+
``connect_timeout``, five seconds by default); statements outside a
|
|
7
|
+
transaction each take a connection for themselves, a transaction keeps
|
|
8
|
+
one until it ends, and its savepoints are never released (SQL Server has
|
|
9
|
+
no ``RELEASE SAVEPOINT``). Streams are not implemented, as in Effect::
|
|
10
|
+
|
|
11
|
+
program.pipe(fx.provide(mssql.layer(mssql.MssqlClientConfig(server="localhost", ...))))
|
|
12
|
+
"""
|
|
13
|
+
|
|
14
|
+
from __future__ import annotations
|
|
15
|
+
|
|
16
|
+
from collections.abc import Callable, Mapping, Sequence
|
|
17
|
+
from typing import Any, Never, Unpack, cast
|
|
18
|
+
|
|
19
|
+
from pyfect._core import Effect
|
|
20
|
+
from pyfect._effect_args import require_effect
|
|
21
|
+
from pyfect.config import Config, ConfigError, unwrap
|
|
22
|
+
from pyfect.context import Context, Tag
|
|
23
|
+
from pyfect.effect import fail
|
|
24
|
+
from pyfect.layer import Layer, provide
|
|
25
|
+
from pyfect.layer import effect_context as layer_effect_context
|
|
26
|
+
from pyfect.reactivity import Reactivity
|
|
27
|
+
from pyfect.reactivity import layer as reactivity_layer
|
|
28
|
+
from pyfect.redacted import value as redacted_value
|
|
29
|
+
from pyfect.scope import Scope, scoped
|
|
30
|
+
from pyfect.sql.client import (
|
|
31
|
+
ClientFields,
|
|
32
|
+
SqlClient,
|
|
33
|
+
SqlClientService,
|
|
34
|
+
TransactionOptions,
|
|
35
|
+
make_with_transaction,
|
|
36
|
+
transaction_connection,
|
|
37
|
+
)
|
|
38
|
+
from pyfect.sql.client import make as make_sql_client
|
|
39
|
+
from pyfect.sql.connection import Connection
|
|
40
|
+
from pyfect.sql.error import ConnectionError, SqlError
|
|
41
|
+
from pyfect.sql.mssql import _pool
|
|
42
|
+
from pyfect.sql.mssql._connection import MssqlConnection, is_mssql_param, mssql_param
|
|
43
|
+
from pyfect.sql.mssql._driver import ConnectOptions
|
|
44
|
+
from pyfect.sql.mssql._names import escape
|
|
45
|
+
from pyfect.sql.mssql.config import (
|
|
46
|
+
DEFAULT_CANCEL_TIMEOUT,
|
|
47
|
+
DEFAULT_CHECK_TIMEOUT,
|
|
48
|
+
DEFAULT_CONNECT_TIMEOUT,
|
|
49
|
+
DEFAULT_CONNECTION_RETRY_INTERVAL,
|
|
50
|
+
DEFAULT_CONNECTION_TTL,
|
|
51
|
+
DEFAULT_MAX_CONNECTIONS,
|
|
52
|
+
DEFAULT_MAX_RETRIES_ON_TRANSIENT_ERRORS,
|
|
53
|
+
DEFAULT_MIN_CONNECTIONS,
|
|
54
|
+
DEFAULT_PORT,
|
|
55
|
+
MssqlClientConfig,
|
|
56
|
+
driver_seconds_or,
|
|
57
|
+
seconds_or,
|
|
58
|
+
)
|
|
59
|
+
from pyfect.sql.mssql.error import classify
|
|
60
|
+
from pyfect.sql.mssql.procedure import ProcedureWithValues, Result
|
|
61
|
+
from pyfect.sql.mssql.types import DataType, ParameterOptions
|
|
62
|
+
from pyfect.sql.statement import (
|
|
63
|
+
Compiled,
|
|
64
|
+
Compiler,
|
|
65
|
+
Custom,
|
|
66
|
+
Fragment,
|
|
67
|
+
Placeholder,
|
|
68
|
+
default_transforms,
|
|
69
|
+
)
|
|
70
|
+
from pyfect.sql.statement import make_compiler as make_statement_compiler
|
|
71
|
+
|
|
72
|
+
|
|
73
|
+
class MssqlClientService(SqlClientService):
|
|
74
|
+
"""A SQL Server client: an ``SqlClientService``, its config, typed
|
|
75
|
+
parameters and stored procedure calls."""
|
|
76
|
+
|
|
77
|
+
__slots__ = ("config",)
|
|
78
|
+
|
|
79
|
+
config: MssqlClientConfig
|
|
80
|
+
|
|
81
|
+
def __init__(self, *, config: MssqlClientConfig, **fields: Unpack[ClientFields]) -> None:
|
|
82
|
+
super().__init__(**fields)
|
|
83
|
+
self.config = config
|
|
84
|
+
|
|
85
|
+
def param(
|
|
86
|
+
self,
|
|
87
|
+
type: DataType,
|
|
88
|
+
value: object,
|
|
89
|
+
options: ParameterOptions | None = None,
|
|
90
|
+
) -> Fragment:
|
|
91
|
+
"""A parameter sent as ``type`` (``client.param(TYPES.BigInt, 1)``)."""
|
|
92
|
+
return Fragment((mssql_param(type, value, options or ParameterOptions()),))
|
|
93
|
+
|
|
94
|
+
def call[A](self, procedure: ProcedureWithValues[A]) -> Effect[Result[A], SqlError]:
|
|
95
|
+
"""Run a stored procedure (on the transaction's connection inside
|
|
96
|
+
one): its output parameters and rows, renamed as the client renames
|
|
97
|
+
results."""
|
|
98
|
+
transform_rows = self.transform_rows
|
|
99
|
+
return scoped(
|
|
100
|
+
self.acquirer.flat_map(
|
|
101
|
+
lambda conn: cast("MssqlConnection", conn).call(procedure, transform_rows)
|
|
102
|
+
)
|
|
103
|
+
)
|
|
104
|
+
|
|
105
|
+
|
|
106
|
+
class MssqlClient(Tag[MssqlClientService], key="@effect/sql-mssql/MssqlClient"):
|
|
107
|
+
"""The SQL Server client service."""
|
|
108
|
+
|
|
109
|
+
|
|
110
|
+
def _on_insert(
|
|
111
|
+
columns: Sequence[str],
|
|
112
|
+
placeholders: str,
|
|
113
|
+
values: Sequence[Sequence[object]],
|
|
114
|
+
returning: Compiled | None,
|
|
115
|
+
) -> Compiled:
|
|
116
|
+
output = ""
|
|
117
|
+
binds: list[object] = []
|
|
118
|
+
if returning is not None:
|
|
119
|
+
output = " OUTPUT INSERTED.*" if returning[0] == "*" else f" OUTPUT {returning[0]}"
|
|
120
|
+
binds.extend(returning[1])
|
|
121
|
+
for row in values:
|
|
122
|
+
binds.extend(row)
|
|
123
|
+
return (f"({','.join(columns)}){output} VALUES {placeholders}", tuple(binds))
|
|
124
|
+
|
|
125
|
+
|
|
126
|
+
def _on_record_update(
|
|
127
|
+
placeholders: str,
|
|
128
|
+
alias: str,
|
|
129
|
+
columns: str,
|
|
130
|
+
values: Sequence[Sequence[object]],
|
|
131
|
+
returning: Compiled | None,
|
|
132
|
+
) -> Compiled:
|
|
133
|
+
output = ""
|
|
134
|
+
binds: list[object] = []
|
|
135
|
+
if returning is not None:
|
|
136
|
+
output = "OUTPUT INSERTED.* " if returning[0] == "*" else f"OUTPUT {returning[0]} "
|
|
137
|
+
binds.extend(returning[1])
|
|
138
|
+
for row in values:
|
|
139
|
+
binds.extend(row)
|
|
140
|
+
return (f"{output}FROM (values {placeholders}) AS {alias}{columns}", tuple(binds))
|
|
141
|
+
|
|
142
|
+
|
|
143
|
+
def _on_custom(
|
|
144
|
+
segment: Custom[Any, Any, Any, Any], placeholder: Placeholder, _without_transform: bool
|
|
145
|
+
) -> Compiled:
|
|
146
|
+
if is_mssql_param(segment):
|
|
147
|
+
return (placeholder(None), (segment,))
|
|
148
|
+
return ("", ())
|
|
149
|
+
|
|
150
|
+
|
|
151
|
+
def make_compiler(transform: Callable[[str], str] | None = None) -> Compiler:
|
|
152
|
+
"""SQL Server's compiler: ``@1``, ``@2``, ... placeholders, ``[quoted]``
|
|
153
|
+
names (renamed with ``transform`` unless a statement runs without
|
|
154
|
+
transforms) and ``OUTPUT INSERTED`` for ``returning``."""
|
|
155
|
+
|
|
156
|
+
def on_identifier(value: str, without_transform: bool) -> str:
|
|
157
|
+
if transform is None or without_transform:
|
|
158
|
+
return escape(value)
|
|
159
|
+
return escape(transform(value))
|
|
160
|
+
|
|
161
|
+
return make_statement_compiler(
|
|
162
|
+
dialect="mssql",
|
|
163
|
+
placeholder=lambda index, _value: f"@{index}",
|
|
164
|
+
on_identifier=on_identifier,
|
|
165
|
+
on_record_update=_on_record_update,
|
|
166
|
+
on_custom=_on_custom,
|
|
167
|
+
on_insert=_on_insert,
|
|
168
|
+
)
|
|
169
|
+
|
|
170
|
+
|
|
171
|
+
def connect_options(config: MssqlClientConfig) -> ConnectOptions:
|
|
172
|
+
"""What the driver connects with, for ``config``."""
|
|
173
|
+
retries = config.max_retries_on_transient_errors
|
|
174
|
+
return ConnectOptions(
|
|
175
|
+
server=config.server,
|
|
176
|
+
port=config.port if config.port is not None or config.instance_name else DEFAULT_PORT,
|
|
177
|
+
instance_name=config.instance_name,
|
|
178
|
+
database=config.database,
|
|
179
|
+
username=config.username,
|
|
180
|
+
password=None if config.password is None else redacted_value(config.password),
|
|
181
|
+
encrypt=config.encrypt,
|
|
182
|
+
trust_server=config.trust_server,
|
|
183
|
+
connect_timeout=driver_seconds_or(config.connect_timeout, DEFAULT_CONNECT_TIMEOUT),
|
|
184
|
+
cancel_timeout=driver_seconds_or(config.cancel_timeout, DEFAULT_CANCEL_TIMEOUT),
|
|
185
|
+
connection_retry_interval=driver_seconds_or(
|
|
186
|
+
config.connection_retry_interval, DEFAULT_CONNECTION_RETRY_INTERVAL
|
|
187
|
+
),
|
|
188
|
+
max_retries_on_transient_errors=DEFAULT_MAX_RETRIES_ON_TRANSIENT_ERRORS
|
|
189
|
+
if retries is None
|
|
190
|
+
else retries,
|
|
191
|
+
)
|
|
192
|
+
|
|
193
|
+
|
|
194
|
+
def _span_attributes(config: MssqlClientConfig) -> list[tuple[str, object]]:
|
|
195
|
+
return [
|
|
196
|
+
*config.span_attributes.items(),
|
|
197
|
+
("db.system.name", "microsoft.sql_server"),
|
|
198
|
+
("db.namespace", config.database or "master"),
|
|
199
|
+
("server.address", config.server),
|
|
200
|
+
("server.port", config.port or DEFAULT_PORT),
|
|
201
|
+
]
|
|
202
|
+
|
|
203
|
+
|
|
204
|
+
def _connect_failed(error: SqlError) -> SqlError:
|
|
205
|
+
return SqlError(
|
|
206
|
+
reason=classify(
|
|
207
|
+
error,
|
|
208
|
+
message="MssqlClient: Failed to connect",
|
|
209
|
+
operation="connect",
|
|
210
|
+
fallback="connection",
|
|
211
|
+
)
|
|
212
|
+
)
|
|
213
|
+
|
|
214
|
+
|
|
215
|
+
def _connect_timed_out() -> Effect[Never, SqlError]:
|
|
216
|
+
return fail(
|
|
217
|
+
SqlError(
|
|
218
|
+
reason=ConnectionError(
|
|
219
|
+
cause=Exception("connection timeout"),
|
|
220
|
+
message="MssqlClient: Connection timeout",
|
|
221
|
+
operation="connect",
|
|
222
|
+
)
|
|
223
|
+
)
|
|
224
|
+
)
|
|
225
|
+
|
|
226
|
+
|
|
227
|
+
def make(config: MssqlClientConfig) -> Effect[MssqlClientService, SqlError, Scope | Reactivity]:
|
|
228
|
+
"""A client on a pool of connections; it closes with the scope."""
|
|
229
|
+
pool = _pool.make(
|
|
230
|
+
connect_options(config),
|
|
231
|
+
config.parameter_types,
|
|
232
|
+
min_connections=DEFAULT_MIN_CONNECTIONS
|
|
233
|
+
if config.min_connections is None
|
|
234
|
+
else config.min_connections,
|
|
235
|
+
max_connections=config.max_connections or DEFAULT_MAX_CONNECTIONS,
|
|
236
|
+
ttl=seconds_or(config.connection_ttl, DEFAULT_CONNECTION_TTL),
|
|
237
|
+
)
|
|
238
|
+
span_attributes = _span_attributes(config)
|
|
239
|
+
|
|
240
|
+
def build(
|
|
241
|
+
check_out: Effect[MssqlConnection, SqlError, Scope],
|
|
242
|
+
) -> Effect[MssqlClientService, SqlError, Reactivity]:
|
|
243
|
+
acquirer = cast("Effect[Connection, SqlError, Scope]", check_out)
|
|
244
|
+
connected = (
|
|
245
|
+
scoped(
|
|
246
|
+
acquirer.flat_map(
|
|
247
|
+
lambda conn: conn.execute_unprepared("SELECT 1", (), None)
|
|
248
|
+
).map_error(_connect_failed)
|
|
249
|
+
)
|
|
250
|
+
.timeout_or_else(
|
|
251
|
+
seconds_or(config.connect_timeout, DEFAULT_CHECK_TIMEOUT), _connect_timed_out
|
|
252
|
+
)
|
|
253
|
+
.as_void()
|
|
254
|
+
)
|
|
255
|
+
transform_rows = (
|
|
256
|
+
None
|
|
257
|
+
if config.transform_result_names is None
|
|
258
|
+
else default_transforms(config.transform_result_names).array
|
|
259
|
+
)
|
|
260
|
+
service = transaction_connection()
|
|
261
|
+
|
|
262
|
+
def mssql(conn: Connection) -> MssqlConnection:
|
|
263
|
+
return cast("MssqlConnection", conn)
|
|
264
|
+
|
|
265
|
+
with_transaction = make_with_transaction(
|
|
266
|
+
TransactionOptions(
|
|
267
|
+
transaction_service=service,
|
|
268
|
+
span_attributes=tuple(span_attributes),
|
|
269
|
+
acquire_connection=lambda restore: restore(acquirer),
|
|
270
|
+
begin=lambda conn, _statement: mssql(conn).begin,
|
|
271
|
+
savepoint=lambda conn, depth: mssql(conn).savepoint(f"effect_sql_{depth}"),
|
|
272
|
+
commit=lambda conn: mssql(conn).commit,
|
|
273
|
+
rollback=lambda conn: mssql(conn).rollback(),
|
|
274
|
+
rollback_savepoint=lambda conn, depth: mssql(conn).rollback(f"effect_sql_{depth}"),
|
|
275
|
+
)
|
|
276
|
+
)
|
|
277
|
+
client = make_sql_client(
|
|
278
|
+
driver="mssql",
|
|
279
|
+
acquirer=acquirer,
|
|
280
|
+
compiler=make_compiler(config.transform_query_names),
|
|
281
|
+
transaction_service=service,
|
|
282
|
+
span_attributes=span_attributes,
|
|
283
|
+
transform_rows=transform_rows,
|
|
284
|
+
)
|
|
285
|
+
|
|
286
|
+
def finish(c: SqlClientService) -> MssqlClientService:
|
|
287
|
+
fields = c.fields()
|
|
288
|
+
fields["with_transaction"] = with_transaction
|
|
289
|
+
return MssqlClientService(config=config, **fields)
|
|
290
|
+
|
|
291
|
+
return connected.and_then(client.map(finish))
|
|
292
|
+
|
|
293
|
+
return pool.flat_map(build)
|
|
294
|
+
|
|
295
|
+
|
|
296
|
+
def _contexts(client: MssqlClientService) -> Context[MssqlClient | SqlClient]:
|
|
297
|
+
return Context.empty().add(MssqlClient, client).add(SqlClient, client)
|
|
298
|
+
|
|
299
|
+
|
|
300
|
+
def layer_from[E](
|
|
301
|
+
acquire: Effect[MssqlClientService, E, Scope | Reactivity],
|
|
302
|
+
) -> Layer[MssqlClient | SqlClient, E]:
|
|
303
|
+
"""Provides the client ``acquire`` makes as ``MssqlClient`` and
|
|
304
|
+
``SqlClient``, with its own ``Reactivity``."""
|
|
305
|
+
require_effect(acquire, "acquire")
|
|
306
|
+
return layer_effect_context(acquire.map(_contexts)).pipe(provide(reactivity_layer))
|
|
307
|
+
|
|
308
|
+
|
|
309
|
+
def layer(config: MssqlClientConfig) -> Layer[MssqlClient | SqlClient, SqlError]:
|
|
310
|
+
"""Provides a client (``make``)."""
|
|
311
|
+
return layer_from(make(config))
|
|
312
|
+
|
|
313
|
+
|
|
314
|
+
def layer_config(
|
|
315
|
+
config: Config[MssqlClientConfig] | Mapping[str, Any],
|
|
316
|
+
) -> Layer[MssqlClient | SqlClient, ConfigError | SqlError]:
|
|
317
|
+
"""``layer`` with the config read from the ``ConfigProvider``: a
|
|
318
|
+
``Config`` of the whole config, or a mapping from field names to their
|
|
319
|
+
``Config`` (Effect's ``Config.Wrap``)::
|
|
320
|
+
|
|
321
|
+
mssql.layer_config({
|
|
322
|
+
"server": fx.config.string("MSSQL_SERVER"),
|
|
323
|
+
"password": fx.config.redacted("MSSQL_PASSWORD"),
|
|
324
|
+
})
|
|
325
|
+
"""
|
|
326
|
+
read: Effect[MssqlClientConfig, ConfigError] = (
|
|
327
|
+
config
|
|
328
|
+
if isinstance(config, Config)
|
|
329
|
+
else unwrap(config).map(lambda fields: MssqlClientConfig(**fields))
|
|
330
|
+
)
|
|
331
|
+
return layer_from(read.flat_map(make))
|
|
332
|
+
|
|
333
|
+
|
|
334
|
+
__all__ = [
|
|
335
|
+
"MssqlClient",
|
|
336
|
+
"MssqlClientService",
|
|
337
|
+
"SqlError",
|
|
338
|
+
"connect_options",
|
|
339
|
+
"layer",
|
|
340
|
+
"layer_config",
|
|
341
|
+
"layer_from",
|
|
342
|
+
"make",
|
|
343
|
+
"make_compiler",
|
|
344
|
+
]
|