snowpark-checkpoints-configuration 0.2.1__py3-none-any.whl → 0.3.0__py3-none-any.whl

snowflake/snowpark_checkpoints_configuration/__version__.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "0.2.1"
+__version__ = "0.3.0"

snowflake/snowpark_checkpoints_configuration/checkpoint_metadata.py

@@ -18,6 +18,9 @@ import os
 
 from typing import Optional
 
+from snowflake.snowpark_checkpoints_configuration.io_utils.io_file_manager import (
+    get_io_file_manager,
+)
 from snowflake.snowpark_checkpoints_configuration.model.checkpoints import (
     Checkpoint,
     Checkpoints,
@@ -42,16 +45,13 @@ class CheckpointMetadata(metaclass=Singleton):
 
     def __init__(self, path: Optional[str] = None):
         self.checkpoint_model: Checkpoints = Checkpoints(type="", pipelines=[])
-        directory = path if path is not None else os.getcwd()
+        directory = path if path is not None else get_io_file_manager().getcwd()
         checkpoints_file = os.path.join(directory, "checkpoints.json")
-        if os.path.exists(checkpoints_file):
+        if get_io_file_manager().file_exists(checkpoints_file):
             LOGGER.info("Reading checkpoints file: '%s'", checkpoints_file)
             try:
-                with open(checkpoints_file) as f:
-                    checkpoint_json = f.read()
-                    self.checkpoint_model = Checkpoints.model_validate_json(
-                        checkpoint_json
-                    )
+                checkpoint_json = get_io_file_manager().read(checkpoints_file)
+                self.checkpoint_model = Checkpoints.model_validate_json(checkpoint_json)
                 LOGGER.info(
                     "Successfully read and validated checkpoints file: '%s'",
                     checkpoints_file,

snowflake/snowpark_checkpoints_configuration/io_utils/__init__.py

@@ -0,0 +1,26 @@
+# Copyright 2025 Snowflake Inc.
+# SPDX-License-Identifier: Apache-2.0
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+
+# http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+__all__ = ["EnvStrategy", "IOFileManager", "IODefaultStrategy"]
+
+from snowflake.snowpark_checkpoints_configuration.io_utils.io_env_strategy import (
+    EnvStrategy,
+)
+from snowflake.snowpark_checkpoints_configuration.io_utils.io_default_strategy import (
+    IODefaultStrategy,
+)
+from snowflake.snowpark_checkpoints_configuration.io_utils.io_file_manager import (
+    IOFileManager,
+)

snowflake/snowpark_checkpoints_configuration/io_utils/io_default_strategy.py

@@ -0,0 +1,34 @@
+# Copyright 2025 Snowflake Inc.
+# SPDX-License-Identifier: Apache-2.0
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+
+# http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import os
+
+from typing import Optional
+
+from snowflake.snowpark_checkpoints_configuration.io_utils import EnvStrategy
+
+
+class IODefaultStrategy(EnvStrategy):
+    def file_exists(self, path: str) -> bool:
+        return os.path.isfile(path)
+
+    def read(
+        self, file_path: str, mode: str = "r", encoding: Optional[str] = None
+    ) -> str:
+        with open(file_path, mode=mode, encoding=encoding) as file:
+            return file.read()
+
+    def getcwd(self) -> str:
+        return os.getcwd()

snowflake/snowpark_checkpoints_configuration/io_utils/io_env_strategy.py

@@ -0,0 +1,62 @@
+# Copyright 2025 Snowflake Inc.
+# SPDX-License-Identifier: Apache-2.0
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+
+# http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from abc import ABC, abstractmethod
+from typing import Optional
+
+
+class EnvStrategy(ABC):
+
+    """An abstract base class that defines methods for file and directory operations.
+
+    Subclasses should implement these methods to provide environment-specific behavior.
+    """
+
+    @abstractmethod
+    def file_exists(self, path: str) -> bool:
+        """Check if a file exists.
+
+        Args:
+            path: The path to the file.
+
+        Returns:
+            bool: True if the file exists, False otherwise.
+
+        """
+
+    @abstractmethod
+    def read(
+        self, file_path: str, mode: str = "r", encoding: Optional[str] = None
+    ) -> str:
+        """Read content from a file.
+
+        Args:
+            file_path: The path to the file to read from.
+            mode: The mode in which to open the file.
+            encoding: The encoding to use for reading the file.
+
+        Returns:
+            str: The content of the file.
+
+        """
+
+    @abstractmethod
+    def getcwd(self) -> str:
+        """Get the current working directory.
+
+        Returns:
+            str: The current working directory.
+
+        """

snowflake/snowpark_checkpoints_configuration/io_utils/io_file_manager.py

@@ -0,0 +1,57 @@
+# Copyright 2025 Snowflake Inc.
+# SPDX-License-Identifier: Apache-2.0
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+
+# http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from typing import Optional
+
+from snowflake.snowpark_checkpoints_configuration.io_utils import (
+    EnvStrategy,
+    IODefaultStrategy,
+)
+from snowflake.snowpark_checkpoints_configuration.singleton import Singleton
+
+
+class IOFileManager(metaclass=Singleton):
+    def __init__(self, strategy: Optional[EnvStrategy] = None):
+        self.strategy = strategy or IODefaultStrategy()
+
+    def file_exists(self, path: str) -> bool:
+        return self.strategy.file_exists(path)
+
+    def read(
+        self, file_path: str, mode: str = "r", encoding: Optional[str] = None
+    ) -> str:
+        return self.strategy.read(file_path, mode, encoding)
+
+    def getcwd(self) -> str:
+        return self.strategy.getcwd()
+
+    def set_strategy(self, strategy: EnvStrategy):
+        """Set the strategy for file and directory operations.
+
+        Args:
+            strategy (EnvStrategy): The strategy to use for file and directory operations.
+
+        """
+        self.strategy = strategy
+
+
+def get_io_file_manager():
+    """Get the singleton instance of IOFileManager.
+
+    Returns:
+        IOFileManager: The singleton instance of IOFileManager.
+
+    """
+    return IOFileManager()

{snowpark_checkpoints_configuration-0.2.1.dist-info → snowpark_checkpoints_configuration-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowpark-checkpoints-configuration
-Version: 0.2.1
+Version: 0.3.0
 Summary: Migration tools for Snowpark
 Project-URL: Bug Tracker, https://github.com/snowflakedb/snowpark-checkpoints/issues
 Project-URL: Source code, https://github.com/snowflakedb/snowpark-checkpoints/
@@ -29,6 +29,7 @@ Requires-Python: <3.12,>=3.9
 Requires-Dist: pydantic>=1.8.2
 Requires-Dist: snowflake-snowpark-python>=1.23.0
 Provides-Extra: development
+Requires-Dist: certifi==2025.1.31; extra == 'development'
 Requires-Dist: coverage>=7.6.7; extra == 'development'
 Requires-Dist: hatchling==1.25.0; extra == 'development'
 Requires-Dist: pre-commit>=4.0.1; extra == 'development'

snowpark_checkpoints_configuration-0.3.0.dist-info/RECORD

@@ -0,0 +1,14 @@
+snowflake/snowpark_checkpoints_configuration/__init__.py,sha256=xZ2oBwJeEso0laWYDaXQidGt_MdjgNsOZ1_0BcZcXvE,980
+snowflake/snowpark_checkpoints_configuration/__version__.py,sha256=kbbDnlkY7JOLNHvfWYkCO_mOBOV9GniMGdxYoQpLhyg,632
+snowflake/snowpark_checkpoints_configuration/checkpoint_metadata.py,sha256=fdmZpZxzqwtYfCjwjkl_9M8Z1hY04f_vE05HfGVRQc4,2773
+snowflake/snowpark_checkpoints_configuration/checkpoint_name_utils.py,sha256=Xc4k3JU6A96-79VFRR8NrNAUPeO3V1DEAhngg-hLlU4,1787
+snowflake/snowpark_checkpoints_configuration/singleton.py,sha256=7AgIHQBXVRvPBBCkmBplzkdrrm-xVWf_N8svzA2vF8E,836
+snowflake/snowpark_checkpoints_configuration/io_utils/__init__.py,sha256=UgACTfonQw5bMAAuU_CrtVTzVVKxxK6NEDOogNl2qJM,996
+snowflake/snowpark_checkpoints_configuration/io_utils/io_default_strategy.py,sha256=TX0xhezFmO5f37SsNP8nhsfAfRWJ2YwYPOE_pefJvsg,1113
+snowflake/snowpark_checkpoints_configuration/io_utils/io_env_strategy.py,sha256=h6gjwBqOhyJHxAvWa7kL4jvyqF31ssTWGkQCAH99rd8,1735
+snowflake/snowpark_checkpoints_configuration/io_utils/io_file_manager.py,sha256=H_xNLycfia2R_35xF1ery41i1BqFzysEuQjkL5sZeEs,1782
+snowflake/snowpark_checkpoints_configuration/model/checkpoints.py,sha256=R1L9kA6rhZxftw6iq7p4FwswcrJvU8HRn0XOkv6bRJA,4852
+snowpark_checkpoints_configuration-0.3.0.dist-info/METADATA,sha256=Bjd-RHBmSKluYnJGbrsNEggmPwgJacvGGhBESZHT1i4,2782
+snowpark_checkpoints_configuration-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+snowpark_checkpoints_configuration-0.3.0.dist-info/licenses/LICENSE,sha256=DVQuDIgE45qn836wDaWnYhSdxoLXgpRRKH4RuTjpRZQ,10174
+snowpark_checkpoints_configuration-0.3.0.dist-info/RECORD,,

snowpark_checkpoints_configuration-0.2.1.dist-info/RECORD

@@ -1,10 +0,0 @@
-snowflake/snowpark_checkpoints_configuration/__init__.py,sha256=xZ2oBwJeEso0laWYDaXQidGt_MdjgNsOZ1_0BcZcXvE,980
-snowflake/snowpark_checkpoints_configuration/__version__.py,sha256=jEnm4p_P4FqdYsTq3hnGQnhLZ4KwL0_Ew8fDF8BRL98,632
-snowflake/snowpark_checkpoints_configuration/checkpoint_metadata.py,sha256=vr60OMFW5PUR5BSBSi1PqiTRoKi4D64qxSUP5z1Lz4o,2692
-snowflake/snowpark_checkpoints_configuration/checkpoint_name_utils.py,sha256=Xc4k3JU6A96-79VFRR8NrNAUPeO3V1DEAhngg-hLlU4,1787
-snowflake/snowpark_checkpoints_configuration/singleton.py,sha256=7AgIHQBXVRvPBBCkmBplzkdrrm-xVWf_N8svzA2vF8E,836
-snowflake/snowpark_checkpoints_configuration/model/checkpoints.py,sha256=R1L9kA6rhZxftw6iq7p4FwswcrJvU8HRn0XOkv6bRJA,4852
-snowpark_checkpoints_configuration-0.2.1.dist-info/METADATA,sha256=13FFoejcxLnf6xyX_NG7bpqVvWs3rMYNTcdhPqfRszQ,2724
-snowpark_checkpoints_configuration-0.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-snowpark_checkpoints_configuration-0.2.1.dist-info/licenses/LICENSE,sha256=DVQuDIgE45qn836wDaWnYhSdxoLXgpRRKH4RuTjpRZQ,10174
-snowpark_checkpoints_configuration-0.2.1.dist-info/RECORD,,