html_view.py

from flask import Blueprint, render_template, request, redirect, url_for
from flask_login import LoginManager, login_user, login_required, logout_user, current_user
from config import app  # 确保从包根目录导入 app 和 db
from services import *
from jwt.exceptions import ExpiredSignatureError, DecodeError
import jwt
from emails import generate_activation_code,send_verification_code
import pandas as pd
from datetime import datetime
from import_data import *

# 创建蓝图对象
views_blueprint = Blueprint('html_view', __name__, url_prefix='/')

# 初始化 Flask-Login
login_manager = LoginManager()
login_manager.init_app(app)
login_manager.session_protection = 'strong'
login_manager.login_view = 'views.login'  # 使用蓝图名称和视图函数名称
login_manager.login_message = u"请先登录。"




# 用户加载函数
@login_manager.user_loader
def load_user(user_id):
    user = db.session.get(Users, user_id)
    if user:
        return user
    else:
        return None



# 主页
@views_blueprint.route('/')
def index():
    faculty_info = get_faculty_info('全职')
    # 取前六个教师信息
    faculty_info = faculty_info[:5]
    news_list = News.query.order_by(News.create_time.desc()).limit(3).all()
    # 将新闻列表转换为 JSON 格式
    news_list_json = [news.to_dict() for news in news_list]
    teaching_works = get_teaching_works_info()
    teaching_works = teaching_works[:6]
    research_works = get_research_works_info()
    research_works = research_works[:6]
    all_achievements = get_achievements_info()
    research_achievements=slice_achievements(all_achievements, 1)
    admission_info = AdmissionInfo.query.limit(3).all()
    cooperation_info = InternationalPartnership.query.limit(3).all()

    return render_template('index.html', faculty=faculty_info, news_list=news_list_json, 
                           teaching_works=teaching_works, research_works=research_works,
                           research_achievements = research_achievements,admission_info=admission_info
                           ,cooperation_info=cooperation_info)

@views_blueprint.route('/research_achievements')
def research_achievements():
    research_achievements = get_achievements_info()
    # 将研究成果数据传递给模板
    return render_template('research_achievements.html',
                           achievements=research_achievements)
@views_blueprint.route('/research')
def research():
    # 将教学和科研信息传递给模板

    teaching_works = get_teaching_works_info()
    research_works = get_research_works_info()
    return render_template('research.html',
                           teaching_works=teaching_works, research_works=research_works)
@views_blueprint.route('/news')
def news():
    # 渲染新闻动态页面并传递新闻数据
    news_data = get_news_info()
    return render_template('news.html', news_data=news_data)

@views_blueprint.route('/admissions')
def admissions():
    admissions_info =get_admissions_info()
    # 将招生信息数据传递给模板
    return render_template('admissions.html', admissions=admissions_info)
@views_blueprint.route('/internal_management')
@login_required
def internal_management():
    return render_template('internal_management.html')


@views_blueprint.route('/user_help')
def user_help():
    return render_template('user_help.html')

@views_blueprint.route('/download_zone')
def download_zone():
    return render_template('download_zone.html')
@views_blueprint.route('/cooperation')
def cooperation():
    cooperation_info = InternationalPartnership.query.all()
    # 将招生信息数据传递给模板
    return render_template('cooperation.html', cooperation=cooperation_info)


@views_blueprint.route('/about')
def about():
    # 获取师资和学生信息
    faculty_info = get_faculty_info('全职')
    faculty_info_part=get_faculty_info('兼职')
    students_info = get_students_by_year()
    students_info_graduate=get_graduated_students_by_year()
    # 渲染模板并传递数据
    return render_template('about.html',
                           faculty=faculty_info,faculty_part=faculty_info_part,
                           students=students_info,students_graduate=students_info_graduate)

@views_blueprint.route('/book_borrowing_page', methods=['GET', 'POST'])
@login_required
def book_borrowing_page():

    return render_template('borrow_book_page.html')
@views_blueprint.route('/book_management', methods=['GET', 'POST'])
@login_required
def book_management():
    # 获取当前用户的ID
    current_user_id = current_user.id
    # 只有所长和全职教师可以访问此路由
    if current_user.user_type == 'admin' or current_user.user_type == 'teacher':
        return render_template('book_management.html')
    else:

        return '您没有权限访问此页面', 403


@views_blueprint.route('/user_stats', methods=['GET', 'POST'])
@login_required
def user_stats():
    # 获取当前用户的ID
    current_user_id = current_user.work_id

    # 只有所长和全职教师可以访问此路由
    if current_user.user_info.user_type == '所长' or current_user.user_info.user_type == '全职教师':
        login_logs = get_login_logs(current_user.user_info.user_type, current_user_id)
        return render_template('user_stats.html', login_logs=login_logs)
    else:
        return '您没有权限访问此页面', 403
# 登录处理
@views_blueprint.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        userid = request.form['userid']
        password = request.form['password']
        user = Users.query.filter_by(work_id=str(userid)).first()
        print(user)

        # 如果用户存在且密码匹配，登录用户
        if user and user.check_password(password):
            if user.is_active:
                session['user_id'] = user.id
                ip_address = request.remote_addr
                login_event = longin_log(user,ip_address)
                db.session.add(login_event)
                db.session.commit()
                login_user(user)  # 确保用户登录后设置登录状态
                return redirect(url_for('html_view.index'))  # 确保用户登录后重定向到 'user_index' 页面
            else:
                flash('您的账户未激活，请先激活。')
                activation_code = generate_activation_code(userid)
                send_verification_code(user.user_info.email, activation_code, userid)
                return redirect(url_for('html_view.activate'))  # 重定向到激活页面
        # 否则，提示用户名或密码错误
        else:
            flash('用户名或密码错误，请检查输入信息。')
    return render_template('login.html')

#处理借阅请求页面
@views_blueprint.route('/request_process', methods=['GET', 'POST'])
@login_required
def request_process():
    return render_template('request_process.html')
# 注册处理
@views_blueprint.route('/activate', methods=['GET', 'POST'])
def activate():
    if request.method == 'POST':
        activation_code = request.form['activation_code']
        try:
            # 解码令牌
            payload = jwt.decode(activation_code, app.config['SECRET_KEY'], algorithms=['HS256'])
            user_id = payload['user_id']
            user = Users.query.filter_by(work_id=str(user_id)).first()
            if user and not user.is_active:
                # 激活用户账户
                user.is_active = True
                session['user_id'] = user.id
                ip_address = request.remote_addr
                login_event = longin_log(user, ip_address)
                db.session.add(login_event)
                db.session.commit()
                # 登录用户并重定向到首页或某个特定页面
                login_user(user)
                flash('账户已成功激活。')
                # 重定向到首页
                return redirect(url_for('html_view.index'))
            else:
                flash('账户已激活或激活码无效。')
                return render_template('activation_failed.html')

        except ExpiredSignatureError:
            # 令牌已过期
            flash('激活码已过期。')
            return render_template('activation_failed.html')
        except DecodeError:
            # 令牌无效或被篡改
            flash('无效的激活代码。')
            return render_template('activation_failed.html')

    return render_template('activation.html')

# 退出登录路由

@views_blueprint.route('/user_page')
@login_required
def user_page():
    user_id = get_current_user_id()  # 假设这是从会话或令牌中获取当前用户ID的函数
    user = Users.query.get(user_id)
    user_loans = BookLoans.query.filter_by(user_id=user.work_id, status='借阅中').all()
    left_days = []
    for loan in user_loans:
        left_days.append(get_left_days(loan))


    # 渲染模板并将用户信息、借阅列表和request_ids传递给模板
    return render_template('user_page.html', user=user, user_loans=user_loans,left_days=left_days)
@views_blueprint.route('/README', methods=['GET'])
def README():
    return render_template('README.html')


@views_blueprint.route('/logout')
@login_required
def logout():
    # 获取当前登录用户的最后一条登录记录
    last_login = db.session.query(LoginEvent).filter_by(user_id=current_user.work_id
                                ).order_by(LoginEvent.id.desc()).first()
    if last_login:
        # 设置登出时间
        last_login.logout_time = datetime.now()
        # 提交数据库更改
        db.session.commit()
    # 清除用户会话
    logout_user()
    flash('您已成功退出登录。')  # 显示退出登录消息
    # 重定向到首页或登录页面
    return redirect(url_for('html_view.index'))

@views_blueprint.route('/user_index', methods=['GET', 'POST'])
@login_required
def user_index():
    return render_template('user_index.html')
# 修改个人信息的路由
@views_blueprint.route('/update_user_info', methods=['POST'])
@login_required
def update_user_info():
    # 从表单中获取新信息
    name = request.form['name']
    gender = request.form['gender']
    category = request.form['category']
    nationality = request.form['nationality']
    admission_time = request.form['admission_time']
    tutor = request.form['tutor']
    # ... 其他信息 ...

    # 更新数据库中的用户信息
    user = current_user  # 获取当前登录的用户
    user.name = name
    user.gender = gender
    user.category = category
    user.nationality = nationality
    user.admission_time = admission_time
    user.tutor = tutor
    # ... 更新其他信息 ...

    # 提交数据库更改
    db.session.commit()

    # 返回用户主页
    return redirect(url_for('html_view.user_page'))
@views_blueprint.route('/news_posting', methods=['GET', 'POST'])
def news_posting():
    return render_template('news_posting.html')
# 导入用户的路由
@views_blueprint.route('/import_data', methods=['GET', 'POST'])
def import_data():
    if request.method == 'POST':
        # 获取上传的文件
        file = request.files['file']
        data_type = request.form['data_type']
        if not file:
            flash('请选择要上传的文件。')
            return redirect(url_for('import_data_view'))
        file_type = get_file_type(file)
        if (file_type =='csv'):
            df= pd.read_csv(file)

        elif(file_type in [ 'xlsx', 'xls']):
            df= pd.read_excel(file,engine='openpyxl')
        else:
            flash('不支持的文件类型。')
            return redirect(url_for('html_view.import_data'))
        if data_type == 'on_campus_students':
            import_on_campus_students_info(df)
        elif data_type == 'graduated_students':
            import_graduated_students_info(df)
        elif data_type == 'teacher':
            import_teacher_info(df)
        elif data_type == 'research_work':
            import_research_work_info(df)
        elif data_type == 'teaching_work':
            import_teaching_work_info(df)
        elif data_type == 'teaching_achievements':
            import_teaching_achievements_info(df)
        elif data_type == 'teaching_papers':
            import_papers_info(df)
        elif data_type == 'textbooks':
            import_textbooks_info(df)
        elif data_type == 'teaching_reform':
            import_teaching_reform_info(df)
        elif data_type == 'research_achievements':
            import_research_achievements_info(df)
        elif data_type == 'research_papers':
            import_papers_info(df)
        elif data_type == 'patents':
            import_patents_info(df)
        elif data_type == 'copyrights':
            import_copyrights_info(df)
        elif data_type == 'admissions':
            import_admissions_info(df)
        elif data_type == 'international_cooperation':
            import_international_cooperation_info(df)
        elif data_type == 'books':
            import_books_info(df)
    return render_template('import_data.html')

# 检查文件是否允许上传
@views_blueprint.route('/news_model', methods=['GET', 'POST'])
def news_model():
    return render_template('news_model.html')
@views_blueprint.route('/upload_materials', methods=['GET', 'POST'])
def upload_materials():
    return render_template('upload_materials.html')
@views_blueprint.route('/materials_model', methods=['GET', 'POST'])
def materials_model():
    return render_template('materials_model.html')

@views_blueprint.route('/system_settings_page', methods=['GET', 'POST'])
def system_settings_page():
    return render_template('system_settings_page.html')

@views_blueprint.route('/introduction_update_page', methods=['GET', 'POST'])
def introduction_update_page():
    return render_template('introduction_update_page.html')

@views_blueprint.route('/test', methods=['GET', 'POST'])
def test():
    return render_template('test.html')

def blueprint(app):
    app.register_blueprint(views_blueprint)