#import sqlite3
from flask import (Blueprint, flash, g, redirect,
                   render_template, request, session, url_for)
import functools
import configparser

bp = Blueprint('login', __name__, url_prefix='/web')


@bp.route('/login', methods=('GET', 'POST'))
def login():
    # 登录
    if request.method == 'POST':
        username = request.form['username']
        password = request.form['password']
        error = None

        config = configparser.ConfigParser()
        path = r'setting.ini'
        config.read(path, encoding="utf-8")
        USERNAME = config.get('WEB', 'USERNAME')
        PASSWORD = config.get('WEB', 'PASSWORD')

        if username != USERNAME and password != PASSWORD:
            error = '错误的用户名或密码 Incorrect username or password.'

        if error is None:
            session.clear()
            session['user_id'] = USERNAME + PASSWORD
            return redirect(url_for('index.index'))

        flash(error)

    return render_template('web/login.html')


@bp.route('/logout')
def logout():
    # 登出
    session.clear()
    flash('成功登出 Successfully log out.')
    return redirect(url_for('index.index'))


def login_required(view):
    # 登录验证，写成了修饰器
    @functools.wraps(view)
    def wrapped_view(**kwargs):
        x = session.get('user_id')

        config = configparser.ConfigParser()
        path = r'setting.ini'
        config.read(path, encoding="utf-8")
        USERNAME = config.get('WEB', 'USERNAME')
        PASSWORD = config.get('WEB', 'PASSWORD')

        if x != USERNAME + PASSWORD:
            return redirect(url_for('login.login'))

        g.user = {'user_id': x, 'username': USERNAME}
        return view(**kwargs)

    return wrapped_view