【flask】使用类组织配置-使用工厂函数创建程序实例

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

【需求】

使用配置类管理flask管理测试环境，

通过1个参数即可控制Flask是运行develpment环境还是production环境（数据库配置，邮件配置也要根据环境的变化而变化）

【思路】

1.在config.ini中存储所有的配置信息

2.在settings.py通过DdevelopConfig，ProductionConfig类存取不同测试环境的环境变量值

3.在app.py通过实例化DdevelopConfig或ProductionConfig类,获取这些类中的环境变量的值

在创建Flask对象app之后，通过方法app.config.from_object(obj)将配置信息写入到app.config

这样最终的效果就是：

在创建app时，通过create_app(config_name)，即可灵活的切换测试环境

【代码示例】

目录结构：

config.ini

# Flask对象配置

[app]
FLASK_CONFIG=production

JSON_AS_ASCII =False

JSONIFY_MIMETYPE = "application/json;charset=utf-8"

SECRET_KEY='secret string'

# 数据库配置

[sqlalchemy]

SQLALCHEMY_TRACK_MODIFICATIONS=False

# 线上环境

[production]

DATABASE_URL ='mysql+mysqlconnector://username:password@ip:port/dbname?charset=utf8'

# 开发环境

[development]

DATABASE_URL ='mysql+mysqlconnector://username:password@ip:port/dbname?charset=utf8'

# 邮件配置信息

[email]

MAIL_SERVER="smtp.126.com"

MAIL_USERNAME='xxxx@126.com'

MAIL_PASSWORD="授权码"

settings.py

import os

from configparser import ConfigParser

basedir = os.path.abspath(os.path.dirname(__file__))

cfgpath = os.path.join(basedir, 'config.ini')

cf = ConfigParser()

cf.read(cfgpath)

# 获取邮件配置

def get_email_config(option):

    return cf.get('email', option)

# 获取邮件配置

def get_app_config(option):

    return cf.get('app', option)

# 获取数据库配置

def get_sqlalchemy_config(option):

    return cf.get('sqlalchemy', option)

# 获取生产环境配置

def get_production_config(option):

    return cf.get('production', option)

# 获取开发环境配置

def get_development_config(option):

    return cf.get('development', option)

class BaseConfig(object):

    SECRET_KEY = get_app_config('SECRET_KEY')

    SQLALCHMEY_TRACK_MODIFICATIONS = False

    MAIL_SERVER = get_email_config('MAIL_SERVER')

    MAIL_PORT = 465

    MAIL_USE_SSL = True

    MAIL_USERNAME = get_email_config('MAIL_USERNAME')

    MAIL_PASSWORD = get_email_config('MAIL_USERNAME')

    MAIL_DEFAULT_SENDER = ('Bluelog Admin', MAIL_USERNAME)

class DevelopmentConfig(BaseConfig):

    ENV = 'development'

    AAA = 'in DevelopmentConfig'

    SQLALCHEMY_DATABASE_URI = get_development_config('DATABASE_URL')

class ProductingConfig(BaseConfig):

    ENV = 'production'

    AAA='in ProductingConfig'

    SQLALCHEMY_DATABASE_URI = get_production_config('DATABASE_URL')

config = {

    'production': ProductingConfig,

    'development': DevelopmentConfig

}

app.py

from flask import Flask

from settings import config,get_app_config

def create_app(config_name=None):

    # 不指定config_name，默认取配置文件FLASK_CONFIG的值

    # 指定config_name，则以指定的环境为主

    if config_name is None:

        config_name=get_app_config('FLASK_CONFIG')

    app = Flask(__name__)

    cfobj=config[config_name]

    app.config.from_object(cfobj)

    return app

#测试不指定config_name

app = create_app()

print(app.config['FLASK_ENV'])

print(app.config['AAA'])

print('*'*50)

#测试指定config_name

app = create_app('development')

print(app.config['FLASK_ENV'])

print(app.config['AAA'])

运行结果：

production

in ProductingConfig

**************************************************

development

in DevelopmentConfig

重点:

使用该种方法组织Flask程序后,如果想单独运行models文件(使用if __name__ == '__main__'),查询数据,会报错

if __name__ == '__main__':

    from flask import Flask

    from flask_sqlalchemy import SQLAlchemy

    app = Flask(__name__)

    db = SQLAlchemy(app)

    search_case_objs = TestCase.query.filter_by(creater='王菊')

    print(search_case_objs)

报错如下:

RuntimeError: No application found. Either work inside a view function or push an application context. See http://flask-sqlalchemy.pocoo.org/contexts/.

正确的调试方法:

运行wsgi文件
在Terminal中输入flask shell,在flask shell中导入从models中导入要查询的表,接下来就可以执行curd了

>>> from gotest.models import TestCase

>>> a=TestCase.query.filter_by(creater=<name>)

>>> a

<flask_sqlalchemy.BaseQuery object at 0x7f5c5087c080>

>>> type(a)

<class 'flask_sqlalchemy.BaseQuery'>

>>> a=TestCase.query.filter_by(creater=<name>).all()

>>> a

[<TestCase.a11bcdecdefd1111111111111111111111ffd>, <TestCase.获取xxxxxx评价>]