datasette-enrichments-slow 0.2__tar.gz → 0.3__tar.gz

{datasette_enrichments_slow-0.2 → datasette_enrichments_slow-0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: datasette-enrichments-slow
-Version: 0.2
+Version: 0.3
 Summary: An enrichment on a slow loop to help debug progress bars
 Author: Simon Willison
 License: Apache-2.0
@@ -14,7 +14,7 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: datasette
-Requires-Dist: datasette-enrichments
+Requires-Dist: datasette-enrichments>=0.5
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-asyncio; extra == "test"

{datasette_enrichments_slow-0.2 → datasette_enrichments_slow-0.3}/datasette_enrichments_slow/__init__.py

@@ -31,6 +31,16 @@ class SlowEnrichment(Enrichment):
                 description="What portion of rows should be errors? Between 0 and 1.0",
                 default=0,
             )
+            pause_rate = FloatField(
+                "Pause rate",
+                description="Chance an item should pause the run, 0 to 1.0",
+                default=0,
+            )
+            cancel_rate = FloatField(
+                "Cancel rate",
+                description="Chance an item should cancel the run, 0 to 1.0",
+                default=0,
+            )
 
         return ConfigForm
 
@@ -43,7 +53,13 @@ class SlowEnrichment(Enrichment):
         config,
     ):
         error_rate = config.get("error_rate") or 0
+        cancel_rate = config.get("cancel_rate") or 0
+        pause_rate = config.get("pause_rate") or 0
         for row in rows:
             if error_rate and random.random() < error_rate:
                 raise ValueError("Error rate")
+            if cancel_rate and random.random() < cancel_rate:
+                raise self.Cancel()
+            if pause_rate and random.random() < pause_rate:
+                raise self.Pause()
             await asyncio.sleep(config["delay"])

{datasette_enrichments_slow-0.2 → datasette_enrichments_slow-0.3}/datasette_enrichments_slow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: datasette-enrichments-slow
-Version: 0.2
+Version: 0.3
 Summary: An enrichment on a slow loop to help debug progress bars
 Author: Simon Willison
 License: Apache-2.0
@@ -14,7 +14,7 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: datasette
-Requires-Dist: datasette-enrichments
+Requires-Dist: datasette-enrichments>=0.5
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-asyncio; extra == "test"

{datasette_enrichments_slow-0.2 → datasette_enrichments_slow-0.3}/datasette_enrichments_slow.egg-info/requires.txt

@@ -1,5 +1,5 @@
 datasette
-datasette-enrichments
+datasette-enrichments>=0.5
 
 [test]
 pytest

{datasette_enrichments_slow-0.2 → datasette_enrichments_slow-0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "datasette-enrichments-slow"
-version = "0.2"
+version = "0.3"
 description = "An enrichment on a slow loop to help debug progress bars"
 readme = "README.md"
 authors = [{name = "Simon Willison"}]
@@ -12,7 +12,7 @@ classifiers=[
 requires-python = ">=3.9"
 dependencies = [
     "datasette",
-    "datasette-enrichments",
+    "datasette-enrichments>=0.5",
 ]
 
 [build-system]