datasette-enrichments-slow 0.2__py3-none-any.whl → 0.3__py3-none-any.whl

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.dist-info → datasette_enrichments_slow-0.3.dist-info}/METADATA

@@ -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.3.dist-info/RECORD

@@ -0,0 +1,7 @@
+datasette_enrichments_slow/__init__.py,sha256=oLtRUwe3rzp8KX1mu1BLwST_f53Lb2BtwNrQ90aQRMg,1962
+datasette_enrichments_slow-0.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+datasette_enrichments_slow-0.3.dist-info/METADATA,sha256=DTBKLQUV1pLjRCwr83jVltX6UfmD7_FAVZfXGrpLpjc,2298
+datasette_enrichments_slow-0.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+datasette_enrichments_slow-0.3.dist-info/entry_points.txt,sha256=cvHA2Xq1fpH6lmAljMedKADFzSa_4Lj7Px38hCoqHDQ,58
+datasette_enrichments_slow-0.3.dist-info/top_level.txt,sha256=o-W1XO60TGFcagSJ9HljP25g_m1q_qnZxSJ9ftY70XQ,27
+datasette_enrichments_slow-0.3.dist-info/RECORD,,

datasette_enrichments_slow-0.2.dist-info/RECORD

@@ -1,7 +0,0 @@
-datasette_enrichments_slow/__init__.py,sha256=cPnIhtqLPq6mcohBVryJfHFa32WYM3DssRUwgDCp5pY,1292
-datasette_enrichments_slow-0.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-datasette_enrichments_slow-0.2.dist-info/METADATA,sha256=ynTMyO8aqfOJFMOzJlVkJYzYxKhjNMeimK-2fpJJpd8,2293
-datasette_enrichments_slow-0.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-datasette_enrichments_slow-0.2.dist-info/entry_points.txt,sha256=cvHA2Xq1fpH6lmAljMedKADFzSa_4Lj7Px38hCoqHDQ,58
-datasette_enrichments_slow-0.2.dist-info/top_level.txt,sha256=o-W1XO60TGFcagSJ9HljP25g_m1q_qnZxSJ9ftY70XQ,27
-datasette_enrichments_slow-0.2.dist-info/RECORD,,