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

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

@@ -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/datasette_enrichments_slow/__init__.py

@@ -0,0 +1,65 @@
+import asyncio
+from datasette_enrichments import Enrichment
+from datasette import hookimpl
+from wtforms import Form, FloatField
+import random
+
+
+@hookimpl
+def register_enrichments():
+    return [SlowEnrichment()]
+
+
+class SlowEnrichment(Enrichment):
+    name = "Slow"
+    slug = "slow"
+    description = "An enrichment on a slow loop to help debug progress bars"
+    batch_size = 1
+
+    async def initialize(self, datasette, db, table, config):
+        pass
+
+    async def get_config_form(self, db, table):
+        class ConfigForm(Form):
+            delay = FloatField(
+                "Delay (seconds)",
+                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
+
+    async def enrich_batch(
+        self,
+        db,
+        table,
+        rows,
+        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 → datasette_enrichments_slow-0.3}/datasette_enrichments_slow.egg-info/PKG-INFO

@@ -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.1 → 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.1 → datasette_enrichments_slow-0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "datasette-enrichments-slow"
-version = "0.1"
+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]

datasette_enrichments_slow-0.1/datasette_enrichments_slow/__init__.py

@@ -1,40 +0,0 @@
-import asyncio
-from datasette_enrichments import Enrichment
-from datasette import hookimpl
-from wtforms import Form, FloatField
-
-
-@hookimpl
-def register_enrichments():
-    return [SlowEnrichment()]
-
-
-class SlowEnrichment(Enrichment):
-    name = "Slow"
-    slug = "slow"
-    description = "An enrichment on a slow loop to help debug progress bars"
-    batch_size = 1
-
-    async def initialize(self, datasette, db, table, config):
-        pass
-
-    async def get_config_form(self, db, table):
-        class ConfigForm(Form):
-            delay = FloatField(
-                "Delay (seconds)",
-                description="How many seconds to delay for each row",
-                default=0.1,
-            )
-
-        return ConfigForm
-
-    async def enrich_batch(
-        self,
-        db,
-        table,
-        rows,
-        pks,
-        config,
-    ):
-        for row in rows:
-            await asyncio.sleep(config["delay"])