mirror of
https://github.com/Lost-MSth/Arcaea-server.git
synced 2026-02-09 09:17:26 +08:00
64 lines
1.7 KiB
Python
64 lines
1.7 KiB
Python
#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
|