39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
from flask import Flask, render_template, redirect, url_for
|
|
from loguru import logger
|
|
|
|
from src.config import Config
|
|
from src.generator import build_data
|
|
|
|
app = Flask(__name__)
|
|
logger.add("endofyear.log")
|
|
|
|
|
|
@app.route('/')
|
|
def home():
|
|
# 默认主题 painting
|
|
return redirect(url_for('painting'))
|
|
|
|
|
|
@app.route('/painting')
|
|
def painting():
|
|
if Config("config.ini").web_status:
|
|
# web 服务
|
|
# 如果数据存在，直接返回
|
|
if blog_data := Config("config.ini").blog_data:
|
|
return render_template('painting.html', data=blog_data, web_status=1)
|
|
|
|
# 如果数据不存在，需要生成，并写入配置
|
|
return render_template('painting.html', data=build_data(), web_status=1)
|
|
else:
|
|
# Github 静态
|
|
# 数据需要生成，并写入静态文件
|
|
html_data = render_template('painting.html', data=build_data(), web_status=0)
|
|
with open("static/index.html", "w") as f:
|
|
f.write(html_data)
|
|
|
|
return 'OK'
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(host='0.0.0.0', port=7777, debug=True)
|