Flask项目与clery实践

好久没写博客了,前段时间换工作天天加班,最近不加班了长时间不写就不想写了,今天西安凑合一篇,主要记录一下现在公司的项目实践。之前项目一直用Django框架,现在的公司都用flask,于是新项目也用flask搭建的。

项目结构

├── app
│   ├── __init__.py
│   ├── api
│   │   ├── __init__.py
│   │   ├── serializers       # 存放参数检验的目录
│   │   │   ├── __init__.py
│   │   ├── urls.py           # flask-rest的url映射
│   │   └── views             # 存放视图函数的目录
│   │       ├── __init__.py
│   │       ├── hello.py
│   ├── cache                 # 缓存函数目录
│   │   ├── __init__.py
│   ├── db                    # 数据库目录
│   │   ├── __init__.py
│   │   └── mongo.py
│   ├── service               # 公共服务目录
│   │   ├── __init__.py
│   │   └── email.py
│   ├── tasks                 # celery任务目录
│   │   ├── __init__.py
│   └── utils                 # 公共组件目录
│       ├── __init__.py
└── tests
|    └── __init__.py
├── .env                      # 环境变量文件
├── config.py
├── Dockerfile
├── Pipfile
├── Pipfile.lock
├── docker-compose.yml
├── manage.py                 # flask启动文件
├── readme.md

Celery

官方文档有介绍:

http://flask.pocoo.org/docs/1.0/patterns/celery/

在此基础上我将make_celery放到了app/__init__.py中。

# app/__init__.py
from flask import Flask
from config import config
from celery import Celery


def create_app(config_name):
    app = Flask(__name__)
    app.config.from_object(config[config_name])
    config[config_name].init_app(app)

    from .api import api as api_blueprint
    app.register_blueprint(api_blueprint, url_prefix='/doubao-uecv2/api/v1')

    return app


def make_celery(app):
    celery = Celery()
    celery.config_from_object(app.config)
    class ContextTask(celery.Task):
        def __call__(self, *args, **kwargs):
            with app.app_context():
                return self.run(*args, **kwargs)

    celery.Task = ContextTask
    return celery
# manage.py
import os
from dotenv import load_dotenv

dotenv_path = os.path.join(os.path.dirname(__file__), '.env')
if os.path.exists(dotenv_path):
    load_dotenv(dotenv_path)

from app import create_app, make_celery

app = create_app(os.getenv('FLASK_CONFIG') or 'default')

celery = make_celery(app)

自定义celery的task目录

class Config:
    ...
    CELERY_IMPORTS = ('app.tasks', )
    ...

然后在app/tasks/__init__.py中将各个task导入。

总结

啊啊啊,大晚上的了不想写了,架子也睡半个月前搭建的了,现在也组织不起语言来,就简单写写吧。其实写项目的话还是django好使我觉得,不用装那么多插件或者自己造轮子,也可能是我用的少的缘故,像django的中间件flask有各种handle,但用起来比较别扭,有时候一些功能不知道该放到目录的什么位置。还有日志配置也踩了半天坑。但是如果只是写着玩,一两个功能用flask一个文件几行代码就可以搞定,很方便。当然flask也适合大的项目,只不过前期搭架子得多花些时间。