96 lines
2.7 KiB
Python
96 lines
2.7 KiB
Python
from flask import Flask, redirect, url_for, render_template
|
|
from flask_login import LoginManager, current_user
|
|
from flask_apscheduler import APScheduler
|
|
from models import db, User
|
|
from routes.auth import auth_bp
|
|
from routes.temp_spec import temp_spec_bp
|
|
from routes.upload import upload_bp
|
|
from routes.admin import admin_bp
|
|
from routes.api import api_bp
|
|
|
|
app = Flask(__name__)
|
|
app.config.from_object('config.Config')
|
|
|
|
# 初始化資料庫
|
|
db.init_app(app)
|
|
|
|
# 初始化排程器
|
|
scheduler = APScheduler()
|
|
scheduler.init_app(app)
|
|
scheduler.start()
|
|
|
|
# 初始化登入管理
|
|
login_manager = LoginManager()
|
|
login_manager.init_app(app)
|
|
login_manager.login_view = 'auth.login'
|
|
login_manager.login_message = "請先登入以存取此頁面。"
|
|
login_manager.login_message_category = "info"
|
|
|
|
# 預設首頁導向登入畫面
|
|
@app.route('/')
|
|
def index():
|
|
# 檢查使用者是否已經通過驗證 (已登入)
|
|
if current_user.is_authenticated:
|
|
# 如果已登入，直接導向到暫規總表
|
|
return redirect(url_for('temp_spec.spec_list'))
|
|
else:
|
|
# 如果未登入，才導向到登入頁面
|
|
return redirect(url_for('auth.login'))
|
|
|
|
# 載入登入使用者
|
|
@login_manager.user_loader
|
|
def load_user(user_id):
|
|
return User.query.get(int(user_id))
|
|
|
|
# 註冊 Blueprint 模組路由
|
|
app.register_blueprint(auth_bp)
|
|
app.register_blueprint(temp_spec_bp)
|
|
app.register_blueprint(upload_bp)
|
|
app.register_blueprint(admin_bp)
|
|
app.register_blueprint(api_bp)
|
|
|
|
# 導入任務
|
|
from tasks import check_expiring_specs
|
|
|
|
# 註冊排程任務：每天凌晨 2:00 執行一次
|
|
@scheduler.task('cron', id='check_expiring_specs_job', hour=2, minute=0)
|
|
def scheduled_job():
|
|
check_expiring_specs(app)
|
|
|
|
# 註冊錯誤處理函式
|
|
@app.errorhandler(404)
|
|
def not_found_error(error):
|
|
return render_template('404.html'), 404
|
|
|
|
@app.errorhandler(403)
|
|
def forbidden_error(error):
|
|
return render_template('403.html'), 403
|
|
|
|
if __name__ == '__main__':
|
|
# 設定日誌等級以便偵錯
|
|
import logging
|
|
import sys
|
|
|
|
# 設定日誌輸出到 console
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s %(levelname)s %(name)s: %(message)s',
|
|
handlers=[
|
|
logging.StreamHandler(sys.stdout)
|
|
]
|
|
)
|
|
|
|
# 設定 Flask app 的日誌
|
|
app.logger.setLevel(logging.INFO)
|
|
app.logger.addHandler(logging.StreamHandler(sys.stdout))
|
|
|
|
# 確保 LDAP 相關的日誌也能輸出
|
|
ldap_logger = logging.getLogger('ldap_utils')
|
|
ldap_logger.setLevel(logging.INFO)
|
|
|
|
print("=== 暫時規範系統 V3 啟動中 ===")
|
|
print("日誌等級: INFO")
|
|
print("系統準備就緒，可開始登入測試...")
|
|
|
|
app.run(debug=True)
|