databricks-labs-lakebridge 0.10.0__py3-none-any.whl

databricks/labs/lakebridge/reconcile/constants.py

@@ -0,0 +1,37 @@
+from enum import Enum, auto
+
+
+class AutoName(Enum):
+    """
+    This class is used to auto generate the enum values based on the name of the enum in lower case
+
+    Reference: https://docs.python.org/3/howto/enum.html#enum-advanced-tutorial
+    """
+
+    @staticmethod
+    # pylint: disable-next=bad-dunder-name
+    def _generate_next_value_(name, start, count, last_values):  # noqa ARG004
+        return name.lower()
+
+
+class ReconSourceType(AutoName):
+    SNOWFLAKE = auto()
+    ORACLE = auto()
+    DATABRICKS = auto()
+
+
+class ReconReportType(AutoName):
+    DATA = auto()
+    SCHEMA = auto()
+    ROW = auto()
+    ALL = auto()
+
+
+class SamplingSpecificationsType(AutoName):
+    FRACTION = auto()
+    COUNT = auto()
+
+
+class SamplingOptionMethod(AutoName):
+    RANDOM = auto()
+    STRATIFIED = auto()

databricks/labs/lakebridge/reconcile/exception.py

@@ -0,0 +1,42 @@
+from pyspark.errors import PySparkException
+from databricks.labs.lakebridge.reconcile.recon_output_config import ReconcileOutput
+
+
+class ColumnMismatchException(Exception):
+    """Raise the error when there is a mismatch in source and target column names"""
+
+
+class DataSourceRuntimeException(Exception):
+    """Raise the error when there is a runtime exception thrown in DataSource"""
+
+
+class WriteToTableException(Exception):
+    """Raise the error when there is a runtime exception thrown while writing data to table"""
+
+
+class InvalidInputException(ValueError):
+    """Raise the error when the input is invalid"""
+
+
+class ReconciliationException(Exception):
+    """Raise the error when there is an error occurred during reconciliation"""
+
+    def __init__(self, message: str, reconcile_output: ReconcileOutput | None = None):
+        self._reconcile_output = reconcile_output
+        super().__init__(message, reconcile_output)
+
+    @property
+    def reconcile_output(self) -> ReconcileOutput | None:
+        return self._reconcile_output
+
+
+class ReadAndWriteWithVolumeException(PySparkException):
+    """Raise the error when there is a runtime exception thrown while writing data to volume"""
+
+
+class CleanFromVolumeException(PySparkException):
+    """Raise the error when there is a runtime exception thrown while cleaning data from volume"""
+
+
+class InvalidSnowflakePemPrivateKey(Exception):
+    """Raise the error when the input private key is invalid"""