sukiun
85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
from flask import Flask,session,request,redirect,render_template
|
|
import re
|
|
from apscheduler.schedulers.background import BackgroundScheduler
|
|
import subprocess
|
|
import os
|
|
from pytz import utc
|
|
import logging
|
|
|
|
app = Flask(__name__)
|
|
app.secret_key = 'this is secret_key you know ?'
|
|
|
|
from views.page import page
|
|
from views.user import user
|
|
app.register_blueprint(page.pb)
|
|
app.register_blueprint(user.ub)
|
|
|
|
@app.route('/')
|
|
def hello_world(): # put application's code here
|
|
return session.clear()
|
|
|
|
"""
|
|
@app.before_request
|
|
def before_reuqest():
|
|
pat = re.compile(r'^/static')
|
|
if re.search(pat,request.path):return
|
|
elif request.path == '/user/login' or request.path == '/user/register':return
|
|
elif session.get('username'):return
|
|
return redirect('/user/login')
|
|
"""
|
|
#中间件代码逻辑可以优化，以减少重复的 return 语句，并提高可读性：
|
|
@app.before_request
|
|
def before_request():
|
|
# 静态文件路径允许直接访问
|
|
if request.path.startswith('/static'):
|
|
return
|
|
|
|
# 登录和注册页面无需验证会话
|
|
if request.path in ['/user/login', '/user/register']:
|
|
return
|
|
|
|
# 验证用户是否登录
|
|
if not session.get('username'):
|
|
return redirect('/user/login')
|
|
|
|
@app.route('/<path:path>')
|
|
def catch_all(path):
|
|
return render_template('404.html')
|
|
|
|
def run_script():
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
spider_script = os.path.join(current_dir, 'spider', 'main.py')
|
|
# cutComments_script = os.path.join(current_dir, 'utils', 'cutComments.py')
|
|
# cipingTotal_script = os.path.join(current_dir, 'utils', 'cipingTotal.py')
|
|
|
|
scripts = [
|
|
("Spider Script", spider_script),
|
|
# ("Cut Comments Script", cutComments_script),
|
|
# ("Ciping Total Script", cipingTotal_script)
|
|
]
|
|
|
|
for script_name, script_path in scripts:
|
|
try:
|
|
print(f"Running {script_name}...")
|
|
subprocess.run(['python', script_path], check=True)
|
|
print(f"{script_name} finished successfully.")
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"An error occurred while running {script_name}: {e}")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
scheduler = BackgroundScheduler(timezone=utc)
|
|
scheduler.add_job(run_script, 'interval', hours=5)
|
|
scheduler.start()
|
|
|
|
try:
|
|
app.run()
|
|
finally:
|
|
scheduler.shutdown()
|
|
|
|
#为了更好地调试和监控，建议为应用添加日志记录，捕获用户请求和错误：
|
|
logging.basicConfig(level=logging.INFO)
|
|
@app.before_request
|
|
def log_request_info():
|
|
logging.info(f"Request: {request.method} {request.path}")
|