- selector: support for YAML, closes #8

.github/workflows/test.yml

@@ -41,7 +41,11 @@ jobs:
         - output: examples/example.com.png
           url: http://www.example.com/
         - output: examples/w3c.org.png
-          url: https://www.w3.org/' | shot-scraper multi -
+          url: https://www.w3.org/
+        - output: examples/bighead-from-multi.png
+          url: https://simonwillison.net/
+          selector: "#bighead"
+        ' | shot-scraper multi -
     - name: Archive examples
       uses: actions/upload-artifact@v3
       with:

demo.yml

@@ -2,3 +2,6 @@
   url: http://www.example.com/
 - output: w3c.org.png
   url: https://www.w3.org/
+- output: simonwillison-bighead.png
+  url: https://simonwillison.net/
+  selector: "#bighead"

shot_scraper/cli.py

@@ -49,6 +49,7 @@ def shot(url, output, selector):
                 browser,
                 {
                     "url": url,
+                    "selector": selector,
                 },
                 selector=selector,
                 return_bytes=True,
@@ -60,8 +61,8 @@ def shot(url, output, selector):
                 {
                     "url": url,
                     "output": str(output),
+                    "selector": selector,
                 },
-                selector=selector,
             )
         browser.close()
 
@@ -103,7 +104,7 @@ def install():
     run_module("playwright", run_name="__main__")
 
 
-def take_shot(browser, shot, selector=None, return_bytes=False):
+def take_shot(browser, shot, return_bytes=False):
     url = shot.get("url") or ""
     if not (url.startswith("http://") or url.startswith("https://")):
         raise click.ClickException(
@@ -117,6 +118,7 @@ def take_shot(browser, shot, selector=None, return_bytes=False):
     page = browser.new_page()
     page.goto(url)
     message = ""
+    selector = shot.get("selector")
     if selector:
         if return_bytes:
             return page.locator(selector).screenshot()