response.set_cookie() method

[simonw]

Mainly to clean up this code:

datasette/datasette/app.py

Lines 439 to 454 in 4fa7cf6

	def _write_messages_to_response(self, request, response):
	if getattr(request, "_messages", None):
	# Set those messages
	cookie = SimpleCookie()
	cookie["ds_messages"] = self.sign(request._messages, "messages")
	cookie["ds_messages"]["path"] = "/"
	# TODO: Co-exist with existing set-cookie headers
	assert "set-cookie" not in response.headers
	response.headers["set-cookie"] = cookie.output(header="").lstrip()
	elif getattr(request, "_messages_should_clear", False):
	cookie = SimpleCookie()
	cookie["ds_messages"] = ""
	cookie["ds_messages"]["path"] = "/"
	# TODO: Co-exist with existing set-cookie headers
	assert "set-cookie" not in response.headers
	response.headers["set-cookie"] = cookie.output(header="").lstrip()

[simonw]

Also:

datasette/datasette/views/special.py

Lines 63 to 76 in dfff34e

	if secrets.compare_digest(token, self.ds._root_token):
	self.ds._root_token = None
	cookie = SimpleCookie()
	cookie["ds_actor"] = self.ds.sign({"id": "root"}, "actor")
	cookie["ds_actor"]["path"] = "/"
	response = Response(
	body="",
	status=302,
	headers={
	"Location": "/",
	"set-cookie": cookie.output(header="").lstrip(),
	},
	)
	return response

[simonw]

Documentation: https://datasette.readthedocs.io/en/latest/internals.html#setting-cookies-with-response-set-cookie