summaryrefslogtreecommitdiffstats
path: root/third_party/python/aiohttp/examples/web_cookies.py
blob: e7a4a595d779f6bf156e99d9d529654fd423f2ca (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#!/usr/bin/env python3
"""Example for aiohttp.web basic server with cookies.
"""

from pprint import pformat

from aiohttp import web

tmpl = """\
<html>
    <body>
        <a href="/login">Login</a><br/>
        <a href="/logout">Logout</a><br/>
        <pre>{}</pre>
    </body>
</html>"""


async def root(request):
    resp = web.Response(content_type="text/html")
    resp.text = tmpl.format(pformat(request.cookies))
    return resp


async def login(request):
    resp = web.HTTPFound(location="/")
    resp.set_cookie("AUTH", "secret")
    return resp


async def logout(request):
    resp = web.HTTPFound(location="/")
    resp.del_cookie("AUTH")
    return resp


def init(loop):
    app = web.Application(loop=loop)
    app.router.add_get("/", root)
    app.router.add_get("/login", login)
    app.router.add_get("/logout", logout)
    return app


web.run_app(init())