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

datasette_enrichments_slow/__init__.py

@@ -2,6 +2,7 @@ import asyncio
 from datasette_enrichments import Enrichment
 from datasette import hookimpl
 from wtforms import Form, FloatField
+import random
 
 
 @hookimpl
@@ -25,6 +26,21 @@ class SlowEnrichment(Enrichment):
                 description="How many seconds to delay for each row",
                 default=0.1,
             )
+            error_rate = FloatField(
+                "Error rate",
+                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
 
@@ -36,5 +52,14 @@ class SlowEnrichment(Enrichment):
         pks,
         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.1.dist-info → datasette_enrichments_slow-0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: datasette-enrichments-slow
-Version: 0.1
+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.1.dist-info → datasette_enrichments_slow-0.3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.6.0)
+Generator: setuptools (75.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

datasette_enrichments_slow-0.1.dist-info/RECORD

@@ -1,7 +0,0 @@
-datasette_enrichments_slow/__init__.py,sha256=YH4FHP8NmsURC3lYDapLXmMPY_6Fsczrh15wk_6YpWU,924
-datasette_enrichments_slow-0.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-datasette_enrichments_slow-0.1.dist-info/METADATA,sha256=6VJ5WKbUn1tvxxLgJVp-zGUDJCZDLaI4vyGoKWDxris,2293
-datasette_enrichments_slow-0.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-datasette_enrichments_slow-0.1.dist-info/entry_points.txt,sha256=cvHA2Xq1fpH6lmAljMedKADFzSa_4Lj7Px38hCoqHDQ,58
-datasette_enrichments_slow-0.1.dist-info/top_level.txt,sha256=o-W1XO60TGFcagSJ9HljP25g_m1q_qnZxSJ9ftY70XQ,27
-datasette_enrichments_slow-0.1.dist-info/RECORD,,