falcon-dashboard/rrd/view/team/team.py

#-*- coding:utf-8 -*-
import json
from flask import request, g, abort, render_template
from rrd import app
from rrd.view.utils import require_login, require_login_json
from rrd.model.team import Team

@app.route("/team/<int:team_id>/users", methods=["GET",])
@require_login()
def team_users(team_id):
    if request.method == "GET":
        try:
            ret = Team.get_team_users(team_id)
        except Exception as e:
            ret = {"msg":str(e)}
        return json.dumps(ret)

@app.route("/team/<team_name>/users", methods=["GET",])
@require_login()
def team_users_by_name(team_name):
    if request.method == "GET":
        try:
            ret = Team.get_team_users_by_name(team_name)
        except Exception as e:
            ret = {"msg":str(e)}
        return json.dumps(ret)

@app.route("/team/list", methods=["GET",])
@require_login()
def team_list():
    if request.method == "GET":
        query_term = request.args.get("query", "")
        teams = Team.get_teams(query_term, g.limit or 20, g.page or 1)
        return render_template("team/list.html", **locals())


@app.route("/team/create", methods=["GET", "POST"])
@require_login()
def team_create():
    if request.method == "GET":
        return render_template("team/create.html", **locals())
    
    if request.method == "POST":
        ret = {"msg":""}

        name = request.form.get("name", "")
        resume = request.form.get("resume", "")
        users = request.form.get("users", "")

        user_ids = users and users.split(",") or []
        user_ids = [int(x) for x in user_ids]

        if not name:
            ret["msg"] = "empty name"
            return json.dumps(ret)
        
        try:
            Team.create_team(name, resume, user_ids)
        except Exception as e:
            ret['msg'] = str(e)
        return json.dumps(ret)

@app.route("/team/<int:team_id>/edit", methods=["GET", "POST"])
@require_login()
def team_edit(team_id):
    if request.method == "GET":
        j = Team.get_team_users(team_id)
        team = Team(j['id'], j['name'], j['resume'], j['creator'], j['creator_name'], [])
        team_user_ids = ",".join([str(x['id']) for x in j['users']])

        return render_template("team/edit.html", **locals())
    
    if request.method == "POST":
        ret = {"msg":""}

        resume = request.form.get("resume", "")
        users = request.form.get("users", "")

        user_ids = users and users.split(",") or []
        user_ids = [int(x) for x in user_ids]

        try:
            Team.update_team(team_id, resume, user_ids)
        except Exception as e:
            ret['msg'] = str(e)
        return json.dumps(ret)

@app.route("/team/<int:team_id>/delete", methods=["POST"])
@require_login_json()
def team_delete(team_id):
    if request.method == "POST":
        ret = {"msg": ""}
        try:
            Team.delete_team(team_id)
        except Exception as e:
            ret['msg'] = str(e)
        return json.dumps(ret)