great-expectations-cloud 20240523.0.dev0__py3-none-any.whl → 20251124.0.dev1__py3-none-any.whl

great_expectations_cloud/agent/actions/data_assistants/run_missingness_data_assistant.py

@@ -1,45 +0,0 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-from typing_extensions import override
-
-from great_expectations_cloud.agent.actions.agent_action import ActionResult, AgentAction
-from great_expectations_cloud.agent.actions.data_assistants.utils import (
-    DataAssistantType,
-    build_action_result,
-    build_batch_request,
-)
-from great_expectations_cloud.agent.event_handler import register_event_action
-from great_expectations_cloud.agent.models import RunMissingnessDataAssistantEvent
-
-if TYPE_CHECKING:
-    from great_expectations.data_context import CloudDataContext
-
-
-class RunMissingnessDataAssistantAction(AgentAction[RunMissingnessDataAssistantEvent]):
-    def __init__(self, context: CloudDataContext):
-        super().__init__(context=context)
-        self._data_assistant = self._context.assistants.missingness
-
-    @override
-    def run(
-        self,
-        event: RunMissingnessDataAssistantEvent,
-        id: str,
-    ) -> ActionResult:
-        batch_request = build_batch_request(context=self._context, event=event)
-
-        data_assistant_result = self._data_assistant.run(
-            batch_request=batch_request,
-        )
-        return build_action_result(
-            context=self._context,
-            data_assistant_name=DataAssistantType.MISSINGNESS,
-            event=event,
-            data_assistant_result=data_assistant_result,
-            id=id,
-        )
-
-
-register_event_action("0", RunMissingnessDataAssistantEvent, RunMissingnessDataAssistantAction)

great_expectations_cloud/agent/actions/data_assistants/run_onboarding_data_assistant.py

@@ -1,45 +0,0 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-from typing_extensions import override
-
-from great_expectations_cloud.agent.actions.agent_action import ActionResult, AgentAction
-from great_expectations_cloud.agent.actions.data_assistants.utils import (
-    DataAssistantType,
-    build_action_result,
-    build_batch_request,
-)
-from great_expectations_cloud.agent.event_handler import register_event_action
-from great_expectations_cloud.agent.models import RunOnboardingDataAssistantEvent
-
-if TYPE_CHECKING:
-    from great_expectations.data_context import CloudDataContext
-
-
-class RunOnboardingDataAssistantAction(AgentAction[RunOnboardingDataAssistantEvent]):
-    def __init__(self, context: CloudDataContext):
-        super().__init__(context=context)
-        self._data_assistant = self._context.assistants.onboarding
-
-    @override
-    def run(
-        self,
-        event: RunOnboardingDataAssistantEvent,
-        id: str,
-    ) -> ActionResult:
-        batch_request = build_batch_request(context=self._context, event=event)
-
-        data_assistant_result = self._data_assistant.run(
-            batch_request=batch_request,
-        )
-        return build_action_result(
-            context=self._context,
-            data_assistant_name=DataAssistantType.ONBOARDING,
-            event=event,
-            data_assistant_result=data_assistant_result,
-            id=id,
-        )
-
-
-register_event_action("0", RunOnboardingDataAssistantEvent, RunOnboardingDataAssistantAction)

great_expectations_cloud/agent/actions/data_assistants/utils.py

@@ -1,123 +0,0 @@
-from __future__ import annotations
-
-import enum
-from datetime import datetime
-from typing import TYPE_CHECKING
-
-from great_expectations.datasource.fluent import Datasource as FluentDatasource
-from great_expectations.exceptions import DataContextError
-
-from great_expectations_cloud.agent.actions.agent_action import (
-    ActionResult,
-)
-from great_expectations_cloud.agent.models import (
-    CreatedResource,
-    RunDataAssistantEvent,
-)
-
-if TYPE_CHECKING:
-    from great_expectations.core import ExpectationSuite
-    from great_expectations.core.batch import BatchRequest
-    from great_expectations.data_context.data_context import CloudDataContext
-    from great_expectations.rule_based_profiler.data_assistant_result.data_assistant_result import (
-        DataAssistantResult,
-    )
-
-
-class DataAssistantType(str, enum.Enum):
-    MISSINGNESS = "Missingness"
-    ONBOARDING = "Onboarding"
-
-
-def build_batch_request(
-    context: CloudDataContext,
-    event: RunDataAssistantEvent,
-) -> BatchRequest:
-    datasource = context.get_datasource(datasource_name=event.datasource_name)
-    if not isinstance(datasource, FluentDatasource):
-        raise TypeError(  # noqa: TRY003 # one off error
-            "The RunDataAssistant Action can only be used with a fluent-style Data Source."
-        )
-
-    asset = datasource.get_asset(asset_name=event.data_asset_name)
-    try:
-        batch_request = asset.build_batch_request()
-    except ValueError as e:
-        raise ValueError(  # noqa: TRY003 # one off error
-            "The RunDataAssistant Action for data assistant cannot be used with an "
-            "in-memory Data Asset."
-        ) from e
-
-    return batch_request
-
-
-def build_action_result(
-    context: CloudDataContext,
-    data_assistant_name: DataAssistantType,
-    event: RunDataAssistantEvent,
-    data_assistant_result: DataAssistantResult,
-    id: str,
-) -> ActionResult:
-    if event.expectation_suite_name is not None:
-        expectation_suite_name = event.expectation_suite_name
-    else:
-        expectation_suite_name = f"{event.data_asset_name} {data_assistant_name.value} Suite"
-
-    checkpoint_name = f"{event.data_asset_name} {data_assistant_name.value} Checkpoint"
-
-    # build tz aware timestamp
-    tz = datetime.now().astimezone().tzinfo
-    timestamp = datetime.now(tz=tz)
-
-    # ensure we have unique names for created resources
-    try:
-        context.get_expectation_suite(expectation_suite_name=expectation_suite_name)
-        # if that didn't error, this name exists, so we add the timestamp
-        expectation_suite_name = f"{expectation_suite_name} {timestamp}"
-    except DataContextError:
-        # resource is unique
-        pass
-
-    try:
-        context.get_checkpoint(name=checkpoint_name)
-        # if that didn't error, this name exists, so we add the timestamp
-        checkpoint_name = f"{checkpoint_name} {timestamp}"
-    except DataContextError:
-        # resource is unique
-        pass
-
-    expectation_suite: ExpectationSuite = data_assistant_result.get_expectation_suite(
-        expectation_suite_name=expectation_suite_name
-    )
-    context.add_expectation_suite(expectation_suite=expectation_suite)
-
-    checkpoint_config = {
-        "name": checkpoint_name,
-        "validations": [
-            {
-                "expectation_suite_name": expectation_suite_name,
-                "expectation_suite_ge_cloud_id": expectation_suite.ge_cloud_id,
-                "batch_request": {
-                    "datasource_name": event.datasource_name,
-                    "data_asset_name": event.data_asset_name,
-                },
-            }
-        ],
-        "config_version": 1,
-        "class_name": "Checkpoint",
-    }
-
-    checkpoint = context.add_checkpoint(**checkpoint_config)
-
-    expectation_suite_id = expectation_suite.ge_cloud_id
-    checkpoint_id = checkpoint.ge_cloud_id
-    if expectation_suite_id is None or checkpoint_id is None:
-        raise ValueError("Cloud backed resources must have an ID.")  # noqa: TRY003 # one off error
-    return ActionResult(
-        id=id,
-        type=event.type,
-        created_resources=[
-            CreatedResource(resource_id=expectation_suite_id, type="ExpectationSuite"),
-            CreatedResource(resource_id=checkpoint_id, type="Checkpoint"),
-        ],
-    )

great_expectations_cloud/agent/actions/list_table_names.py

@@ -1,76 +0,0 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-from great_expectations.compatibility import pydantic
-from great_expectations.compatibility.sqlalchemy import inspect
-from great_expectations.core.http import create_session
-from great_expectations.datasource.fluent import SQLDatasource
-from great_expectations.exceptions import GXCloudError
-from typing_extensions import override
-
-from great_expectations_cloud.agent.actions.agent_action import (
-    ActionResult,
-    AgentAction,
-)
-from great_expectations_cloud.agent.config import (
-    GxAgentEnvVars,
-    generate_config_validation_error_text,
-)
-from great_expectations_cloud.agent.event_handler import register_event_action
-from great_expectations_cloud.agent.models import (
-    ListTableNamesEvent,
-)
-
-if TYPE_CHECKING:
-    from great_expectations.compatibility.sqlalchemy.engine import Inspector
-
-
-class ListTableNamesAction(AgentAction[ListTableNamesEvent]):
-    @override
-    def run(self, event: ListTableNamesEvent, id: str) -> ActionResult:
-        datasource_name: str = event.datasource_name
-        datasource = self._context.get_datasource(datasource_name=datasource_name)
-        if not isinstance(datasource, SQLDatasource):
-            raise TypeError(  # noqa: TRY003 # one off error
-                f"This operation requires a SQL Data Source but got {type(datasource).__name__}."
-            )
-
-        inspector: Inspector = inspect(datasource.get_engine())
-        table_names: list[str] = inspector.get_table_names()
-
-        self._add_or_update_table_names_list(
-            datasource_id=str(datasource.id), table_names=table_names
-        )
-
-        return ActionResult(
-            id=id,
-            type=event.type,
-            created_resources=[],
-        )
-
-    def _add_or_update_table_names_list(self, datasource_id: str, table_names: list[str]) -> None:
-        try:
-            cloud_config = GxAgentEnvVars()
-        except pydantic.ValidationError as validation_err:
-            raise RuntimeError(
-                generate_config_validation_error_text(validation_err)
-            ) from validation_err
-
-        session = create_session(access_token=cloud_config.gx_cloud_access_token)
-        response = session.patch(
-            url=f"{cloud_config.gx_cloud_base_url}/organizations/"
-            f"{cloud_config.gx_cloud_organization_id}/datasources/{datasource_id}",
-            json={"table_names": table_names},
-        )
-        if response.status_code != 204:  # noqa: PLR2004
-            raise GXCloudError(
-                message=f"ListTableNamesAction encountered an error while connecting to GX Cloud. "
-                f"Unable to update "
-                f"table_names for Data Source with ID"
-                f"={datasource_id}.",
-                response=response,
-            )
-
-
-register_event_action("0", ListTableNamesEvent, ListTableNamesAction)

great_expectations_cloud-20240523.0.dev0.dist-info/RECORD

@@ -1,32 +0,0 @@
-great_expectations_cloud/__init__.py,sha256=1mr5RDyA2N38eynvEfVbuYIbjFadeJfqZ-X9CrqYiVo,150
-great_expectations_cloud/agent/__init__.py,sha256=et60Glanwmrnnj3I_gL7fQDIz-gWB3jgO6u8Ag8Klho,244
-great_expectations_cloud/agent/actions/__init__.py,sha256=Sxiu30lGA-5CB9-WFbKTv4agJP_0hgqvUZBJZFkb8uo,733
-great_expectations_cloud/agent/actions/agent_action.py,sha256=_Is2wPxYirSh6jRVZbo_hYPW3vy5hqsd-tMGgl3-o3U,763
-great_expectations_cloud/agent/actions/data_assistants/__init__.py,sha256=yBfQEyYMBNTMjoMTvaTxuwz6WnOPUhHRUPLoHAINgc4,316
-great_expectations_cloud/agent/actions/data_assistants/run_missingness_data_assistant.py,sha256=zOvGGcSnD9W9FcmhV6LEv3fI8YN31-43-FwaLM47b-I,1511
-great_expectations_cloud/agent/actions/data_assistants/run_onboarding_data_assistant.py,sha256=IlRy8LwMhcllf42OznmL5XF6dUnRz7AnOQ5CHzzGRWc,1503
-great_expectations_cloud/agent/actions/data_assistants/utils.py,sha256=gXOvAoJ_cPUeH8u9H30cPC8zAAfPBr_xICuyztvEuus,4279
-great_expectations_cloud/agent/actions/draft_datasource_config_action.py,sha256=auYIglyrL6ja0Nags8uAOjrLfdtuxP_Rpp_9XIpkvjs,4171
-great_expectations_cloud/agent/actions/list_table_names.py,sha256=1WiuT8IuRmqKYbF7IMGkXBm-VIAaWQgmpUw7UDfJ1So,2881
-great_expectations_cloud/agent/actions/run_checkpoint.py,sha256=CTpD78XwSjj9sKC0yzvmd2t7N1s7kKx3Lp3m1ZERk4A,2041
-great_expectations_cloud/agent/actions/run_metric_list_action.py,sha256=RSjEfESKjn2xKeOoLiWrm0Af6ghh0MWPiMwgu_2BmXY,2993
-great_expectations_cloud/agent/actions/unknown.py,sha256=mtWw9tDZqGZSiUWj7PtIlLFJ1dM-7AHBX3SO16-u2EM,739
-great_expectations_cloud/agent/agent.py,sha256=NOit0OSy4e-ELzeKYM01gnRrZTFFRnNIbwq7YzffzPo,17221
-great_expectations_cloud/agent/agent_warnings.py,sha256=9-xl_AI2V9Py4o7KzFOQjG3lYx-vZ36fq4w2iiPNiUw,362
-great_expectations_cloud/agent/cli.py,sha256=wdf8uhYM_gsQOgGTcu6irRKB8k5HmtB9-fSm1ym8XhM,2851
-great_expectations_cloud/agent/config.py,sha256=bUypxGky-dzTsITtfWOzJjUSoSReKYdOa9e8UFNduIY,858
-great_expectations_cloud/agent/constants.py,sha256=SAEtcOwI-SxZiZSVoCFfEC5oCtYdgmyUmK7for_zg_Q,246
-great_expectations_cloud/agent/event_handler.py,sha256=_0sC_ooay_8dbNx9ddR9eUaGT_cc_QvED7omDfBtL3g,4632
-great_expectations_cloud/agent/exceptions.py,sha256=Gl5QJp1dCjNwFxfWsKY2FShajMlrzkkqVW_1CrJbmGs,1360
-great_expectations_cloud/agent/message_service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-great_expectations_cloud/agent/message_service/asyncio_rabbit_mq_client.py,sha256=mAdXGXZ0nY1o4qCbZRU8CWwA06tRhlhU8-4PFHlOh6Q,9260
-great_expectations_cloud/agent/message_service/subscriber.py,sha256=sCGwee3qmCU0h7dSUAyTUhNTICuP1gqZpr3udrio0VQ,5836
-great_expectations_cloud/agent/models.py,sha256=UHOk2tySSlXSeWBycQ4AIQpbK8BbYlen6hpJ-vPwsdY,3562
-great_expectations_cloud/agent/run.py,sha256=V33RLoB1PFmJ0h0RfHG4SB5lN_Za8tW2Dua6GUpN9yY,639
-great_expectations_cloud/logging/README.md,sha256=vbwU689x8SkGjzoBYQzZOzAvh28fR0RCa1XY5WD-Dgs,1762
-great_expectations_cloud/logging/logging_cfg.py,sha256=TYP4jCuCY4PbMbct7arUhhLEPNJdI0CAtYgPaS80Dlw,5918
-great_expectations_cloud-20240523.0.dev0.dist-info/LICENSE,sha256=_JJnoX6N_OkrAwlCRizCwil0tIjDAy2TG3GiJ50sM6k,2084
-great_expectations_cloud-20240523.0.dev0.dist-info/METADATA,sha256=JBWa_KazpmYPwRI1CIktwwKYwDKdOJsZJo924YKgXyA,11112
-great_expectations_cloud-20240523.0.dev0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-great_expectations_cloud-20240523.0.dev0.dist-info/entry_points.txt,sha256=ofJgdeS2gSzxXLyCAjfNhIaN1wmSyR7EAMs5qhVaXE4,68
-great_expectations_cloud-20240523.0.dev0.dist-info/RECORD,,