添加了api接口需要key的规则

This commit is contained in:
MarkusJoe
2022-04-10 14:31:24 +08:00
parent b59bf10970
commit bf9b3fac1f
9 files changed
+151 -40

No files matched your search

+26 -3
View File
@@ -6,6 +6,8 @@
# @File Name: __init__.py
import os
import random
from flask import Flask
from flask_sslify import SSLify
from .main import main as main_blueprint
@@ -13,17 +15,38 @@ from .api import api as api_blueprint
from app.config import config
def generate_random_string(length):
"""
生成随机密码
:param length:
:return:
"""
letters = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
return ''.join(random.choice(letters) for _ in range(length))
def create_app(config_name):
"""创建主app"""
"""
创建 app
:param config_name:
:return:
"""
app = Flask(__name__)
app.config.from_object(config[config_name])
config[config_name].init_app(app)
if config[config_name].SSL_REDIRECT:
sslify = SSLify(app)
if os.environ.get('SSL_REDIRECT'):
SSLify(app)
app.register_blueprint(main_blueprint)
app.register_blueprint(api_blueprint, url_prefix='/api/v1/')
if not os.getenv('ACCESS_KEY'):
access_key = generate_random_string(32)
os.environ['ACCESS_KEY'] = access_key
else:
access_key = os.getenv('ACCESS_KEY')
print('==========Access Key: {}=========='.format(access_key))
return app