"""
@Project: flask-vue-master
@Author: ZhangWenLong
@Date: 2024/3/17 12:07
@File: hell
@Email: 1013974064@qq.com
"""

from flask import request
from redis import Redis
from utils import RequestUtils, SecurityUtils, ScanPortsUtils
from utils import create_app
from utils.R import R

apps = create_app()



"""
请求拦截器
"""


@apps.before_request
def check_headers():
    # 白名单，不在白名单里的都需要验证请求头是否携带token
    url = request.path
    whitelist = ['/user/login', '/user/register', '/user/logout', '/queryUsername', '/user/sendEmailValidateCode', '/user/validateEmailCode', '/user/forgetPassword']
    if url in whitelist:
        return None
    else:
        headers = request.headers.get('Authorization')
        if headers is None:
            return R(message="未授权，无法访问系统资源", code=401).to_json()


if __name__ == '__main__':
    apps.run(host='127.0.0.1', port=8080, debug=True)