26 lines
1.1 KiB
Python
Executable File
26 lines
1.1 KiB
Python
Executable File
from flask import render_template, request, redirect, url_for, flash
|
|
from flask_login import login_user, login_required, logout_user, current_user
|
|
from scripts.models import db, User
|
|
|
|
def init_auth_routes(app):
|
|
@app.route('/login', methods=['GET', 'POST'])
|
|
def login():
|
|
if request.method == 'POST':
|
|
phone = request.form['phone']
|
|
user = User.query.filter_by(phone=phone).first()
|
|
if user:
|
|
login_user(user)
|
|
app.logger.info(f'用户 {user.name}({user.phone}) 登录成功')
|
|
# 无论是统计员还是装维员，都跳转到仪表盘
|
|
return redirect(url_for('dashboard'))
|
|
else:
|
|
app.logger.warning(f'登录失败：用户不存在 (手机号: {phone})')
|
|
flash('用户不存在')
|
|
return render_template('login.html')
|
|
|
|
@app.route('/logout')
|
|
@login_required
|
|
def logout():
|
|
app.logger.info(f'用户 {current_user.name}({current_user.phone}) 登出')
|
|
logout_user()
|
|
return redirect(url_for('login')) |