File: app_aiohttp.py

package info (click to toggle)
aiodogstatsd 0.16.0-5
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 296 kB
  • sloc: python: 1,080; makefile: 70; sh: 19
file content (45 lines) | stat: -rw-r--r-- 1,270 bytes parent folder | download | duplicates (3)
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
from http import HTTPStatus

from aiohttp import web

from aiodogstatsd.contrib import aiohttp as aiodogstatsd


async def handler_hello(request: web.Request) -> web.Response:
    return web.json_response({"hello": "aiodogstatsd"})


async def handler_bad_request(request: web.Request) -> web.Response:
    return web.json_response({"hello": "bad"}, status=HTTPStatus.BAD_REQUEST)


async def handler_internal_server_error(request: web.Request) -> web.Response:
    raise NotImplementedError()


async def handler_unauthorized(request: web.Request) -> web.Response:
    raise web.HTTPUnauthorized()


def get_application() -> web.Application:
    app = web.Application(middlewares=[aiodogstatsd.middleware_factory()])
    app.cleanup_ctx.append(
        aiodogstatsd.cleanup_context_factory(
            host="0.0.0.0", port=9125, constant_tags={"whoami": "I am Batman!"}
        )
    )
    app.add_routes(
        [
            web.get("/hello", handler_hello),
            web.get("/bad_request", handler_bad_request),
            web.get("/internal_server_error", handler_internal_server_error),
            web.get("/unauthorized", handler_unauthorized),
        ]
    )

    return app


if __name__ == "__main__":
    app = get_application()
    web.run_app(app)