UV package management and Snowflake logging extras

.github/workflows/publish-docs.yml

@@ -9,25 +9,28 @@ jobs:
     build-and-publish-docs:
         name: Build and publish docs
         runs-on: ubuntu-latest
-    
+
         steps:
-          - uses: actions/checkout@v3
-          - name: Set up Python 3.10
-            uses: actions/setup-python@v4
+          - uses: actions/checkout@v4
+
+          - name: Set up Python 3.12
+            uses: actions/setup-python@v5
             with:
-              python-version: "3.10"
-              cache: pip
-              cache-dependency-path: requirements*.txt
-
-          - name: Install dependencies
+              python-version: 3.12
+
+          # Install uv package manager
+          - name: Install uv
             run: |
               python -m pip install --upgrade pip
-              python -m pip install -r requirements.txt
-              python -m pip install -r requirements-dev.txt
-
+              pip install uv
+
+          - name: Install dependencies
+            run: |
+              uv sync --frozen --all-extras
+
           - name: Build docs
-            run: mkdocs build
-    
+            run: uv run mkdocs build
+
           - name: Publish docs
             uses: JamesIves/[email protected]
             with:

.github/workflows/publish-package.yml

@@ -18,15 +18,19 @@ jobs:
           cache: pip
           cache-dependency-path: requirements*.txt
 
-      - name: Install packages
+      # Install uv package manager
+      - name: Install uv
         run: |
-          python -m pip install --upgrade pip build
-          python -m pip install -r requirements.txt
-          python -m pip install -r requirements-dev.txt
+          python -m pip install --upgrade pip
+          pip install uv
+
+      - name: Install dependencies
+        run: |
+          uv sync --frozen --all-extras
 
       - name: Build a binary wheel and a source tarball
         run: |
-          python -m build --sdist --wheel --outdir dist/
+          uv run python -m build --sdist --wheel --outdir dist/
 
       - name: Publish build artifacts
         uses: actions/upload-artifact@v3

.github/workflows/pytest.yml

@@ -4,8 +4,8 @@ on:
   pull_request:
     paths:
       - '.github/workflows/pytest.yml'
-      - 'requirements.txt'
-      - 'requirements-dev.txt'
+      - 'pyproject.toml'
+      - 'uv.lock'
       - 'src/**'
       - 'tests/**'
 
@@ -27,15 +27,17 @@ jobs:
         uses: actions/setup-python@v4
         with:
           python-version: ${{ matrix.python-version }}
-          cache-dependency-path: requirements*.txt
-          cache: pip
-
-      - name: Install dependencies
+
+      # Install uv package manager
+      - name: Install uv
         run: |
           python -m pip install --upgrade pip
-          python -m pip install -r requirements.txt
-          python -m pip install -r requirements-dev.txt
+          pip install uv
+
+      - name: Install dependencies
+        run: |
+          uv sync --frozen --all-extras
 
       - name: Run tests
         run: |
-          python -m pytest tests/ -vv
+          uv run pytest tests/ -vv

.pre-commit-config.yaml

@@ -10,9 +10,3 @@ repos:
     hooks:
       - id: black
         language_version: python3
-  - repo: https://github.com/econchick/interrogate
-    rev: 1.7.0
-    hooks:
-      - id: interrogate
-        args: [-vv]
-        pass_filenames: false

docs/ext/snowflake.md

@@ -0,0 +1,3 @@
+# Snowflake
+
+::: src.prefecto.ext.snowflake

mkdocs.yml

@@ -12,6 +12,8 @@ nav:
       - "Batch Task": concurrency/batch_task.md
       - "Kill Switches": concurrency/kill_switch.md
     - "Logging": logging.md
+    - "Extras":
+      - "Snowflake": ext/snowflake.md
 
 plugins:
   - search:

pyproject.toml

@@ -25,14 +25,18 @@ classifiers = [
   "Topic :: Software Development :: Libraries",
 ]
 dependencies = [
-  "prefect>=2.10",
+  "prefect>=2.10.0",
 ]
 
 [project.urls]
 Documentation = "https://dominictarro.github.io/prefecto/"
 Repository = "https://github.com/dominictarro/prefecto"
 
 [project.optional-dependencies]
+snowflake = [
+    "coolname>=2.2.0",
+    "snowflake-connector-python>=3.0.0",
+]
 
 [tool.interrogate]
 exclude = ["tests/", "setup.py", "versioneer.py", "src/prefecto/_version.py"]
@@ -53,3 +57,13 @@ versionfile_source = "src/prefecto/_version.py"
 versionfile_build = "prefecto/_version.py"
 tag_prefix=""
 parentdir_prefix="prefecto-"
+
+[tool.uv]
+dev-dependencies = [
+    "mkdocs>=1.4.3",
+    "mkdocstrings-python>=1.1.2",
+    "pre-commit>=3.5.0",
+    "pytest-asyncio>=0.23.6",
+    "pytest>=7.4.3",
+    "versioneer>=0.29",
+]

src/prefecto/__init__.py

@@ -1,3 +1,9 @@
 from . import _version
 
 __version__ = _version.get_versions()["version"]
+
+from . import (
+    blocks,
+    logging,
+    states,
+)