shot-scraper multi shots.yml -o output.png to selectively run shots from a YAML file

[simonw]

In adding new screenshots to this multi file: https://github.com/simonw/datasette-screenshots/blob/main/shots.yml

I found myself wanting a way to run just a subset of the shots defined in the file.

Here's a design for that:

shot-scraper multi shots.yml -o binary-data.png
Screenshot of 'table' on 'https://latest.datasette.io/fixtures/binary_data' written to 'binary-data.png'

[simonw]

Here's a prototype that works pretty well already:

diff --git a/shot_scraper/cli.py b/shot_scraper/cli.py
index fc69fe6..aded7ef 100644
--- a/shot_scraper/cli.py
+++ b/shot_scraper/cli.py
@@ -315,6 +315,13 @@ def _browser_context(
     is_flag=True,
     help="Skip images that already exist",
 )
+@click.option(
+    "outputs",
+    "-o",
+    "--output",
+    help="Just take shots matching this output file",
+    multiple=True,
+)
 @browser_option
 @user_agent_option
 @reduced_motion_option
@@ -325,6 +332,7 @@ def multi(
     timeout,
     fail_on_error,
     noclobber,
+    outputs,
     browser,
     user_agent,
     reduced_motion,
@@ -366,6 +374,8 @@ def multi(
                 and pathlib.Path(shot["output"]).exists()
             ):
                 continue
+            if outputs and shot.get("output") not in outputs:
+                continue
             try:
                 take_shot(context, shot)
             except TimeoutError as e:

New help output:

Options:
  -a, --auth FILENAME             Path to JSON authentication context file
  --retina                        Use device scale factor of 2
  --timeout INTEGER               Wait this many milliseconds before failing
  --fail-on-error                 Fail noisily on error
  -n, --no-clobber                Skip images that already exist
  -o, --output TEXT               Just take shots matching this output file