Update screenshot with shot-scraper, closes #40

README.md

@@ -21,9 +21,9 @@ Datasette plugin for modifying table schemas
 ## Installation
 
 Install this plugin in the same environment as Datasette.
-
-    $ pip install datasette-edit-schema
-
+```bash
+pip install datasette-edit-schema
+```
 ## Usage
 
 Navigate to `/-/edit-schema/dbname/tablename` on your Datasette instance to edit a specific table.
@@ -49,19 +49,20 @@ These permission checks will call the `permission_allowed()` plugin hook with th
 ## Development
 
 To set up this plugin locally, first checkout the code. Then create a new virtual environment:
-
-    cd datasette-edit-schema
-    python3 -mvenv venv
-    source venv/bin/activate
-
+```bash
+cd datasette-edit-schema
+python3 -mvenv venv
+source venv/bin/activate
+```
 Or if you are using `pipenv`:
-
-    pipenv shell
-
+```bash
+pipenv shell
+```
 Now install the dependencies and tests:
-
-    pip install -e '.[test]'
-
+```bash
+pip install -e '.[test]'
+```
 To run the tests:
-
-    pytest
+```bash
+pytest
+```

update-screenshot.sh

@@ -0,0 +1,39 @@
+#!/bin/bash
+
+# Grab the database
+wget https://datasette.io/content.db
+
+# Setup the root plugin
+mkdir shot-plugins
+cat > shot-plugins/root.py <<EOL
+from datasette import hookimpl, Response
+
+@hookimpl
+def register_routes():
+    def login_as_root(datasette, request):
+        response = Response.redirect("/-/edit-schema/content/licenses")
+        response.set_cookie(
+            "ds_actor", datasette.sign({"a": {"id": "root"}}, "actor")
+        )
+        return response
+    return (
+        ("/-/root", login_as_root),
+    )
+EOL
+
+# Start the server in the background and capture its PID
+datasette -p 8045 content.db --plugins-dir shot-plugins &
+SERVER_PID=$!
+
+# Wait for server to start
+sleep 2
+
+# Take the screenshot
+shot-scraper http://localhost:8045/-/root -w 800 -o datasette-edit-schema.png
+
+# Kill the server
+kill $SERVER_PID
+
+# Cleanup
+rm content.db
+rm -rf shot-plugins