Ejemplo n.º 1
0
def home_data():
    action = request.form['submit']

    if action == "infer":
        info = {}
        info['weather'] = request.form.get("weather")
        info['temperature'] = request.form.get("temperature")
        info['humidity'] = request.form.get("humidity")
        info['wind'] = request.form.get("wind")
        info['gender'] = request.form.get("gender")
        info['cold'] = request.form.get("cold")
        info['style'] = request.form.get("style")

        ret = Interaction.GetRet(info)

        return render_template('index.html',
                               Suggestion=ret[0],
                               Rules=ret[1],
                               Description=ret[2])

    elif action == "all_check":
        with open("rules/rules.txt") as f:
            rules = f.read()
        return render_template('index.html', all=rules.decode('utf-8'))

    elif action == "add":
        rule = {}
        rule["condition"] = request.form.get("add_condition")
        rule["result"] = request.form.get("add_result")
        rule["description"] = request.form.get("add_description")
        ret = Interaction.AddRule(rule)

        with open("rules/rules.txt") as f:
            rules = f.read()

        if ret == True:
            flash(u"添加规则成功", 'success')
        else:
            flash(u"添加规则失败", 'error')

        return render_template('index.html', all=rules.decode('utf-8'))

    elif action == "delete":
        id = request.form.get("delete_id")
        ret = Interaction.DeleteRule(int(id))

        with open("rules/rules.txt") as f:
            rules = f.read()

        if ret == True:
            flash(u"删除规则成功", 'success')
        else:
            flash(u"删除规则失败", 'error')

        return render_template('index.html', all=rules.decode('utf-8'))

    elif action == "modify":
        rule = {}
        rule["id"] = int(request.form.get("modify_id"))
        rule["condition"] = request.form.get("modify_condition")
        rule["result"] = request.form.get("modify_result")
        rule["description"] = request.form.get("modify_description")
        ret = Interaction.ModifyRule(rule)

        with open("rules/rules.txt") as f:
            rules = f.read()

        if ret == True:
            flash(u"修改规则成功", 'success')
        else:
            flash(u"修改规则失败", 'error')

        return render_template('index.html', all=rules.decode('utf-8'))

    elif action == "reset":
        Interaction.Reset()
        with open("rules/rules.txt") as f:
            rules = f.read()
        return render_template('index.html', all=rules.decode('utf-8'))

    else:
        return render_template('index.html')