prefect-gitlab 0.2.0__tar.gz → 0.2.2__tar.gz

{prefect-gitlab-0.2.0/prefect_gitlab.egg-info → prefect-gitlab-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect-gitlab
-Version: 0.2.0
+Version: 0.2.2
 Summary: A Prefect collection for working with GitLab repositories.
 Home-page: https://github.com/prefecthq/prefect-gitlab
 Author: Prefect Technologies, Inc.
@@ -19,25 +19,41 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Topic :: Software Development :: Libraries
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
-Provides-Extra: dev
 License-File: LICENSE
+Requires-Dist: prefect>=2.13.5
+Requires-Dist: python-gitlab>=3.12.0
+Requires-Dist: tenacity>=8.2.3
+Provides-Extra: dev
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: flake8; extra == "dev"
+Requires-Dist: mypy; extra == "dev"
+Requires-Dist: mkdocs; extra == "dev"
+Requires-Dist: mkdocs-material; extra == "dev"
+Requires-Dist: mkdocstrings[python]; extra == "dev"
+Requires-Dist: isort; extra == "dev"
+Requires-Dist: pre-commit; extra == "dev"
+Requires-Dist: pytest-asyncio; extra == "dev"
+Requires-Dist: mock; python_version < "3.8" and extra == "dev"
+Requires-Dist: mkdocs-gen-files; extra == "dev"
+Requires-Dist: interrogate; extra == "dev"
+Requires-Dist: coverage; extra == "dev"
+Requires-Dist: pillow; extra == "dev"
 
 # prefect-gitlab
 
 <p align="center">
     <a href="https://pypi.python.org/pypi/prefect-gitlab/" alt="PyPI version">
-        <img alt="PyPI" src="https://img.shields.io/pypi/v/prefect-gitlab?color=0052FF&labelColor=090422"></a>
+        <img alt="PyPI" src="https://img.shields.io/pypi/v/prefect-gitlab?color=26272B&labelColor=090422"></a>
     <a href="https://github.com/prefecthq/prefect-gitlab/" alt="Stars">
-        <img src="https://img.shields.io/github/stars/prefecthq/prefect-gitlab?color=0052FF&labelColor=090422" /></a>
+        <img src="https://img.shields.io/github/stars/prefecthq/prefect-gitlab?color=26272B&labelColor=090422" /></a>
     <a href="https://pepy.tech/badge/prefect-gitlab/" alt="Downloads">
-        <img src="https://img.shields.io/pypi/dm/prefect-gitlab?color=0052FF&labelColor=090422" /></a>
+        <img src="https://img.shields.io/pypi/dm/prefect-gitlab?color=26272B&labelColor=090422" /></a>
     <a href="https://github.com/prefecthq/prefect-gitlab/pulse" alt="Activity">
-        <img src="https://img.shields.io/github/commit-activity/m/prefecthq/prefect-gitlab?color=0052FF&labelColor=090422" /></a>
+        <img src="https://img.shields.io/github/commit-activity/m/prefecthq/prefect-gitlab?color=26272B&labelColor=090422" /></a>
     <br>
     <a href="https://prefect-community.slack.com" alt="Slack">
-        <img src="https://img.shields.io/badge/slack-join_community-red.svg?color=0052FF&labelColor=090422&logo=slack" /></a>
-    <a href="https://discourse.prefect.io/" alt="Discourse">
-        <img src="https://img.shields.io/badge/discourse-browse_forum-red.svg?color=0052FF&labelColor=090422&logo=discourse" /></a>
+        <img src="https://img.shields.io/badge/slack-join_community-red.svg?color=26272B&labelColor=090422&logo=slack" /></a>
 </p>
 
 ## Welcome!
@@ -48,11 +64,11 @@ License-File: LICENSE
 
 ### Python setup
 
-Requires an installation of Python 3.7+.
+Requires an installation of Python 3.8 or higher.
 
-We recommend using a Python virtual environment manager such as pipenv, conda or virtualenv.
+We recommend using a Python virtual environment manager such as pipenv, conda, or virtualenv.
 
-This collection is designed to work with Prefect 2.3.0 or higher. For more information about how to use Prefect, please refer to the [Prefect documentation](https://orion-docs.prefect.io/).
+This integration is designed to work with Prefect 2.3.0 or higher. For more information about how to use Prefect, please refer to the [Prefect documentation](https://docs.prefect.io/).
 
 ### Installation
 
@@ -62,13 +78,13 @@ Install `prefect-gitlab` with `pip`:
 pip install prefect-gitlab
 ```
 
-Then, register to [view the storage block](https://orion-docs.prefect.io/ui/blocks/) on Prefect Cloud:
+Then, register the [block types](https://docs.prefect.io/concepts/blocks/)) in this integration to view the storage block type on Prefect Cloud:
 
 ```bash
 prefect block register -m prefect_gitlab
 ```
 
-Note, to use the `load` method on Blocks, you must already have a block document [saved through code](https://orion-docs.prefect.io/concepts/blocks/#saving-blocks) or [saved through the UI](https://orion-docs.prefect.io/ui/blocks/).
+Note, to use the `load` method on a block, you must already have a block document [saved](https://docs.prefect.io/concepts/blocks/).
 
 ## Creating a GitLab storage block
 
@@ -121,9 +137,9 @@ Finally, enter your repository information in the form and click **Create**:
 
 If you encounter any bugs while using `prefect-gitlab`, feel free to open an issue in the [prefect-gitlab](https://github.com/prefecthq/prefect-gitlab) repository.
 
-If you have any questions or issues while using `prefect-gitlab`, you can find help in either the [Prefect Discourse forum](https://discourse.prefect.io/) or the [Prefect Slack community](https://prefect.io/slack).
+If you have any questions or issues while using `prefect-gitlab`, you can find help in the [Prefect Slack community](https://prefect.io/slack).
 
-Feel free to ⭐️ or watch [`prefect-gitlab`](https://github.com/prefecthq/prefect-gitlab) for updates too!
+Feel free to ⭐️ or watch [`prefect-gitlab`](https://github.com/prefecthq/prefect-gitlab) for updates!
 
 ## Development
 

{prefect-gitlab-0.2.0 → prefect-gitlab-0.2.2}/README.md

@@ -2,18 +2,16 @@
 
 <p align="center">
     <a href="https://pypi.python.org/pypi/prefect-gitlab/" alt="PyPI version">
-        <img alt="PyPI" src="https://img.shields.io/pypi/v/prefect-gitlab?color=0052FF&labelColor=090422"></a>
+        <img alt="PyPI" src="https://img.shields.io/pypi/v/prefect-gitlab?color=26272B&labelColor=090422"></a>
     <a href="https://github.com/prefecthq/prefect-gitlab/" alt="Stars">
-        <img src="https://img.shields.io/github/stars/prefecthq/prefect-gitlab?color=0052FF&labelColor=090422" /></a>
+        <img src="https://img.shields.io/github/stars/prefecthq/prefect-gitlab?color=26272B&labelColor=090422" /></a>
     <a href="https://pepy.tech/badge/prefect-gitlab/" alt="Downloads">
-        <img src="https://img.shields.io/pypi/dm/prefect-gitlab?color=0052FF&labelColor=090422" /></a>
+        <img src="https://img.shields.io/pypi/dm/prefect-gitlab?color=26272B&labelColor=090422" /></a>
     <a href="https://github.com/prefecthq/prefect-gitlab/pulse" alt="Activity">
-        <img src="https://img.shields.io/github/commit-activity/m/prefecthq/prefect-gitlab?color=0052FF&labelColor=090422" /></a>
+        <img src="https://img.shields.io/github/commit-activity/m/prefecthq/prefect-gitlab?color=26272B&labelColor=090422" /></a>
     <br>
     <a href="https://prefect-community.slack.com" alt="Slack">
-        <img src="https://img.shields.io/badge/slack-join_community-red.svg?color=0052FF&labelColor=090422&logo=slack" /></a>
-    <a href="https://discourse.prefect.io/" alt="Discourse">
-        <img src="https://img.shields.io/badge/discourse-browse_forum-red.svg?color=0052FF&labelColor=090422&logo=discourse" /></a>
+        <img src="https://img.shields.io/badge/slack-join_community-red.svg?color=26272B&labelColor=090422&logo=slack" /></a>
 </p>
 
 ## Welcome!
@@ -24,11 +22,11 @@
 
 ### Python setup
 
-Requires an installation of Python 3.7+.
+Requires an installation of Python 3.8 or higher.
 
-We recommend using a Python virtual environment manager such as pipenv, conda or virtualenv.
+We recommend using a Python virtual environment manager such as pipenv, conda, or virtualenv.
 
-This collection is designed to work with Prefect 2.3.0 or higher. For more information about how to use Prefect, please refer to the [Prefect documentation](https://orion-docs.prefect.io/).
+This integration is designed to work with Prefect 2.3.0 or higher. For more information about how to use Prefect, please refer to the [Prefect documentation](https://docs.prefect.io/).
 
 ### Installation
 
@@ -38,13 +36,13 @@ Install `prefect-gitlab` with `pip`:
 pip install prefect-gitlab
 ```
 
-Then, register to [view the storage block](https://orion-docs.prefect.io/ui/blocks/) on Prefect Cloud:
+Then, register the [block types](https://docs.prefect.io/concepts/blocks/)) in this integration to view the storage block type on Prefect Cloud:
 
 ```bash
 prefect block register -m prefect_gitlab
 ```
 
-Note, to use the `load` method on Blocks, you must already have a block document [saved through code](https://orion-docs.prefect.io/concepts/blocks/#saving-blocks) or [saved through the UI](https://orion-docs.prefect.io/ui/blocks/).
+Note, to use the `load` method on a block, you must already have a block document [saved](https://docs.prefect.io/concepts/blocks/).
 
 ## Creating a GitLab storage block
 
@@ -97,9 +95,9 @@ Finally, enter your repository information in the form and click **Create**:
 
 If you encounter any bugs while using `prefect-gitlab`, feel free to open an issue in the [prefect-gitlab](https://github.com/prefecthq/prefect-gitlab) repository.
 
-If you have any questions or issues while using `prefect-gitlab`, you can find help in either the [Prefect Discourse forum](https://discourse.prefect.io/) or the [Prefect Slack community](https://prefect.io/slack).
+If you have any questions or issues while using `prefect-gitlab`, you can find help in the [Prefect Slack community](https://prefect.io/slack).
 
-Feel free to ⭐️ or watch [`prefect-gitlab`](https://github.com/prefecthq/prefect-gitlab) for updates too!
+Feel free to ⭐️ or watch [`prefect-gitlab`](https://github.com/prefecthq/prefect-gitlab) for updates!
 
 ## Development
 

{prefect-gitlab-0.2.0 → prefect-gitlab-0.2.2}/prefect_gitlab/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2023-10-04T15:50:26-0400",
+ "date": "2024-02-07T14:31:34-0500",
  "dirty": false,
  "error": null,
- "full-revisionid": "95d9ae7c574b182d45377c28a0c06217b806062e",
- "version": "0.2.0"
+ "full-revisionid": "9cd1a3e9ec4f374db98d832ea60d0acba6a416f7",
+ "version": "0.2.2"
 }
 '''  # END VERSION_JSON
 

{prefect-gitlab-0.2.0 → prefect-gitlab-0.2.2}/prefect_gitlab/repositories.py

@@ -52,6 +52,7 @@ from prefect.filesystems import ReadableDeploymentStorage
 from prefect.utilities.asyncutils import sync_compatible
 from prefect.utilities.processutils import run_process
 from pydantic import VERSION as PYDANTIC_VERSION
+from tenacity import retry, stop_after_attempt, wait_fixed, wait_random
 
 if PYDANTIC_VERSION.startswith("2."):
     from pydantic.v1 import Field, HttpUrl, validator
@@ -60,6 +61,13 @@ else:
 
 from prefect_gitlab.credentials import GitLabCredentials
 
+# Create get_directory retry settings
+
+MAX_CLONE_ATTEMPTS = 3
+CLONE_RETRY_MIN_DELAY_SECONDS = 1
+CLONE_RETRY_MIN_DELAY_JITTER_SECONDS = 0
+CLONE_RETRY_MAX_DELAY_JITTER_SECONDS = 3
+
 
 class GitLabRepository(ReadableDeploymentStorage):
     """
@@ -153,6 +161,15 @@ class GitLabRepository(ReadableDeploymentStorage):
         return str(content_source), str(content_destination)
 
     @sync_compatible
+    @retry(
+        stop=stop_after_attempt(MAX_CLONE_ATTEMPTS),
+        wait=wait_fixed(CLONE_RETRY_MIN_DELAY_SECONDS)
+        + wait_random(
+            CLONE_RETRY_MIN_DELAY_JITTER_SECONDS,
+            CLONE_RETRY_MAX_DELAY_JITTER_SECONDS,
+        ),
+        reraise=True,
+    )
     async def get_directory(
         self, from_path: Optional[str] = None, local_path: Optional[str] = None
     ) -> None:

{prefect-gitlab-0.2.0 → prefect-gitlab-0.2.2/prefect_gitlab.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect-gitlab
-Version: 0.2.0
+Version: 0.2.2
 Summary: A Prefect collection for working with GitLab repositories.
 Home-page: https://github.com/prefecthq/prefect-gitlab
 Author: Prefect Technologies, Inc.
@@ -19,25 +19,41 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Topic :: Software Development :: Libraries
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
-Provides-Extra: dev
 License-File: LICENSE
+Requires-Dist: prefect>=2.13.5
+Requires-Dist: python-gitlab>=3.12.0
+Requires-Dist: tenacity>=8.2.3
+Provides-Extra: dev
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: flake8; extra == "dev"
+Requires-Dist: mypy; extra == "dev"
+Requires-Dist: mkdocs; extra == "dev"
+Requires-Dist: mkdocs-material; extra == "dev"
+Requires-Dist: mkdocstrings[python]; extra == "dev"
+Requires-Dist: isort; extra == "dev"
+Requires-Dist: pre-commit; extra == "dev"
+Requires-Dist: pytest-asyncio; extra == "dev"
+Requires-Dist: mock; python_version < "3.8" and extra == "dev"
+Requires-Dist: mkdocs-gen-files; extra == "dev"
+Requires-Dist: interrogate; extra == "dev"
+Requires-Dist: coverage; extra == "dev"
+Requires-Dist: pillow; extra == "dev"
 
 # prefect-gitlab
 
 <p align="center">
     <a href="https://pypi.python.org/pypi/prefect-gitlab/" alt="PyPI version">
-        <img alt="PyPI" src="https://img.shields.io/pypi/v/prefect-gitlab?color=0052FF&labelColor=090422"></a>
+        <img alt="PyPI" src="https://img.shields.io/pypi/v/prefect-gitlab?color=26272B&labelColor=090422"></a>
     <a href="https://github.com/prefecthq/prefect-gitlab/" alt="Stars">
-        <img src="https://img.shields.io/github/stars/prefecthq/prefect-gitlab?color=0052FF&labelColor=090422" /></a>
+        <img src="https://img.shields.io/github/stars/prefecthq/prefect-gitlab?color=26272B&labelColor=090422" /></a>
     <a href="https://pepy.tech/badge/prefect-gitlab/" alt="Downloads">
-        <img src="https://img.shields.io/pypi/dm/prefect-gitlab?color=0052FF&labelColor=090422" /></a>
+        <img src="https://img.shields.io/pypi/dm/prefect-gitlab?color=26272B&labelColor=090422" /></a>
     <a href="https://github.com/prefecthq/prefect-gitlab/pulse" alt="Activity">
-        <img src="https://img.shields.io/github/commit-activity/m/prefecthq/prefect-gitlab?color=0052FF&labelColor=090422" /></a>
+        <img src="https://img.shields.io/github/commit-activity/m/prefecthq/prefect-gitlab?color=26272B&labelColor=090422" /></a>
     <br>
     <a href="https://prefect-community.slack.com" alt="Slack">
-        <img src="https://img.shields.io/badge/slack-join_community-red.svg?color=0052FF&labelColor=090422&logo=slack" /></a>
-    <a href="https://discourse.prefect.io/" alt="Discourse">
-        <img src="https://img.shields.io/badge/discourse-browse_forum-red.svg?color=0052FF&labelColor=090422&logo=discourse" /></a>
+        <img src="https://img.shields.io/badge/slack-join_community-red.svg?color=26272B&labelColor=090422&logo=slack" /></a>
 </p>
 
 ## Welcome!
@@ -48,11 +64,11 @@ License-File: LICENSE
 
 ### Python setup
 
-Requires an installation of Python 3.7+.
+Requires an installation of Python 3.8 or higher.
 
-We recommend using a Python virtual environment manager such as pipenv, conda or virtualenv.
+We recommend using a Python virtual environment manager such as pipenv, conda, or virtualenv.
 
-This collection is designed to work with Prefect 2.3.0 or higher. For more information about how to use Prefect, please refer to the [Prefect documentation](https://orion-docs.prefect.io/).
+This integration is designed to work with Prefect 2.3.0 or higher. For more information about how to use Prefect, please refer to the [Prefect documentation](https://docs.prefect.io/).
 
 ### Installation
 
@@ -62,13 +78,13 @@ Install `prefect-gitlab` with `pip`:
 pip install prefect-gitlab
 ```
 
-Then, register to [view the storage block](https://orion-docs.prefect.io/ui/blocks/) on Prefect Cloud:
+Then, register the [block types](https://docs.prefect.io/concepts/blocks/)) in this integration to view the storage block type on Prefect Cloud:
 
 ```bash
 prefect block register -m prefect_gitlab
 ```
 
-Note, to use the `load` method on Blocks, you must already have a block document [saved through code](https://orion-docs.prefect.io/concepts/blocks/#saving-blocks) or [saved through the UI](https://orion-docs.prefect.io/ui/blocks/).
+Note, to use the `load` method on a block, you must already have a block document [saved](https://docs.prefect.io/concepts/blocks/).
 
 ## Creating a GitLab storage block
 
@@ -121,9 +137,9 @@ Finally, enter your repository information in the form and click **Create**:
 
 If you encounter any bugs while using `prefect-gitlab`, feel free to open an issue in the [prefect-gitlab](https://github.com/prefecthq/prefect-gitlab) repository.
 
-If you have any questions or issues while using `prefect-gitlab`, you can find help in either the [Prefect Discourse forum](https://discourse.prefect.io/) or the [Prefect Slack community](https://prefect.io/slack).
+If you have any questions or issues while using `prefect-gitlab`, you can find help in the [Prefect Slack community](https://prefect.io/slack).
 
-Feel free to ⭐️ or watch [`prefect-gitlab`](https://github.com/prefecthq/prefect-gitlab) for updates too!
+Feel free to ⭐️ or watch [`prefect-gitlab`](https://github.com/prefecthq/prefect-gitlab) for updates!
 
 ## Development
 

{prefect-gitlab-0.2.0 → prefect-gitlab-0.2.2}/prefect_gitlab.egg-info/requires.txt

@@ -1,5 +1,6 @@
-prefect>=2.7.0
+prefect>=2.13.5
 python-gitlab>=3.12.0
+tenacity>=8.2.3
 
 [dev]
 pytest

prefect-gitlab-0.2.2/requirements.txt

@@ -0,0 +1,3 @@
+prefect>=2.13.5
+python-gitlab>=3.12.0
+tenacity>=8.2.3

{prefect-gitlab-0.2.0 → prefect-gitlab-0.2.2}/tests/test_repositories.py

@@ -15,7 +15,7 @@ else:
 
 import prefect_gitlab
 from prefect_gitlab.credentials import GitLabCredentials
-from prefect_gitlab.repositories import GitLabRepository
+from prefect_gitlab.repositories import GitLabRepository  # noqa: E402
 
 
 class TestGitLab:
@@ -243,3 +243,22 @@ class TestGitLab:
 
                 assert set(os.listdir(tmp_dst)) == set([sub_dir_name])
                 assert set(os.listdir(Path(tmp_dst) / sub_dir_name)) == child_contents
+
+    async def test_get_directory_retries(self, monkeypatch):
+        # Constants for the retry decorator
+        MAX_CLONE_ATTEMPTS = 3
+
+        # Create an instance of GitLabRepository
+        g = GitLabRepository(repository="https://gitlab.com/prefectHQ/prefect.git")
+
+        # Prepare a MagicMock to simulate the process call within get_directory
+        mock = AsyncMock()
+        mock.return_value = AsyncMock(returncode=1)  # Simulate failure
+        monkeypatch.setattr(prefect_gitlab.repositories, "run_process", mock)
+
+        # Call get_directory and expect it to raise a RetryError after maximum attempts
+        with pytest.raises(OSError):
+            await g.get_directory()
+        print(mock.call_count)
+        # Verify that the function retried the expected number of times
+        assert mock.call_count == MAX_CLONE_ATTEMPTS

prefect-gitlab-0.2.0/requirements.txt

@@ -1,2 +0,0 @@
-prefect>=2.7.0
-python-gitlab>=3.12.0