快速创建Flask Restful API项目
快速创建Flask Restful API项目
前言
Python必学的两大web框架之一Flask,俗称微框架。它只需要一个文件,几行代码就可以完成一个简单的http请求服务。
但是我们需要用flask来提供中型甚至大型web restful api服务的时候就需要我们来规划项目结构。
Django 框架有自身的生成项目结构的命令和生成app的命令,但是Flask没有。
这篇文章主要介绍使用工具帮助我们进行一些Flask rest api项目初始化的工作
cookiecutter-flask-restful 介绍
cookiecutter-flask-restful是一个基于cookiecutter的Flask rest api项目生成器,你可以用它快速生成Flask restful api项目。
目前支持的python版本:3.6 | 3.7 | 3.8
项目特性
Flask应用工厂函数,使用蓝图
集成Flask命令行接口
数据迁移
JWT登录认证
分页工具类
单元测试
使用环境变量配置
集成swagger UI
主要使用的包
Flask
Flask-RESTful
Flask-Migrate
Flask-SQLAlchemy
Flask-Marshmallow
Flask-JWT-Extended
marshmallow-sqlalchemy
passlib
tox
pytest
factoryboy
dotenv
apispec
快速开始
安装cookiecutter
$ pip3 install cookiecutter
创建第一个项目
$ cookiecutter https://github.com/karec/cookiecutter-flask-restful
一路按回车键,就会生成一个新项目。
项目名字暂且叫myproject, app名字暂且叫myapi。
运行项目
如果项目使用了pipenv管理虚拟环境:
安装依赖:
$ cd myproject$ pipenv install -r requirements.txt# 进入到虚拟环境$ pipenv shell$ pip install -e .$ myapi db upgrade$ myapi init$ myapi --help
启动项目:
$ pipenv run flask run
测试项目:
登录请求:
$ curl -X POST -H "Content-Type: application/json" -d '{"username": "admin", "password": "admin"}' http://localhost:5000/auth/login
登录返回:
{ "access_token": "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJpYXQiOjE1ODYyMjUyNzIsIm5iZiI6MTU4NjIyNTI3MiwianRpIjoiZDM3NmViODktYWFkOC00ODZkLWE1NmUtZDdiNmQ3NjNhYmI2IiwiZXhwIjoxNTg2MjI2MTcyLCJpZGVudGl0eSI6MSwiZnJlc2giOmZhbHNlLCJ0eXBlIjoiYWNjZXNzIn0.-xCHCJHS3lKeiar-Z12HSOwd2DxAb9EeKFg9zMKWHfU", "refresh_token": "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJpYXQiOjE1ODYyMjUyNzIsIm5iZiI6MTU4NjIyNTI3MiwianRpIjoiMTcyNmI0ODEtMDJjYi00NWFhLTgyOWUtOGY0ODJlYmFlZTEyIiwiZXhwIjoxNTg4ODE3MjcyLCJpZGVudGl0eSI6MSwidHlwZSI6InJlZnJlc2gifQ.fpFH_vw8Lp-_xjDVoFLjF6HnjOKXR7PCLKIp8v9gSSU"}
获取所有用户请求:
curl -X GET -H "Content-Type: application/json" -H "Authorization: Bearer eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJpYXQiOjE1ODYyMjUyNzIsIm5iZiI6MTU4NjIyNTI3MiwianRpIjoiZDM3NmViODktYWFkOC00ODZkLWE1NmUtZDdiNmQ3NjNhYmI2IiwiZXhwIjoxNTg2MjI2MTcyLCJpZGVudGl0eSI6MSwiZnJlc2giOmZhbHNlLCJ0eXBlIjoiYWNjZXNzIn0.-xCHCJHS3lKeiar-Z12HSOwd2DxAb9EeKFg9zMKWHfU" http://127.0.0.1:5000/api/v1/users
获取所有用户返回:
{ "total": 3, "pages": 1, "next": "/api/v1/users?page=1&per_page=50", "prev": "/api/v1/users?page=1&per_page=50", "results": [ { "active": true, "id": 1, "email": "admin@mail.com", "username": "admin" }, { "active": true, "id": 2, "email": "", "username": "" }, { "active": true, "id": 3, "email": "string", "username": "string" } ]}
刷新token请求:
curl -X POST -H "Content-Type: application/json" -H "Authorization: Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZGVudGl0eSI6MSwiaWF0IjoxNTEwMDAwNDQxLCJ0eXBlIjoicmVmcmVzaCIsImp0aSI6IjRmMjgxOTQxLTlmMWYtNGNiNi05YmI1LWI1ZjZhMjRjMmU0ZSIsIm5iZiI6MTUxMDAwMDQ0MSwiZXhwIjoxNTEyNTkyNDQxfQ.SJPsFPgWpZqZpHTc4L5lG_4aEKXVVpLLSW1LO7g4iU0" http://127.0.0.1:5000/auth/refresh
刷新token返回:
{ "access_token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ0eXBlIjoiYWNjZXNzIiwiaWRlbnRpdHkiOjEsImlhdCI6MTUxMDAwMDYxOCwiZnJlc2giOmZhbHNlLCJqdGkiOiIzODcxMzg4Ni0zNGJjLTRhOWQtYmFlYS04MmZiNmQwZjEyNjAiLCJuYmYiOjE1MTAwMDA2MTgsImV4cCI6MTUxMDAwMTUxOH0.cHuNf-GxVFJnUZ_k9ycoMMb-zvZ10Y4qbrW8WkXdlpw"}
如果没有使用pipenv管理虚拟环境,直接使用的系统python环境:
$ cd myproject$ pip install -r requirements.txt$ pip install -e .# 初始化项目$ myapi db upgrade$ myapi init$ myapi --help
测试项目同上
项目环境变量配置
环境变量在.flaskenv文件中
FLASK_ENV=developmentFLASK_APP="myapp.app:create_app"SECRET_KEY=changemeDATABASE_URI="sqlite:////tmp/myapp.db"CELERY_BROKER_URL=amqp://guest:guest@localhost/ # only present when celery is enabledCELERY_RESULT_BACKEND_URL=amqp://guest:guest@localhost/ # only present when celery is enabled
请根据你的实际情况进行配置。
测试
直接用tox
$ tox
使用pytest
$ pip install pytest pytest-runner pytest-flask pytest-factoryboy factory_boy$ pytest
使用docker-compose 和 Makefile
make tests
如果出现连接失败,你需要更改redis | rabbitmq地址
使用wsgi服务启动
gunicorn示例:
$ pip install gunicorn$ gunicorn myapi.wsgi:app
uwsgi示例:
$ pip install uwsgi$ uwsgi --http 127.0.0.1:5000 --module myapi.wsgi:app
使用celery
启动celery
$ celery worker -A myapi.celery_app:app --loglevel=info
如果你的配置没有问题,在终端你会看到:
[tasks] . myapi.tasks.example.dummy_task
调用task:
>>> from myapi.tasks.example import dummy_task>>> result = dummy_task.delay()>>> result.get()'OK'
使用celery extension:
>>> from myapi.extensions import celery>>> celery.send_task('myapi.tasks.example.dummy_task').get()'OK'
使用Docker
注意:目前Docker的配置只有开发环境
构建镜像:
$ docker build -t myapp .
启动镜像:
$ docker run --env-file=.flaskenv myapp myapi init$ docker run --env-file=.flaskenv -p 5000:5000 myapp myapi run -h 0.0.0.0 * Serving Flask app "myapi.app:create_app" (lazy loading) * Environment: development * Debug mode: on * Running on http://0.0.0.0:5000/ (Press CTRL+C to quit) * Restarting with stat * Debugger is active! * Debugger PIN: 214-619-010
使用docker-compose启动:
$ docker-compose up -d$ docker ps$ docker exec -it <container_id> myapi init
使用docker-compose 和 Makefile启动
make initMakefile 命令make init # 初始化环境make build # 构建镜像make run # 启动镜像make db-migrate # 迁移数据make db-upgrade # 数据映射到数据库make test # 执行测试
使用Swagger
打开UI页面地址:http://localhost:5000/swagger-ui
页面如下:
在config.py文件进行配置:
APISPEC_TITLE: 页面标题,默认: {{cookiecutter.project_name}}
APISPEC_VERSION: api版本,默认: 1.0.0
OPENAPI_VERSION: OpenAPI 版本,默认: 3.0.2
SWAGGER_JSON_URL: JSON规范校验地址, 默认: /swagger.json
SWAGGER_UI_URL: 页面UI地址,默认: /swagger-ui
SWAGGER_URL_PREFIX: 蓝图URL前缀, 默认: None