"""An example of sending and receiving cookies."""
import logging
from typing import Dict
from urllib.parse import parse_qsl
from bareutils import header
from bareasgi import (
Application,
HttpRequest,
HttpResponse,
text_reader,
text_writer
)
logging.basicConfig(level=logging.DEBUG)
FORM_HTML = """
HTML Form
Cookies
{cookies}
"""
async def index(_request: HttpRequest) -> HttpResponse:
"""Redirect to the test page"""
return HttpResponse(303, [(b'Location', b'/get_form')])
async def get_form(request: HttpRequest) -> HttpResponse:
"""A response handler which returns a form and sets some cookies"""
cookies = header.cookie(request.scope['headers'])
first_name = cookies.get(b'first_name', [b'Micky'])[0]
last_name = cookies.get(b'last_name', [b'Mouse'])[0]
html_list = ''
for name, values in cookies.items():
for value in values:
html_list += f'- {name.decode()}
- {value.decode()}
'
html_list += '
'
html = FORM_HTML.format(
first_name=first_name.decode(),
last_name=last_name.decode(),
cookies=html_list
)
headers = [
(b'content-type', b'text/html'),
]
return HttpResponse(200, headers, text_writer(html))
async def post_form(request: HttpRequest) -> HttpResponse:
"""A response handler that reads the cookies from a posted form."""
content_type = header.find(b'content-type', request.scope['headers'])
if content_type != b'application/x-www-form-urlencoded':
return HttpResponse(500)
variables = await text_reader(request.body)
values: Dict[str, str] = dict(parse_qsl(variables))
first_name = values['first_name']
last_name = values['last_name']
headers = [
(b'location', b'/get_form'),
(b'set-cookie', f'first_name={first_name}'.encode()),
(b'set-cookie', f'last_name={last_name}'.encode()),
]
return HttpResponse(303, headers)
if __name__ == "__main__":
import uvicorn
app = Application()
app.http_router.add({'GET'}, '/', index)
app.http_router.add({'GET'}, '/get_form', get_form)
app.http_router.add({'POST'}, '/post_form', post_form)
uvicorn.run(app, port=9009)