初始提交:人事共享服务中心钉钉登录功能
This commit is contained in:
44
__init__.py
Normal file
44
__init__.py
Normal file
@@ -0,0 +1,44 @@
|
||||
import os
|
||||
from flask import Flask, request
|
||||
from flask_login import LoginManager
|
||||
from config import Config
|
||||
from models import db, User
|
||||
|
||||
try:
|
||||
from dotenv import load_dotenv
|
||||
load_dotenv()
|
||||
except ImportError:
|
||||
pass
|
||||
|
||||
app = Flask(__name__)
|
||||
app.config.from_object(Config)
|
||||
|
||||
dingtalk_app_key = os.environ.get('DINGTALK_APP_KEY', '')
|
||||
dingtalk_app_secret = os.environ.get('DINGTALK_APP_SECRET', '')
|
||||
dingtalk_agent_id = os.environ.get('DINGTALK_AGENT_ID', '')
|
||||
dingtalk_corp_id = os.environ.get('DINGTALK_CORP_ID', '')
|
||||
|
||||
app.config['DINGTALK_APP_KEY'] = dingtalk_app_key
|
||||
app.config['DINGTALK_APP_SECRET'] = dingtalk_app_secret
|
||||
app.config['DINGTALK_AGENT_ID'] = dingtalk_agent_id
|
||||
app.config['DINGTALK_CORP_ID'] = dingtalk_corp_id
|
||||
app.config['DINGTALK_TARGET_URL'] = os.environ.get('DINGTALK_TARGET_URL', 'http://localhost:5001/requirement-collection')
|
||||
app.config['APPLICATION_ROOT'] = '/requirement-collection'
|
||||
|
||||
db.init_app(app)
|
||||
|
||||
login_manager = LoginManager()
|
||||
login_manager.init_app(app)
|
||||
login_manager.login_view = 'index'
|
||||
|
||||
@login_manager.user_loader
|
||||
def load_user(user_id):
|
||||
return User.query.get(int(user_id))
|
||||
|
||||
from routes import *
|
||||
from dingtalk import dingtalk_bp
|
||||
|
||||
app.register_blueprint(dingtalk_bp, url_prefix='/dingtalk')
|
||||
|
||||
with app.app_context():
|
||||
db.create_all()
|
||||
Reference in New Issue
Block a user