服务端指南

准备使用aiohttp但不知道如何开始?这里有一些小例子来快速熟悉下。接下来我们一起来试着开发一个小投票系统。

如果你想改进或与之对比学习,可以查看demo source 来获取全部源码。

准备好我们的开发环境
首先检查下python版本:

  1. $ python -V
  2. Python 3.5.0

我们需要python 3.5.0及以上版本。

假设你已经安装好aiohttp库了。你可以用以下命令来查询当前aiohttp库的版本。

  1. $ python3 -c 'import aiohttp; print(aiohttp.__version__)'
  2. 2.0.5

项目结构与其他以python为基础的web项目大同小异:

  1. .
  2. ├── README.rst
  3. └── polls
  4. ├── Makefile
  5. ├── README.rst
  6. ├── aiohttpdemo_polls
  7. ├── __init__.py
  8. ├── __main__.py
  9. ├── db.py
  10. ├── main.py
  11. ├── routes.py
  12. ├── templates
  13. ├── utils.py
  14. └── views.py
  15. ├── config
  16. └── polls.yaml
  17. ├── images
  18. └── example.png
  19. ├── setup.py
  20. ├── sql
  21. ├── create_tables.sql
  22. ├── install.sh
  23. └── sample_data.sql
  24. └── static
  25. └── style.css

开始用aiohttp构建我们的第一个应用程序

该指南借鉴了Django投票系统指南。

创建应用程序

aiohttp的服务端程序都是 aiohttp.web.Application实例对象。用于创建信号,连接路由等。

使用下列代码可以创建一个应用程序:

  1. from aiohttp import web
  2. app = web.Application()
  3. web.run_app(app, host='127.0.0.1', port=8080)

将其保存在aiohttpdemo_polls/main.py然后开启服务器:
$ python3 main.py

你会在命令行中看到如下输出:

  1. ======== Running on http://127.0.0.1:8080 ========
  2. (Press CTRL+C to quit)

在浏览器中打开 http://127.0.0.1:8080 或在命令行中使用curl:
$ curl -X GET localhost:8080

啊咧,出现了404: Not Found. 呃…因为我们并没有创建路由和和展示页面。

创建视图

我们来一起创建第一个展示页面(视图)。我们先创建个文件aiohttpdemo_polls/views.py然后写入:

  1. from aiohttp import web
  2. async def index(request):
  3. return web.Response(text='Hello Aiohttp!')

index就是我们创建的展示页,然后我们创建个路由连接到这个展示页上。我们来把路由放在aiohttpdemo_polls/routes.py文件中(将路由表和模型分开写是很好的实践。创建实际项目时可能会有多个同类文件,这样分开放可以让自己很清楚。):

  1. from views import index
  2. def setup_routes(app):
  3. app.router.add_get('/', index)

我们还要在main.py中调用setup_routes

  1. from aiohttp import web
  2. from routes import setup_routes
  3. app = web.Application()
  4. setup_routes(app)
  5. web.run_app(app, host='127.0.0.1', port=8080)

然后我们重新开启服务器,现在我们从浏览器中访问:

  1. $ curl -X GET localhost:8080
  2. Hello Aiohttp!

啊哈!成功了!我们现在应该有了一个如下所示的目录结构:

  1. .
  2. ├── ..
  3. └── polls
  4. ├── aiohttpdemo_polls
  5. ├── main.py
  6. ├── routes.py
  7. └── views.py

使用配置文件

aiohttp不需要任何配置文件,也没有内置支持任何配置架构。
但考虑到这些事实:

  1. 99%的服务器都有配置文件。
  2. 其他的同类程序(除了以Python为基础的像Django和Flask的)都不会将配置文件作为源码的一部分。
    1. 比如Nginx将配置文件保存在 /etc/nginx文件夹里。
    2. mongo则保存在 /etc/mongodb.conf里。
  3. 使用配置文件是公认的好方法,在部署产品时可以预防一些小错误。

所以我们建议用以下途径(进行配置文件):

  1. 将配置信息写在yaml文件中。(json或ini都可以,但yaml最好用。)
  2. 在一个预先设定好的目录中加载yaml。
  3. 拥有能通过命令行来设置配置文件的功能。如: ./run_app —config=/opt/config/app_cfg.yaml
  4. 对要加载的字典执行严格检测,以确保其数据类型符合预期。可以使用: trafaret, colander 或 JSON schema等库。

以下代码会加载配置文件并设置到应用程序中:

  1. # load config from yaml file in current dir
  2. conf = load_config(str(pathlib.Path('.') / 'config' / 'polls.yaml'))
  3. app['config'] = conf

构建数据库

准备工作

这份指南中我们使用最新版的PostgreSQL数据库。 你可访问以下链接下载: http://www.postgresql.org/download/

数据库架构

我们使用SQLAlchemy来写数据库架构。我们只要创建两个简单的模块——questionchoice:

  1. import sqlalchemy as sa
  2. meta = sa.MetaData()
  3. question - sq.Table(
  4. 'question', meta,
  5. sa.Column('id', sa.Integer, nullable=False),
  6. sa.Column('question_text', sa.String(200), nullable=False),
  7. sa.Column('pub_date', sa.Date, nullable=False),
  8. # Indexes #
  9. sa.PrimaryKeyConstraint('id', name='question_id_pkey')
  10. )
  11. choice = sa.Table(
  12. 'choice', meta,
  13. sa.Column('id', sa.Integer, nullable=False),
  14. sa.Column('question_id', sa.Integer, nullable=False),
  15. sa.Column('choice_text', sa.String(200), nullable=False),
  16. sa.Column('votes', server_default="0", nullable=False),
  17. # Indexes #
  18. sa.PrimayKeyConstraint('id', name='choice_id_pkey'),
  19. sa.ForeignKeyContraint(['question_id'], [question.c.id],
  20. name='choice_question_id_fkey',
  21. ondelete='CASCADE'),
  22. )

你会看到如下数据库结构:

第一张表 question:
|question|
|id|
|question_text|
|pub_date|

第二张表 choice:
|choice|
|id|
|choice_text|
|votes|
|question_id|

创建连接引擎

为了从数据库中查询数据我们需要一个引擎实例对象。假设conf变量是一个带有连接信息的配置字典,Postgres会使用异步的方式完成该操作:

  1. async def init_pg(app):
  2. conf = app['config']
  3. engine = await aiopg.sa.create_engine(
  4. database=conf['database'],
  5. user=conf['user'],
  6. password=conf['password'],
  7. host=conf['host'],
  8. port=conf['host'],
  9. minsize=conf['minsize'],
  10. maxsize=conf['maxsize'])
  11. app['db'] = engine

最好将连接数据库的函数放在on_startup信号中:

  1. app.on_startup.append(init_pg)

关闭数据库

程序退出时一块关闭所有的资源接口是一个很好的做法。
使用on_cleanup信号来关闭数据库接口:

  1. async def close_pg(app):
  2. app['db'].close()
  3. await app['db'].wait_closed()
  4. app.on_cleanup.append(close_pg)

使用模板

我们来添加些更有用的页面:

  1. @aiohttp_jinja2.template('detail.html')
  2. async def poll(request):
  3. async with request['db'].acquire() as conn:
  4. question_id = request.match_info['question_id']
  5. try:
  6. question, choices = await db.get_question(conn,
  7. question_id)
  8. except db.RecordNotFound as e:
  9. raise web.HTTPNotFound(text=str(e))
  10. return {
  11. 'question': question,
  12. 'choices': choices
  13. }

编写页面时使用模板是很方便的。我们返回带有页面内容的字典,aiohttp_jinja2.template装饰器会用jinja2模板加载它。

当然我们要先安装下aiohttp_jinja2:

  1. $ pip install aiohttp_jinja2

安装完成后我们使用时要适配下:

  1. import aiohttp_jinja2
  2. import jinja2
  3. aiohttp_jinja2.setup(
  4. app, loader=jinja2.PackageLoader('aiohttpdemo_polls', 'templates'))

我们将其放在polls/aiohttpdemo_polls/templates文件夹中。

静态文件

每个web站点都有一些静态文件: 图片啦,JavaScript,CSS文件啦等等。
在生产环境中处理这些静态文件最好的方法是使用NGINX或CDN服务做反向代理。
但在开发环境中使用aiohttp服务器处理静态文件是很方便的。

只需要简单的调用一个信号即可:

  1. app.router.add_static('/static/',
  2. path=str(project_root / 'static'),
  3. name='static')

project_root表示根目录。

使用中间件

中间件是每个web处理器必不可少的组件。它的作用是在处理器处理请求前预处理请求以及在得到响应后发送出去。

我们下面来实现一个用于显示漂亮的404和500页面的简单中间件。

  1. def setup_middlewares(app):
  2. error_middleware = error_pages({404: handle_404,
  3. 500: handle_500})
  4. app.middlewares.append(error_middleware)

中间件(middleware)本身是一个接受应用程序(application)后续处理器(next handler)的加工厂。

中间件工厂返回一个与web处理器一样接受请求并返回响应的中间件处理器。

下面实现一个用于处理HTTP异常的中间件:

  1. def error_pages(overrides):
  2. async def middleware(app, handler):
  3. async def middleware_handler(request):
  4. try:
  5. response = await handler(request)
  6. override = overrides.get(response.status)
  7. if override is None:
  8. return response
  9. else:
  10. return await override(request, response)
  11. except web.HTTPException as ex:
  12. override = overrides.get(ex.status)
  13. if override is None:
  14. raise
  15. else:
  16. return await override(request, ex)
  17. return middleware_handler
  18. return middleware

这些overrides(handle_404和handle_500)只是简单的用Jinja2模板渲染:

  1. async def handle_404(request, response):
  2. response = aiohttp_jinja2.render_template('404.html',
  3. request,
  4. {})
  5. return response
  6. async def handle_500(request, response):
  7. response = aiohttp_jinja2.render_template('500.html',
  8. request,
  9. {})
  10. return response

详情看 Middlewares.