Add paste remove to the CLI

Enable admins to remove pastes without having to access the DB directly.

pbnh/cli.py

@@ -60,3 +60,14 @@ def info(ctx: click.Context, show_data: bool, hashids: tuple[str]) -> None:
             + [("data", paste["data"] if show_data else f"({len(value)} bytes)")]
         ):
             click.echo(f"{column + ':':>15} {value}")
+
+
+@paste.command()  # type: ignore
+@click.argument("hashids", type=str, nargs=-1)
+@click.pass_context
+def remove(ctx: click.Context, hashids: tuple[str]) -> None:
+    """Remove pastes."""
+    for hashid in hashids:
+        removed = ctx.obj.data["paster"].delete(hashid=hashid)
+        message = "removed" if removed else "not found"
+        click.echo(f"{hashid} {message}")

pbnh/db.py

@@ -107,11 +107,13 @@ def query(self, *, hashid: str) -> dict[str, Any] | None:
                 }
         return None
 
-    def delete(self, *, hashid: str) -> None:
+    def delete(self, *, hashid: str) -> bool:
         with self._session.begin():
             result = self._query(hashid=hashid)
             if result:
                 self._session.delete(result)
+                return True
+        return False
 
 
 def _get_engine() -> Engine:

tests/test_cli.py

@@ -67,3 +67,11 @@ def test_cli_paste_info_hash_mismatch(test_cli_runner, monkeypatch):
     result = test_cli_runner.invoke(args=["paste", "info", hashid])
     assert hashid in result.output
     assert "WARNING" in result.output
+
+
+def test_cli_paste_remove(app, test_cli_runner):
+    """Pastes can be removed from the cLI."""
+    hashid = "abc123"
+    with app.app_context():
+        result = test_cli_runner.invoke(args=["paste", "remove", hashid])
+    assert hashid in result.output

tests/test_db.py

@@ -88,4 +88,4 @@ def test_delete(paster):
 
 def test_delete_nonexistent(paster):
     with paster as p:
-        assert p.delete(hashid="nonexistent") is None
+        assert not p.delete(hashid="nonexistent")