101 lines
3.3 KiB
Python
101 lines
3.3 KiB
Python
|
# _*_ coding: utf-8 _*_
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
import GetData_zhihu
|
|||
|
|
|
|||
|
|
# flask
|
|||
|
|
from flask import Flask, session, request
|
|||
|
|
from flask import render_template, flash, redirect, url_for, jsonify
|
|||
|
|
|
|||
|
|
# flask extends
|
|||
|
|
from flask_bootstrap import Bootstrap
|
|||
|
|
from flask_wtf import FlaskForm
|
|||
|
|
from wtforms import StringField, PasswordField, SubmitField
|
|||
|
|
from wtforms.validators import Length, Email
|
|||
|
|
|
|||
|
|
# application
|
|||
|
|
app = Flask(__name__)
|
|||
|
|
app.config["SECRET_KEY"] = "hard to guess string"
|
|||
|
|
|
|||
|
|
# manager and bootstrap
|
|||
|
|
bootstrap = Bootstrap(app=app)
|
|||
|
|
|
|||
|
|
# global data
|
|||
|
|
zhihu_all_topics = GetData_zhihu.get_all_topics()
|
|||
|
|
zhihu_all_topics_key = {}
|
|||
|
|
zhihu_init_topics = GetData_zhihu.get_topic_data(topic_id="19559424", topic_name="数据分析")
|
|||
|
|
|
|||
|
|
|
|||
|
|
# form class
|
|||
|
|
class UserForm(FlaskForm):
|
|||
|
|
name = StringField("name", validators=[Email(message="邮箱格式不正确！")])
|
|||
|
|
password = PasswordField("password", validators=[Length(min=6, message="密码长度至少6位！")])
|
|||
|
|
submit = SubmitField("提 交")
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.route("/", methods=["GET", "POST"])
|
|||
|
|
def temp():
|
|||
|
|
return redirect(url_for("index"))
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.route("/index/", methods=["GET", "POST"])
|
|||
|
|
def index():
|
|||
|
|
user_form = UserForm()
|
|||
|
|
if request.method == "POST":
|
|||
|
|
if user_form.validate_on_submit():
|
|||
|
|
session["username"] = user_form.name.data
|
|||
|
|
else:
|
|||
|
|
flash(user_form.errors["name"][0] if "name" in user_form.errors else user_form.errors["password"][0])
|
|||
|
|
else:
|
|||
|
|
if request.args.get("action") == "login_out":
|
|||
|
|
flash("您已成功退出系统！")
|
|||
|
|
session["username"] = None
|
|||
|
|
return redirect(url_for("index"))
|
|||
|
|
elif request.args.get("action") == "overview":
|
|||
|
|
session["page_type"] = "overview"
|
|||
|
|
return redirect(url_for("index"))
|
|||
|
|
elif request.args.get("action") == "zhihu_topics":
|
|||
|
|
session["page_type"] = "zhihu_topics"
|
|||
|
|
return redirect(url_for("index"))
|
|||
|
|
return render_template("index.html", name=session.get("username"), page_type=session.get("page_type", "overview"), form=user_form)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.route("/zhihu_get_topics_list/", methods=["post"])
|
|||
|
|
def zhihu_get_topics_list():
|
|||
|
|
key = request.form.get("key")
|
|||
|
|
result = {"success": 1, "data": []}
|
|||
|
|
if key:
|
|||
|
|
if key in zhihu_all_topics_key:
|
|||
|
|
result = zhihu_all_topics_key[key]
|
|||
|
|
else:
|
|||
|
|
for item in zhihu_all_topics:
|
|||
|
|
if item[1].find(key) >= 0:
|
|||
|
|
result["data"].append({"id": item[0], "name": item[1]})
|
|||
|
|
if len(result["data"]) > 0:
|
|||
|
|
result["success"] = 1
|
|||
|
|
zhihu_all_topics_key[key] = result
|
|||
|
|
logging.debug("all_topics_key increase: %s", len(zhihu_all_topics_key))
|
|||
|
|
return jsonify(result)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.route("/zhihu_get_topics_data/", methods=["post"])
|
|||
|
|
def zhihu_get_topics_data():
|
|||
|
|
if request.form["id"] == "19554449":
|
|||
|
|
result = zhihu_init_topics
|
|||
|
|
else:
|
|||
|
|
result = GetData_zhihu.get_topic_data(request.form["id"], request.form["name"])
|
|||
|
|
return jsonify(result)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@app.errorhandler(404)
|
|||
|
|
def page_not_found(excep):
|
|||
|
|
return render_template("error.html", error=excep, name=session.get("username")), 404
|
|||
|
|
|
|||
|
|
|
|||
|
|
# main process
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
logging.basicConfig(level=logging.DEBUG, format="%(asctime)s\t%(levelname)s\t%(message)s")
|
|||
|
|
logging.debug("app url_map: %s", app.url_map)
|
|||
|
|
|
|||
|
|
app.run()
|