"""
main.py - 

Author: Hao
Date: 2021/9/9
"""
import flask
from flask import redirect, session, request, make_response
from flask_cors import CORS

import captcha
from utils import random_captcha_code

app = flask.Flask(__name__)
app.secret_key = '1Qaz2Wsx'
CORS(app)


@app.before_request
def check_login():
    url = request.path
    user_id = session.get('user_id')
    if url == '/' and user_id is None:
        return redirect('/static/lyear_pages_login.html')


@app.route('/')
def show_index():
    return redirect('/static/index.html')


@app.route('/captcha')
def get_captcha_image():
    cap = captcha.Captcha.instance()  # type: captcha.Captcha
    captcha_code = random_captcha_code()
    session['captcha_code'] = captcha_code.lower()
    cap_image_data = cap.generate(captcha_code)
    resp = make_response(cap_image_data)
    resp.headers['content-type'] = 'image/png'
    return resp


@app.route('/login', methods=['POST', ])
def login():
    params = request.json
    captcha_from_user = params.get('captcha').lower()
    captcha_from_sess = session.get('captcha_code')
    if captcha_from_sess != captcha_from_user:
        return {'code': 10001, 'message': '验证码错误'}
    username = params.get('username')
    password = params.get('password')

    return {'code': 10000, 'message': '登录成功'}


if __name__ == '__main__':
    app.run(port=8000, debug=True)