| def main(request, response): | |
| """ | |
| Returns a response with a Set-Cookie header based on the query params. | |
| The body will be "1" if the cookie is present in the request and `drop` parameter is "0", | |
| otherwise the body will be "0". | |
| """ | |
| same_site = request.GET.first(b"same-site") | |
| cookie_name = request.GET.first(b"cookie-name") | |
| drop = request.GET.first(b"drop") | |
| cookie_in_request = b"0" | |
| cookie = b"%s=1; Secure; SameSite=%s" % (cookie_name, same_site) | |
| if drop == b"1": | |
| cookie += b"; Max-Age=0" | |
| if request.cookies.get(cookie_name): | |
| cookie_in_request = request.cookies[cookie_name].value | |
| headers = [(b'Content-Type', b'text/html'), (b'Set-Cookie', cookie)] | |
| return (200, headers, cookie_in_request) |