index.py

# -*- coding: utf-8 -*-

from flask import Flask, session, escape
from flask import request, url_for, render_template, jsonify, redirect
from db import userdb, newdb
import json
from newsCatalog import NEWSCATALOG
import sendEmail
NEWSCATALOG = {
    "data": [
        {
            "source": "中山大学教务处",
            "list": ["辅修双专业双学位", "转专业", "信息荟萃", "教学改革研究项目", "交流生课程建设项目", \
            "毕业学位", "证件办理", "专业培养方案", "课程简介", "人才培养基地建设", "课程建设", \
            "学籍信息管理", "专业建设", "卓越人才培养项目", "教学论文集", "教学成果奖",\
             "学生科研", "推荐免试研究生", "选课管理" ,"课室管理", "考试", "教学质量检查与分析", "其他事项","其他项目"]
        },
        {
            "source": "中山大学软件学院",
            "list": ["学院概况", "新闻中心", "招生信息", "教学教务", "学生工作", "学术科研"]
        }
    ]
}

app = Flask(__name__)

@app.route("/")
def main_page():
  user = session.get('user', None)
  if user  != None:
    return render_template('index.html', path='/', user=user)
  else:
    return redirect('/login')

@app.route("/login")
def  login():
    return render_template('login.html', path='/login')

@app.route("/register")
def  register():
    return render_template('register.html', path='/register')

@app.route("/ucenter")
def  ucenter():
  user = session.get('user', None)
  if user  != None:
    return render_template('ucenter.html', path='/ucenter', user=user)
  else:
    return redirect('/login')

@app.route("/news")
def  news():
  user = session.get('user', None)
  if user  != None:
    catalog =  request.args.get('catalog');
    return render_template('news.html', path='/news', catalog=catalog, user=user)
  else:
    return redirect('/login')

@app.route("/api/<param>")
def api_default(param):
    return  "%s" % param

@app.route("/apiTemp/login", methods=["POST"])
def apiTemp_login():
    if request.method == "POST":
        formJson = request.get_json()
        print formJson
        returnJson = userdb.login({ 'loginID': escape(formJson["loginID"]),
                                                   'password': escape(formJson['password']) })
        if returnJson.has_key("uuid"):
            session["user"] = userdb.get({"uuid": returnJson["uuid"]}).get("user", {})
        return jsonify(returnJson)

@app.route("/apiTemp/signup", methods=["POST"])
def apiTemp_signup():
    if request.method == "POST":
        formJson = request.get_json()
        returnJson = userdb.register({  'loginID': escape(formJson["loginID"]), \
                                                        'password': escape(formJson['password']), \
                                                        'name': escape(formJson["name"]) if formJson.has_key("name") else "", \
                                                        'subscribe': escape(formJson["subscribe"]) if formJson.has_key("subscribe") else [], \
                                                        'email': escape(formJson["email"] if formJson.has_key("email") else "")})
        return jsonify(returnJson)

@app.route("/apiTemp/update", methods=["POST"])
def apiTemp_update():
    if request.method == "POST": 
        formJson = request.get_json()
        returnJson = userdb.update({  'uuid': escape(formJson['uuid']) if formJson.has_key("uuid") else session["user"]["uuid"], \
                                                        'password': escape(formJson['password']) if formJson.has_key("password") else '', \
                                                        'name': escape(formJson["name"]) if formJson.has_key("name") else session["user"]["name"], \
                                                        # subscribe might cause secure problem
                                                        'subscribe': formJson["subscribe"] if formJson.has_key("subscribe") else session["user"]["subscribe"], \
                                                        'email': escape(formJson["email"]) if formJson.has_key("email") else session["user"]["email"] }) 
        return jsonify(returnJson)

@app.route("/apiTemp/get", methods=["POST"])
def apiTemp_get():
    if request.method == "POST":
        returnJson = userdb.get({  'uuid': session["user"]["uuid"] if session["user"].has_key("uuid") else "" })
        return jsonify(returnJson)

@app.route("/apiTemp/getInfo", methods=["POST"])
def apiTemp_getInfo():
    if request.method == "POST":
        formJson = request.get_json()
        returnJson = userdb.getUserInfo({ 'loginID': escape(formJson["loginID"]),
                                                   'password': escape(formJson['password']) })
        return jsonify(returnJson)

# return newsCatalog
@app.route("/apiTemp/getNewsCatalog", methods=["POST"])
def apiTemp_getNewsCatalog():
    if request.method == "POST":
        return jsonify(NEWSCATALOG)


# return news When you give a newCatalog
@app.route("/apiTemp/getNewsByCatalog", methods=["POST"])
def apiTemp_getNewsByCatalog():
    if request.method == "POST":
        formJson = request.get_json()
        returnJson = newdb.get({"newCatalog": escape(formJson["newCatalog"]) if formJson.has_key("newCatalog") else ''})
        return jsonify(returnJson)

# return newContent When you give a new uuid
@app.route("/apiTemp/getNewContentByuuid", methods=["POST"])
def apiTemp_getNewContentByuuid():
    if request.method == "POST":
        formJson = request.get_json()
        returnJson = newdb.getContent({"uuid": escape(formJson["uuid"]) if formJson.has_key("uuid") else ''})
        return jsonify(returnJson)

@app.route("/apiTemp/sendEmail", methods=["POST"])
def apiTemp_sendEmail():
    if request.method == "POST":
        formJson = request.get_json()
        return jsonify(sendEmail.sendEmail(escape(formJson["loginID"]) if formJson.has_key("loginID") else session["user"]["loginID"], \
            formJson["subscribe"] if formJson.has_key("subscribe") else session["user"]["subscribe"]))


@app.route("/apiTemp/logout", methods=["POST"])
def apiTemp_logout():
    if request.method == "POST":
        session["user"] = None
        return jsonify({"result": True})

@app.route("/apiTemp/<param>")
def apiTemp_default(param):
    return  "%s" % param

app.secret_key = '我有一头小毛驴我从来也不骑啊~'

if __name__ == '__main__':
    app.debug = True
    app.run(host='0.0.0.0')