python＋flak请假系统

from flask import Flask, render_template, redirect, url_for, request, flash from flask_sqlalchemy import SQLAlchemy from flask_login import LoginManager, UserMixin, login_user, login_required, logout_user from werkzeug.security import generate_password_hash, check_password_hash app = Flask(__name__) app.config['SECRET_KEY'] = 'your_secret_key' app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///site.db' db = SQLAlchemy(app) login_manager = LoginManager(app) login_manager.login_view = 'login' # 用户模型 class User(db.Model, UserMixin): id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(150), unique=True, nullable=False) password = db.Column(db.String(150), nullable=False) @login_manager.user_loader def load_user(user_id): return User.query.get(int(user_id)) # 路由定义 @app.route('/') def home(): return redirect(url_for('login')) @app.route('/register', methods=['GET', 'POST']) def register(): if request.method == 'POST': username = request.form['username'] password = request.form['password'] if User.query.filter_by(username=username).first(): flash('用户名已存在，请选择其他用户名', 'danger') return render_template('register.html') new_user = User(username=username, password=generate_password_hash(password)) try: db.session.add(new_user) db.session.commit() flash('注册成功！', 'success') return redirect(url_for('login')) except Exception: db.session.rollback() flash('注册失败，请稍后再试', 'danger') return render_template('register.html') @app.route('/login', methods=['GET', 'POST']) def login(): if request.method == 'POST': username = request.form['username'] password = request.form['password'] user = User.query.filter_by(username=username).first() if user and check_password_hash(user.password, password): login_user(user) return redirect(url_for('index')) # 重定向到 index flash('登录失败，请检查用户名和密码', 'danger') return render_template('login.html') @app.route('/index') @login_required def index(): return render_template('index.html') @app.route('/dashboard') @login_required def dashboard(): return render_template('dashboard.html') @app.route('/leave_request', methods=['GET', 'POST']) @login_required def leave_request(): if request.method == 'POST': flash('请假申请已提交！', 'success') return redirect(url_for('dashboard')) return render_template('leave_request.html') @app.route('/leave_history') @login_required def leave_history(): return render_template('leave_history.html') @app.route('/leave_progress') @login_required def leave_progress(): return render_template('leave_progress.html') @app.route('/logout') @login_required def logout(): logout_user() flash('您已成功登出！', 'success') return redirect(url_for('login')) @app.route('/profile/settings', methods=['GET', 'POST']) @login_required def profile_settings(): if request.method == 'POST': flash('设置已更新！', 'success') return redirect(url_for('profile_settings')) return render_template('profile_settings.html') @app.route('/feedback', methods=['GET', 'POST']) @login_required def feedback(): if request.method == 'POST': flash('反馈已提交！', 'success') return redirect(url_for('dashboard')) return render_template('feedback.html') @app.route('/help', methods=['GET']) def help(): return render_template('help.html') @app.route('/faq', methods=['GET']) def faq(): return render_template('faq.html') if __name__ == '__main__': with app.app_context(): db.create_all() app.run(debug=True)